Update app.py

app.py

@@ -9,6 +9,10 @@ def log_message(msg, logs):
     return logs
 
 def upload_and_index(files, logs):
+    """
+    1) رفع وبناء الفهرس من المقاطع
+    2) يقوم تلقائيًا بتلخيص كل مقطع بعد الانتهاء من بناء الفهرس
+    """
     logs = log_message("[RAG] بدء معالجة الملفات...", logs)
     all_chunks = []
     for file in files:
@@ -17,56 +21,70 @@ def upload_and_index(files, logs):
         all_chunks.extend(chunks)
         logs = log_message(f"[RAG] تم استخراج {len(chunks)} مقطع من {file.name}", logs)
 
+    # بناء الفهرس
     rag.build_index(all_chunks)
     logs = log_message("[RAG] تم بناء الفهرس.", logs)
-    return logs, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
 
-def summarize_chunks(logs):
-    logs = log_message("[RAG] بدأ التلخيص للمقاطع...", logs)
+    # تلخيص المقاطع تلقائيًا
+    logs = log_message("[RAG] بدأ التلخيص التلقائي لجميع المقاطع...", logs)
     rag.summarize_all_chunks()
     logs = log_message("[RAG] تم التلخيص لجميع المقاطع.", logs)
-    return logs
 
-def answer_question(question, logs, summary_input):
+    # إظهار زر الإجابة بعد اكتمال التلخيص
+    return logs, gr.update(visible=True), gr.update(visible=True)
+
+def answer_question(question, logs):
+    """
+    توليد الإجابة بناءً على الملخصات المحفوظة
+    """
     logs = log_message(f"[RAG] استلام السؤال: {question}", logs)
 
-    if not summary_input.strip():
-        logs = log_message("[RAG] لا يوجد ملخص صالح. يرجى تلخيص المقاطع أولاً.", logs)
-        return "", summary_input, logs
-
-    # توليد الإجابة بناءً على الملخص
-    qa_prompt = f"أجب عن السؤال التالي بناء على النص:\n\n{summary_input}\n\nالسؤال: {question}\nالإجابة:"
-    try:
-        inputs = rag.tokenizer(qa_prompt, return_tensors="pt", truncation=True, max_length=512)
-        output_ids = rag.model.generate(inputs["input_ids"], max_length=200)
-        answer = rag.tokenizer.decode(output_ids[0], skip_special_tokens=True).strip()
-        if not answer:
-            logs = log_message("[RAG] فشل في توليد إجابة من النموذج.", logs)
-    except Exception as e:
-        logs = log_message(f"[RAG] خطأ أثناء توليد الإجابة: {str(e)}", logs)
-        answer = ""
-
-    logs = log_message("[RAG] تم توليد الإجابة.", logs)
-    return answer, summary_input, logs
+    # استدعاء إجابة باستخدام التلخيص المحفوظ
+    answer, sources, combined_summary = rag.answer(question)
+
+    if not combined_summary.strip():
+        logs = log_message("[RAG] لم يتم إنشاء ملخص صالح. تأكد من رفع وتلخيص المقاطع.", logs)
+        return "", logs
+
+    if not answer.strip():
+        logs = log_message("[RAG] فشل في توليد الإجابة من النموذج.", logs)
+    else:
+        logs = log_message("[RAG] تم توليد الإجابة.", logs)
+        logs = log_message(f"[RAG] المصادر: {sources}", logs)
+
+    return answer, combined_summary, logs
 
 with gr.Blocks() as demo:
     logs = gr.State("")
     gr.Markdown("# 🕌 نظام استرجاع المعرفة باللغة العربية")
 
     with gr.Row():
-        files_input = gr.File(file_types=[".pdf", ".docx", ".txt"], file_count="multiple", label="📂 رفع الملفات")
+        files_input = gr.File(
+            file_types=[".pdf", ".docx", ".txt"],
+            file_count="multiple",
+            label="📂 رفع الملفات"
+        )
         upload_btn = gr.Button("🔄 رفع وبناء الفهرس")
 
-    summarize_btn = gr.Button("🧠 تلخيص المقاطع", visible=False)
-    question_input = gr.Textbox(label="❓ اكتب سؤالك هنا", visible=False)
     answer_btn = gr.Button("✍️ أجب عن السؤال", visible=False)
+    question_input = gr.Textbox(label="❓ اكتب سؤالك هنا", visible=False)
 
     logs_output = gr.Textbox(label="📜 سجل العمليات", lines=10, interactive=False)
     summary_output = gr.Textbox(label="📌 الملخص المستخدم", lines=5)
     answer_output = gr.Textbox(label="✅ الإجابة النهائية", lines=5)
 
-    upload_btn.click(upload_and_index, inputs=[files_input, logs], outputs=[logs_output, summarize_btn, question_input, answer_btn])
-    summarize_btn.click(summarize_chunks, inputs=logs, outputs=logs_output)
-    answer_btn.click(answer_question, inputs=[question_input, logs, summary_output], outputs=[answer_output, summary_output, logs_output])
+    upload_btn.click(
+        upload_and_index,
+        inputs=[files_input, logs],
+        outputs=[logs_output, answer_btn, question_input]
+    )
+
+    # بعد اكتمال التلخيص التلقائي، نج��ل زر "أجب عن السؤال" و "اكتب سؤالك" مرئيين
+    # ومن ثم يمكن للمستخدم كتابة السؤال والضغط مباشرة على "أجب عن السؤال"
+    answer_btn.click(
+        answer_question,
+        inputs=[question_input, logs],
+        outputs=[answer_output, summary_output, logs_output]
+    )
 
 demo.launch()