Update main.py

from fastapi import FastAPI, File, UploadFile, HTTPException
from fastapi.responses import JSONResponse
from gradio_client import Client

app = FastAPI()
client = Client("https://ashrafb-image-to-sketch.hf.space/")



@app

.post("/predict")
async def predict(file: UploadFile = File(...)):
try:
# Save the uploaded image to a temporary file
with NamedTemporaryFile(delete=False) as tmp:
shutil.copyfileobj(file.file, tmp)
tmp_path = tmp.name

# Call the Gradio client to predict
result = client.predict(tmp_path, api_name="/predict")

# Delete the temporary file
os.unlink(tmp_path)

# Parse and return the result
sketch_image_url, result_file_url = result
return JSONResponse(content={"sketch_image_url": sketch_image_url, "result_file_url": result_file_url}, status_code=200)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))

main.py

@@ -3,6 +3,8 @@ from fastapi import FastAPI, File, UploadFile, Form, Request
 from fastapi.responses import HTMLResponse, FileResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import JSONResponse
 from fastapi.responses import StreamingResponse
 from gradio_client import Client
 import os
@@ -13,12 +15,24 @@ hf_token = os.environ.get('HF_TOKEN')
 client = Client("https://ashrafb-image-to-sketch.hf.space/", hf_token=hf_token)
 
 @app.post("/predict")
-async def predict_sketch(file: UploadFile = File(...)):
-    content = await file.read()
-    # Call the Gradio client to get the sketch result
-    result = client.predict(content, api_name="/predict")
-    sketch_image_bytes = result[0].encode('utf-8')  # Assuming result[0] is a base64 encoded image
-    return StreamingResponse(io.BytesIO(sketch_image_bytes), media_type="image/png")
+async def predict(file: UploadFile = File(...)):
+    try:
+        # Save the uploaded image to a temporary file
+        with NamedTemporaryFile(delete=False) as tmp:
+            shutil.copyfileobj(file.file, tmp)
+            tmp_path = tmp.name
+
+        # Call the Gradio client to predict
+        result = client.predict(tmp_path, api_name="/predict")
+
+        # Delete the temporary file
+        os.unlink(tmp_path)
+
+        # Parse and return the result
+        sketch_image_url, result_file_url = result
+        return JSONResponse(content={"sketch_image_url": sketch_image_url, "result_file_url": result_file_url}, status_code=200)
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
 
 app.mount("/", StaticFiles(directory="static", html=True), name="static")