Update main.py

main.py

@@ -14,12 +14,16 @@ client = Client("Ashrafb/moondream1", hf_token=hf_token)
 
 # Function to generate captions for uploaded images
 def get_caption(image, additional_context):
-    if additional_context.strip():  
-        context = additional_context
-    else:
-        context = "What is this image? Describe this image to someone who is visually impaired."
-    result = client.predict(image, context, api_name="/predict")
-    return result  
+    try:
+        if additional_context.strip():  
+            context = additional_context
+        else:
+            context = "What is this image? Describe this image to someone who is visually impaired."
+        result = client.predict(image, context, api_name="/predict")
+        return result
+    except Exception as e:
+        error_msg = "Error occurred while generating caption: {}".format(str(e))
+        raise HTTPException(status_code=500, detail={"error": error_msg})
 
 @app.post("/uploadfile/")
 async def upload_file(image: UploadFile = File(...), additional_context: str = Form(...)):
@@ -32,14 +36,9 @@ async def upload_file(image: UploadFile = File(...), additional_context: str = F
                 image_data = image_file.read()
             result = get_caption(image_data, additional_context)
             return {"result": result}
-    except HTTPException as http_exc:
-        # Handle HTTPException raised by FastAPI
-        raise http_exc
     except Exception as e:
-        # Handle other unexpected exceptions
-        error_msg = "An unexpected error occurred while processing the request."
-        detail_msg = str(e)
-        raise HTTPException(status_code=500, detail={"error": error_msg, "detail": detail_msg})
+        error_msg = "Error occurred while processing image: {}".format(str(e))
+        raise HTTPException(status_code=500, detail={"error": error_msg})
 
 # Serve static files
 app.mount("/", StaticFiles(directory="static", html=True), name="static")