Update app.py

app.py

@@ -7,6 +7,10 @@ from qdrant_search import QdrantSearch
 from langchain_groq import ChatGroq
 from nomic_embeddings import EmbeddingsModel
 
+from fastapi import FastAPI, HTTPException
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+
 # Load environment variables
 load_dotenv()
 
@@ -80,6 +84,40 @@ def chat_function(question: str):
 
     return answer.content.strip(), sources
 
+# Define Pydantic model for request
+class ChatRequest(BaseModel):
+    question: str
+
+# Initialize FastAPI app
+app = FastAPI()
+
+# Define allowed origins
+origins = [
+    "*",  # Allow all origins; for production, specify your frontend domains
+    # Example:
+    # "http://localhost",
+    # "http://localhost:3000",
+    # "https://your-frontend-domain.com",
+]
+
+# Add CORS middleware
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=origins,            # Allows all origins. Replace "*" with specific domains in production.
+    allow_credentials=True,
+    allow_methods=["*"],              # Allows all HTTP methods.
+    allow_headers=["*"],              # Allows all headers.
+)
+
+# Define API endpoint
+@app.post("/api/chat")
+async def api_chat(request: ChatRequest):
+    try:
+        answer, sources = chat_function(request.question)
+        return {"answer": answer, "sources": sources}
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+
 # Create Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("# 🗨️ LangAssist Chat")
@@ -107,9 +145,13 @@ with gr.Blocks() as demo:
     ---
     ## 📡 API Endpoint
 
-    Since we're running on Hugging Face Spaces, direct API access is not available. Please use the interactive interface above to ask your questions.
+    You can access the API endpoint at `/api/chat`. For example, send a POST request to `http://localhost:8000/api/chat` with JSON body `{"question": "Your question here"}`.
     """)
 
-# Launch the Gradio App
+# Mount Gradio app on FastAPI
+app = gr.mount_gradio_app(app, demo, path="/gradio")
+
+# To run, use: uvicorn app:app --host 0.0.0.0 --port 8000
 if __name__ == "__main__":
-    demo.launch()
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)