Update main.py

main.py

@@ -1,4 +1,5 @@
 import os
+import tempfile
 from fastapi import FastAPI, File, UploadFile, Form
 from fastapi.responses import JSONResponse
 from gradio_client import Client, handle_file
@@ -18,18 +19,33 @@ except Exception as e:
 @app.post("/get_caption")
 async def get_caption(image: UploadFile = File(...), context: str = Form(...)):
     try:
-        contents = await image.read()
-        # Remove orig_name if it's not supported by handle_file
-        image_data = handle_file(contents)
+        # Create a temporary file
+        with tempfile.NamedTemporaryFile(delete=False) as temp_file:
+            # Write the uploaded file contents to the temp file
+            contents = await image.read()
+            temp_file.write(contents)
+            temp_file_path = temp_file.name
 
+        # Use the temporary file path with handle_file or any other processing
+        image_data = handle_file(temp_file_path)
+
+        # Call the Gradio API
         result = client.predict(
             image=image_data,
             question=context,
             api_name="/answer_question"
         )
 
+        # Return the result as a JSON response
         return JSONResponse(content={"caption": result})
+
     except Exception as e:
         print(f"Error during prediction: {e}")
         return JSONResponse(content={"error": str(e)}, status_code=500)
 
+    finally:
+        # Remove the temporary file
+        if os.path.exists(temp_file_path):
+            os.remove(temp_file_path)
+
+# Run the server with: uvicorn main:app --reload