Update main.py

main.py

@@ -44,9 +44,13 @@ async def enhance_image(
         )
 
         # Read the result image and encode it in base64
-        with open(result[0], "rb") as img_file:
-            b64_string = base64.b64encode(img_file.read()).decode('utf-8')
+        with open(result[0], "rb") as image_file:
+                image_data = base64.b64encode(image_file.read()).decode("utf-8")
 
+            return {
+                "sketch_image_base64": f"data:image/png;base64,{image_data}",
+                "result_file": result[1]
+            }
         # Clean up the temporary file
         os.remove(temp_file_path)