Update app.py

app.py

@@ -1,31 +1,18 @@
-# app.py
-
-from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+from fastapi import FastAPI, Request
 from rag_system import RAGSystem
 
-# نموذج بيانات الطلب
-class QuestionRequest(BaseModel):
-    question: str
-
-# إنشاء التطبيق
-app = FastAPI(
-    title="ECC Compliance Assistant API",
-    description="Ask questions based on the ECC Cybersecurity Guide using RAG",
-    version="1.0.0"
-)
+app = FastAPI()
 
-# تحميل النظام عند بدء التشغيل
 rag = RAGSystem(vector_db_path="vector_db")
 rag.load_vectorstore()
 rag.load_llm()
 rag.get_prompt_template()
 
-# نقطة النهاية
 @app.post("/ask")
-async def ask_question(data: QuestionRequest):
-    try:
-        answer = rag.ask_question(data.question)
-        return {"question": data.question, "answer": answer}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+async def ask_question(request: Request):
+    data = await request.json()
+    question = data.get("question")
+    if not question:
+        return {"error": "Question is required"}
+    answer = rag.ask_question(question)
+    return {"answer": answer}