Update app.py

app.py

@@ -1,30 +1,31 @@
-import os
-import gradio as gr
-from rag_system import RAGSystem  # تأكد أن الملف اسمه rag_system.py وموجود في نفس المجلد
+# app.py
 
-# --- تحميل النظام ---
-rag = RAGSystem(
-    vector_db_path="vector_db",  # تأكد أن قاعدة البيانات مضغوطة ومرفوعة ضمن files
-    model_name="meta-llama/Llama-3-8b-chat-hf",
-    embedding_model_name="sentence-transformers/all-mpnet-base-v2"
+from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel
+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"
 )
 
+# تحميل النظام عند بدء التشغيل
+rag = RAGSystem(vector_db_path="vector_db")
 rag.load_vectorstore()
 rag.load_llm()
 rag.get_prompt_template()
 
-# --- دالة الاستجابة ---
-def ask_question_interface(question):
-    return rag.ask_question(question)
-
-# --- واجهة Gradio ---
-demo = gr.Interface(
-    fn=ask_question_interface,
-    inputs=gr.Textbox(label="Ask a question about the ECC Guide"),
-    outputs=gr.Textbox(label="Answer"),
-    title="ECC Cybersecurity Assistant",
-    description="Ask questions based on the Essential Cybersecurity Controls (ECC) Implementation Guide issued by the NCA (Saudi Arabia).",
-)
-
-if __name__ == "__main__":
-    demo.launch()
+# نقطة النهاية
+@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))