changes

README.md

@@ -1,11 +0,0 @@
----
-title: Moffitt Rag Demo
-emoji: 💬
-colorFrom: yellow
-colorTo: purple
-sdk: gradio
-sdk_version: 5.0.1
-app_file: app.py
-pinned: false
-short_description: API Hosting
----

app.py

@@ -1,47 +1,43 @@
-"""
-FastAPI + Gradio combo app for Hugging Face Space
-"""
-import os, gradio as gr
+# app.py  – FastAPI core with public Gradio UI
+import os
 from fastapi import FastAPI, Header, HTTPException
 from pydantic import BaseModel
-from rag import retrieve_info     # your existing function
+import gradio as gr
+from rag import retrieve_info  # <-- function defined in rag.py
 
-# -------- 1)  Private key list ----------
-VALID_KEYS = set(os.getenv("RAG_KEYS", "alpha,beta").split(","))  # store secret in Space > Settings > Secrets
+# =============== Config ===============
+VALID_KEYS = set(os.getenv("RAG_KEYS", "alpha").split(","))  # add real keys in Space » Settings » Secrets
 
-# -------- 2)  FastAPI core app ----------
-app = FastAPI(title="Moffitt RAG API")
+# =============== FastAPI ==============
+api = FastAPI(title="Moffitt RAG API")
 
 class QueryIn(BaseModel):
     query: str
     k: int = 5
 
-@app.post("/v1/query")
+@api.post("/v1/query")
 async def rag_query(body: QueryIn, x_api_key: str = Header(None)):
-    """
-    Secure JSON endpoint.
-    Caller must send:  X-API-Key: <one-of-valid-keys>
-    """
     if x_api_key not in VALID_KEYS:
         raise HTTPException(status_code=401, detail="Invalid or missing X-API-Key")
+    answer = retrieve_info(body.query, body.k)
+    return {"answer": answer}
 
-    text = retrieve_info(body.query, body.k)
-    return {"answer": text}
-
-# -------- 3)  Public Gradio UI ----------
-def run(q, k):
+# =============== Public Gradio UI ==============
+def ui_fn(q, k):
     return retrieve_info(q, int(k))
 
 demo = gr.Interface(
-    fn=run,
-    inputs=["text", gr.Number(label="k (Number of chunks to retrieve)")],
+    fn=ui_fn,
+    inputs=["text", gr.Slider(1, 20, value=5, step=1, label="Top-k")],
     outputs=gr.Textbox(lines=25, label="Retrieved chunks"),
+    title="🔍 Moffitt Retrieval Demo",
     allow_flagging="never",
-    title="Moffitt RAG Demo",
-    description="Type a question; we search Chroma with E5 embeddings."
 )
 
-# Mount Gradio at ROOT path "/"
-if __name__ == "__main__":
-    demo.launch()
+# Mount UI at "/"
+app = gr.mount_gradio_app(api, demo, path="/")
 
+# ---------- Local debugging ----------
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)