Upload 2 files

Dockerfile

@@ -22,4 +22,4 @@ RUN pip install --no-cache-dir -r requirements.txt
 EXPOSE $PORT
 
 # Command to run FastAPI with uvicorn
-CMD ["sh", "-c", "uvicorn main:app --host 0.0.0.0 --port ${PORT:-8000}"]
+CMD ["sh", "-c", "uvicorn main:app --host 0.0.0.0 --port ${PORT:-7860}"]

main.py

@@ -6,6 +6,7 @@ import faiss
 import pickle
 import redis
 from fastapi import FastAPI, HTTPException
+from fastapi.responses import RedirectResponse
 from pydantic import BaseModel
 from sentence_transformers import SentenceTransformer
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
@@ -98,6 +99,10 @@ def clear_pending(session_id: str):
 # Core endpoint
 # --------------------------
 
+@app.get("/", include_in_schema=False)
+def root():
+    return RedirectResponse(url="/docs")
+
 @app.post("/chat", response_model=ChatResponse)
 def chat(req: ChatRequest):
     # Check if there is a pending confirmation for this session