File size: 4,206 Bytes
ea85083 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 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
# 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) |