Create app.py

app.py

@@ -0,0 +1,24 @@
+from fastapi import FastAPI, UploadFile, File
+from pydantic import BaseModel
+from router import route_request
+from PIL import Image
+import io
+
+app = FastAPI()
+
+class ChatRequest(BaseModel):
+    prompt: str
+
+@app.post("/chat")
+async def chat(request: ChatRequest):
+    response = route_request(request.prompt)
+    return {"response": response}
+
+@app.post("/chat-image")
+async def chat_image(prompt: str, file: UploadFile = File(...)):
+    image_bytes = await file.read()
+    image = Image.open(io.BytesIO(image_bytes))
+
+    response = route_request(prompt, image=image)
+    return {"response": response}
+