Update main.py

main.py

@@ -7,7 +7,6 @@ import os
 
 app = FastAPI()
 
-
 HF_TOKEN = os.getenv("HF_TOKEN")
 
 # Initialize the Gradio client with the token
@@ -45,19 +44,17 @@ async def enhance_image(
 
         # Read the result image and encode it in base64
         with open(result[0], "rb") as image_file:
-                image_data = base64.b64encode(image_file.read()).decode("utf-8")
+            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)
 
-        return JSONResponse(content={"sketch_image_base64": f"data:image/png;base64,{b64_string}"})
+        return JSONResponse(content={
+            "sketch_image_base64": f"data:image/png;base64,{image_data}",
+            "result_file": result[1]
+        })
 
     except Exception as e:
         # Log the error message for debugging
         print(f"Error processing image: {e}")
-        return JSONResponse(status_code=500, content={"message": "Internal Server Error"})
-
+        return JSONResponse(status_code=500, content={"message": "Internal Server Error"})