Update main.py

main.py

@@ -1,49 +1,20 @@
 from fastapi import FastAPI, File, UploadFile, Form, HTTPException
-from fastapi.responses import HTMLResponse, FileResponse
-from fastapi.staticfiles import StaticFiles
+from fastapi.responses import JSONResponse
 from gradio_client import Client
 import os
-import tempfile
-import shutil
 
 app = FastAPI()
 
 # Initialize Gradio Client
 hf_token = os.environ.get('HF_TOKEN')
-client = Client("Ashrafb/moondream1", hf_token=hf_token)
+client = Client("https://ashrafb-moondream1.hf.space/--replicas/o2fdl/")
 
-# Function to generate captions for uploaded images
-def get_caption(image, additional_context):
+@app.post("/predict/")
+async def predict(image_path: str = Form(...), question: str = Form(...)):
     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
+        result = client.predict(image_path, question, api_name="/predict")
+        return JSONResponse(content={"result": 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(...)):
-    try:
-        with tempfile.TemporaryDirectory() as temp_dir:
-            temp_image_path = os.path.join(temp_dir, image.filename)
-            with open(temp_image_path, "wb") as temp_image:
-                shutil.copyfileobj(image.file, temp_image)
-            with open(temp_image_path, "rb") as image_file:
-                image_data = image_file.read()
-            result = get_caption(image_data, additional_context)
-            return {"result": result}
-    except Exception as e:
-        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")
-
-# Serve index.html
-@app.get("/")
-def index() -> FileResponse:
-    return FileResponse(path="/app/static/index.html", media_type="text/html")
+        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})