Fix: Run backend on port 7860 for HF Spaces

Co-authored-by: Qwen-Coder <qwen-coder@alibabacloud.com>

Dockerfile

@@ -2,19 +2,16 @@ FROM python:3.10-slim
 
 WORKDIR /app
 
-# Install system deps
-RUN apt-get update && apt-get install -y curl && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get install -y gcc g++ curl && rm -rf /var/lib/apt/lists/*
 
-# Set env
 ENV PYTHONUNBUFFERED=1
 
-# Install deps
-COPY backend/requirements.txt .
+COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy code
-COPY backend/ .
+COPY backend/ ./backend/
+COPY app.py ./app.py
 
-# Start
-EXPOSE 8000
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+EXPOSE 7860
+
+CMD ["uvicorn", "backend.main:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,66 @@
+"""
+Simple RAG - Streamlit App
+Combined frontend and backend for HF Spaces
+"""
+import os
+import sys
+import logging
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+logger.info("=== Simple RAG Starting ===")
+
+import streamlit as st
+import httpx
+
+# Configuration
+st.set_page_config(page_title="Simple RAG", page_icon="🤖", layout="centered")
+
+SUPPORTED_LANGUAGES = [
+    {"code": "hi", "name": "Hindi", "native": "हिंदी"},
+    {"code": "bn", "name": "Bengali", "native": "বাংলা"},
+    {"code": "gu", "name": "Gujarati", "native": "ગુજરાતી"},
+    {"code": "kn", "name": "Kannada", "native": "ಕನ್ನಡ"},
+    {"code": "ml", "name": "Malayalam", "native": "മലയാളം"},
+    {"code": "mr", "name": "Marathi", "native": "मराठी"},
+    {"code": "or", "name": "Odia", "native": "ଓଡ଼ିଆ"},
+    {"code": "pa", "name": "Punjabi", "native": "ਪੰਜਾਬੀ"},
+    {"code": "ta", "name": "Tamil", "native": "தமிழ்"},
+    {"code": "te", "name": "Telugu", "native": "తెలుగు"},
+]
+
+st.title("🤖 Simple RAG System")
+st.markdown("Ask questions in any supported Indic language.")
+
+# Language selection
+language_options = [f"{lang['native']} ({lang['name']})" for lang in SUPPORTED_LANGUAGES]
+selected_language = st.selectbox("Choose a language", options=language_options, index=0)
+selected_lang_code = SUPPORTED_LANGUAGES[language_options.index(selected_language)]["code"]
+
+# Query input
+query = st.text_area("Ask Your Question", height=100, placeholder=f"Type in {selected_language}...")
+
+if st.button("🔍 Search"):
+    if not query.strip():
+        st.warning("Please enter a question")
+    else:
+        with st.spinner("Searching..."):
+            try:
+                response = httpx.post(
+                    "http://localhost:8000/query",
+                    json={"query": query, "top_k": 5, "language": selected_lang_code},
+                    timeout=120.0
+                )
+                if response.status_code == 200:
+                    result = response.json()
+                    st.write("### Answer")
+                    st.write(result["answer"])
+                else:
+                    st.error(f"Error: {response.status_code}")
+            except Exception as e:
+                st.error(f"Error: {str(e)}")
+
+st.markdown("---")
+st.markdown("Simple RAG - Powered by Sarvam-1 and Qdrant")

backend/main.py

@@ -1,9 +1,10 @@
 """
-Minimal FastAPI backend - test if container starts
+FastAPI Backend for Simple RAG - HF Spaces
 """
 import os
 import sys
 import logging
+from pathlib import Path
 
 # Setup logging FIRST
 logging.basicConfig(
@@ -13,30 +14,91 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
-logger.info("=== CONTAINER STARTING ===")
-logger.info(f"Python version: {sys.version}")
-logger.info(f"QDRANT_URL env set: {bool(os.getenv('QDRANT_URL'))}")
-logger.info(f"HF_TOKEN env set: {bool(os.getenv('HF_TOKEN'))}")
+logger.info("=== Simple RAG Backend Starting ===")
+logger.info(f"Python: {sys.version}")
 
-from fastapi import FastAPI
+# Add src to path
+sys.path.insert(0, str(Path(__file__).parent / "src"))
+
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from typing import Optional, List, Dict
+
+logger.info("Importing modules...")
+
+try:
+    from rag_system import SimpleRAGSystem
+    from language_constants import get_supported_languages
+    logger.info("✓ Modules imported")
+except Exception as e:
+    logger.error(f"✗ Import failed: {e}")
+    raise
 
 app = FastAPI(title="Simple RAG API")
 
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+class QueryRequest(BaseModel):
+    query: str
+    top_k: int = 5
+    language: Optional[str] = None
+
+class QueryResponse(BaseModel):
+    question: str
+    answer: str
+    user_selected_language: Optional[str]
+    supported_languages: List[Dict[str, str]]
+
+rag_system: Optional[SimpleRAGSystem] = None
+
+def get_rag_system():
+    global rag_system
+    if rag_system is None:
+        logger.info("Initializing RAG system...")
+        rag_system = SimpleRAGSystem()
+        logger.info("✓ RAG initialized")
+    return rag_system
+
+@app.on_event("startup")
+async def startup():
+    logger.info("=== Startup ===")
+    logger.info(f"QDRANT_URL: {bool(os.getenv('QDRANT_URL'))}")
+    logger.info(f"HF_TOKEN: {bool(os.getenv('HF_TOKEN'))}")
+    try:
+        get_rag_system()
+        logger.info("✓ Startup complete")
+    except Exception as e:
+        logger.error(f"✗ Startup failed: {e}")
+
 @app.get("/")
 def root():
-    logger.info("Root endpoint called")
     return {"status": "running", "message": "Simple RAG API"}
 
 @app.get("/health")
 def health():
-    logger.info("Health check called")
     return {"status": "healthy"}
 
-@app.get("/test")
-def test():
-    return {"env": dict(os.environ)}
+@app.get("/languages")
+def languages():
+    return {"languages": get_supported_languages()}
+
+@app.post("/query")
+async def query(req: QueryRequest):
+    try:
+        result = get_rag_system().query(req.query, req.top_k, req.language)
+        return QueryResponse(**result)
+    except Exception as e:
+        logger.error(f"Query error: {e}")
+        raise HTTPException(500, str(e))
 
 if __name__ == "__main__":
     import uvicorn
-    logger.info("=== Starting uvicorn on 0.0.0.0:8000 ===")
-    uvicorn.run(app, host="0.0.0.0", port=8000, log_level="info")
+    logger.info("Starting uvicorn on 0.0.0.0:8000")
+    uvicorn.run(app, host="0.0.0.0", port=8000)

docker-compose.yml

@@ -2,15 +2,9 @@ version: '3.8'
 
 services:
   backend:
-    build:
-      context: .
-      dockerfile: Dockerfile
+    build: .
     ports:
-      - "8000:8000"
+      - "7860:7860"
     env_file:
       - .env
-    environment:
-      - QDRANT_URL=${QDRANT_URL:-}
-      - QDRANT_API_KEY=${QDRANT_API_KEY:-}
-      - HF_TOKEN=${HF_TOKEN:-}
     restart: unless-stopped