Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
import os
|
| 2 |
os.environ["HF_HOME"] = "/tmp/huggingface"
|
| 3 |
-
|
| 4 |
from fastapi import FastAPI
|
| 5 |
from fastapi.middleware.cors import CORSMiddleware
|
| 6 |
from pydantic import BaseModel
|
|
@@ -60,6 +60,21 @@ def ask_question(request: QueryRequest):
|
|
| 60 |
"RetrievedContext": context,
|
| 61 |
"answer": answer
|
| 62 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 63 |
@app.get("/")
|
| 64 |
def read_root():
|
| 65 |
return {"message": "✅ RAG Chatbot API is running. Use POST /query to ask questions."}
|
|
|
|
| 1 |
import os
|
| 2 |
os.environ["HF_HOME"] = "/tmp/huggingface"
|
| 3 |
+
from fastapi.responses import HTMLResponse
|
| 4 |
from fastapi import FastAPI
|
| 5 |
from fastapi.middleware.cors import CORSMiddleware
|
| 6 |
from pydantic import BaseModel
|
|
|
|
| 60 |
"RetrievedContext": context,
|
| 61 |
"answer": answer
|
| 62 |
}
|
| 63 |
+
|
| 64 |
+
|
| 65 |
+
|
| 66 |
+
|
| 67 |
+
@app.get("/", response_class=HTMLResponse)
|
| 68 |
+
def home():
|
| 69 |
+
return """
|
| 70 |
+
<h2>🤖 RAG Chatbot</h2>
|
| 71 |
+
<form method="post" action="/query">
|
| 72 |
+
<input name="query" placeholder="Ask a question" style="width:300px;" />
|
| 73 |
+
<button type="submit">Ask</button>
|
| 74 |
+
</form>
|
| 75 |
+
"""
|
| 76 |
+
|
| 77 |
+
|
| 78 |
@app.get("/")
|
| 79 |
def read_root():
|
| 80 |
return {"message": "✅ RAG Chatbot API is running. Use POST /query to ask questions."}
|