Update backend/Server.py

backend/Server.py

@@ -1,71 +1,146 @@
-from fastapi import FastAPI, HTTPException, Request
-from fastapi.middleware.cors import CORSMiddleware
-from rag_engine import GetRagResponse, ExtractStructure
-import uvicorn
-import os
-from dotenv import load_dotenv
-
-load_dotenv()
-
-App = FastAPI()
-
-App.add_middleware(
-    CORSMiddleware,
-    allow_origins=["*"],
-    allow_methods=["*"],
-    allow_headers=["*"],
-)
-
-@App.get("/")
-def HealthCheck():
-    return {"status": "Online"}
-
-# --- CHAT ENDPOINT ---
-@App.post("/chat")
-@App.post("/api/chat")
-async def ChatRoute(request: Request):
-    # 1. Read JSON body flexibly
-    try:
-        data = await request.json()
-    except:
-        raise HTTPException(status_code=400, detail="Invalid JSON")
-
-    # 2. Look for ANY common key
-    user_text = data.get("message") or data.get("Msg") or data.get("query") or data.get("question")
-    
-    if not user_text:
-        raise HTTPException(status_code=422, detail="No message found. Please send JSON with 'message' key.")
-
-    # 3. Process
-    try:
-        return GetRagResponse(user_text)
-    except Exception as e:
-        print(f"Error: {e}")
-        return {"answer": "Sorry, the backend encountered an error processing your request.", "sources": []}
-
-# --- EXTRACT ENDPOINT ---
-@App.post("/extract")
-@App.post("/api/extract")
-async def ExtractRoute(request: Request):
-    # 1. Read JSON
-    try:
-        data = await request.json()
-    except:
-        raise HTTPException(status_code=400, detail="Invalid JSON")
-
-    # 2. Look for ANY common key
-    user_text = data.get("message") or data.get("Cmd") or data.get("instruction") or data.get("query")
-
-    if not user_text:
-        raise HTTPException(status_code=422, detail="No command found. Please send JSON with 'message' key.")
-
-    # 3. Process
-    try:
-        return ExtractStructure(user_text)
-    except Exception as e:
-        print(f"Error: {e}")
-        return {"data": [], "sources": []}
-
-if __name__ == "__main__":
-    # Listen on all interfaces
-    uvicorn.run(App, host="0.0.0.0", port=8000)
+# from fastapi import FastAPI, HTTPException, Request
+# from fastapi.middleware.cors import CORSMiddleware
+# from rag_engine import GetRagResponse, ExtractStructure
+# import uvicorn
+# import os
+# from dotenv import load_dotenv
+
+# load_dotenv()
+
+# App = FastAPI()
+
+# App.add_middleware(
+#     CORSMiddleware,
+#     allow_origins=["*"],
+#     allow_methods=["*"],
+#     allow_headers=["*"],
+# )
+
+# @App.get("/")
+# def HealthCheck():
+#     return {"status": "Online"}
+
+# # --- CHAT ENDPOINT ---
+# @App.post("/chat")
+# @App.post("/api/chat")
+# async def ChatRoute(request: Request):
+#     # 1. Read JSON body flexibly
+#     try:
+#         data = await request.json()
+#     except:
+#         raise HTTPException(status_code=400, detail="Invalid JSON")
+
+#     # 2. Look for ANY common key
+#     user_text = data.get("message") or data.get("Msg") or data.get("query") or data.get("question")
+    
+#     if not user_text:
+#         raise HTTPException(status_code=422, detail="No message found. Please send JSON with 'message' key.")
+
+#     # 3. Process
+#     try:
+#         return GetRagResponse(user_text)
+#     except Exception as e:
+#         print(f"Error: {e}")
+#         return {"answer": "Sorry, the backend encountered an error processing your request.", "sources": []}
+
+# # --- EXTRACT ENDPOINT ---
+# @App.post("/extract")
+# @App.post("/api/extract")
+# async def ExtractRoute(request: Request):
+#     # 1. Read JSON
+#     try:
+#         data = await request.json()
+#     except:
+#         raise HTTPException(status_code=400, detail="Invalid JSON")
+
+#     # 2. Look for ANY common key
+#     user_text = data.get("message") or data.get("Cmd") or data.get("instruction") or data.get("query")
+
+#     if not user_text:
+#         raise HTTPException(status_code=422, detail="No command found. Please send JSON with 'message' key.")
+
+#     # 3. Process
+#     try:
+#         return ExtractStructure(user_text)
+#     except Exception as e:
+#         print(f"Error: {e}")
+#         return {"data": [], "sources": []}
+
+# if __name__ == "__main__":
+#     # Listen on all interfaces
+#     uvicorn.run(App, host="0.0.0.0", port=8000)
+
+
+
+from fastapi import FastAPI, HTTPException, Request
+from fastapi.middleware.cors import CORSMiddleware
+# 1. IMPORT THE INGESTION LOGIC
+from ingestion import IngestDocuments 
+from rag_engine import GetRagResponse, ExtractStructure
+import uvicorn
+import os
+from dotenv import load_dotenv
+
+load_dotenv()
+
+App = FastAPI()
+
+App.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+@App.get("/")
+def HealthCheck():
+    return {"status": "Online", "platform": "Hugging Face Spaces"}
+
+# --- INGEST ENDPOINT ---
+@App.post("/ingest")
+@App.post("/api/ingest")
+async def TriggerIngest():
+    print("--- INGESTION TRIGGERED BY API ---")
+    try:
+        # This actually runs the code inside ingestion.py
+        IngestDocuments() 
+        return {"status": "Success", "message": "Ingestion process finished. Embeddings loaded."}
+    except Exception as e:
+        print(f"INGESTION FAILED: {e}")
+        return {"status": "Error", "message": str(e)}
+
+# --- CHAT ENDPOINT ---
+@App.post("/chat")
+@App.post("/api/chat")
+async def ChatRoute(request: Request):
+    try:
+        data = await request.json()
+    except:
+        raise HTTPException(status_code=400, detail="Invalid JSON")
+
+    user_text = data.get("message") or data.get("Msg") or data.get("query") or data.get("question")
+    
+    if not user_text:
+        raise HTTPException(status_code=422, detail="No message found.")
+
+    return GetRagResponse(user_text)
+
+# --- EXTRACT ENDPOINT ---
+@App.post("/extract")
+@App.post("/api/extract")
+async def ExtractRoute(request: Request):
+    try:
+        data = await request.json()
+    except:
+        raise HTTPException(status_code=400, detail="Invalid JSON")
+
+    user_text = data.get("message") or data.get("Cmd") or data.get("instruction") or data.get("query")
+
+    if not user_text:
+        raise HTTPException(status_code=422, detail="No command found.")
+
+    return ExtractStructure(user_text)
+
+if __name__ == "__main__":
+    # PORT TO 7860 (REQUIRED FOR HUGGING FACE DOCKER)
+    uvicorn.run(App, host="0.0.0.0", port=7860)