add FastAPI wrapper for /api/chat endpoint

Dockerfile

@@ -7,7 +7,7 @@ RUN apt-get update -qq && apt-get install -y -qq python3 python3-pip python3-ven
 RUN python3 -m venv /venv
 ENV PATH=/venv/bin:$PATH
 
-RUN pip3 install -q gradio requests
+RUN pip3 install -q gradio requests fastapi uvicorn python-multipart
 
 WORKDIR /app
 COPY app.py .

app.py

@@ -1,9 +1,23 @@
 import gradio as gr
 import requests, json, os, time
+from fastapi import FastAPI, Request
+from fastapi.responses import JSONResponse
+import uvicorn
 
 OLLAMA = "http://localhost:11434"
 MODEL = "FableForge-AI/shellwhisperer"
 
+fastapi_app = FastAPI(docs_url=None, redoc_url=None)
+
+@fastapi_app.post("/api/chat")
+async def api_chat(request: Request):
+    body = await request.json()
+    try:
+        r = requests.post(f"{OLLAMA}/api/chat", json=body, timeout=120)
+        return JSONResponse(content=r.json())
+    except Exception as e:
+        return JSONResponse(status_code=500, content={"error": str(e)})
+
 # ── helpers ──
 
 def model_info():
@@ -81,7 +95,14 @@ def generate(prompt, temp):
 
 with gr.Blocks(
     title="ShellWhisperer-1.5B · API",
+    theme=gr.themes.Soft(),
     fill_height=True,
+    css="""footer { display: none !important; }
+.status-ok { color: #22c55e; font-weight: 600; }
+.status-loading { color: #f59e0b; font-weight: 600; }
+.status-err { color: #ef4444; font-weight: 600; }
+.api-box { background: #1f2937; color: #e5e7eb; padding: 1em; border-radius: 8px; font-family: monospace; font-size: 0.9em; overflow-x: auto; }
+""",
 ) as demo:
 
     gr.Markdown("""# 🐚 ShellWhisperer-1.5B · API Demo
@@ -201,15 +222,7 @@ print(r.json()["message"]["content"])
         outputs=out,
     )
 
+app = gr.mount_gradio_app(fastapi_app, demo, path="/")
+
 if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        theme=gr.themes.Soft(),
-        css="""footer { display: none !important; }
-.status-ok { color: #22c55e; font-weight: 600; }
-.status-loading { color: #f59e0b; font-weight: 600; }
-.status-err { color: #ef4444; font-weight: 600; }
-.api-box { background: #1f2937; color: #e5e7eb; padding: 1em; border-radius: 8px; font-family: monospace; font-size: 0.9em; overflow-x: auto; }
-""",
-    )
+    uvicorn.run(app, host="0.0.0.0", port=7860)