Update app.py

app.py

@@ -1,33 +1,17 @@
-from fastapi import FastAPI, File, UploadFile
-from fastapi.responses import JSONResponse
-from ollama import Ollama
+from fastapi import FastAPI, UploadFile, File
 from PIL import Image
-import io
+import base64
+import ollama
 
-app = FastAPI(title="Civic Image Analysis API - LLaVA")
-
-# Initialize Ollama client
-client = Ollama()
+app = FastAPI()
+client = ollama.Client()
 
 @app.post("/analyze")
-async def analyze(file: UploadFile = File(...), prompt: str = "Classify the civic issue in this image"):
-    try:
-        # Read image bytes
-        image_bytes = await file.read()
-        image = Image.open(io.BytesIO(image_bytes)).convert("RGB")
-        
-        # Save temporarily to pass path to Ollama (if required)
-        image_path = f"/tmp/{file.filename}"
-        image.save(image_path)
-        
-        # Call Ollama model
-        response = client.call(
-            model="llava",
-            prompt=prompt,
-            image_path=image_path
-        )
-        
-        return JSONResponse({"analysis": response})
-
-    except Exception as e:
-        return JSONResponse({"error": str(e)}, status_code=500)
+async def analyze(file: UploadFile = File(...)):
+    image_bytes = await file.read()
+    image_b64 = base64.b64encode(image_bytes).decode("utf-8")
+    
+    prompt = "Classify the civic issue in this image as JSON with keys: issue_type, severity, department."
+    response = client.run("llava", f"{prompt}", image=image_b64)
+    
+    return {"analysis": response}