Update app.py

app.py

@@ -9,10 +9,6 @@ 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:
@@ -21,36 +17,30 @@ 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)
 
-    # تلخيص المقاطع تلقائيًا
-    logs = log_message("[RAG] بدأ التلخيص التلقائي لجميع المقاطع...", logs)
-    rag.summarize_all_chunks()
-    logs = log_message("[RAG] تم التلخيص لجميع المقاطع.", logs)
+    # Summarize only first N chunks for performance
+    logs = log_message("[RAG] بدأ التلخيص التلقائي لأول 20 مقطع...", logs)
+    rag.summarize_all_chunks(max_chunks=20)
+    logs = log_message("[RAG] تم التلخيص لجميع المقاطع المحددة.", logs)
 
-    # إظهار زر الإجابة بعد اكتمال التلخيص
     return logs, gr.update(visible=True), gr.update(visible=True)
 
 def answer_question(question, logs):
-    """
-    توليد الإجابة بناءً على الملخصات المحفوظة
-    """
     logs = log_message(f"[RAG] استلام السؤال: {question}", logs)
 
-    # استدعاء إجابة باستخدام التلخيص المحفوظ
     answer, sources, combined_summary = rag.answer(question)
 
     if not combined_summary.strip():
-        logs = log_message("[RAG] لم يتم إنشاء ملخص صالح. تأكد من رفع وتلخيص المقاطع.", logs)
+        logs = log_message("[RAG] لم يتم إنشاء ملخص صالح.", logs)
         return "", logs
 
     if not answer.strip():
-        logs = log_message("[RAG] فشل في توليد الإجابة من النموذج.", logs)
+        logs = log_message("[RAG] لم يتم توليد إجابة. حاول صياغة السؤال بشكل أوضح.", logs)
     else:
-        logs = log_message("[RAG] تم توليد الإجابة.", logs)
-        logs = log_message(f"[RAG] المصادر: {sources}", logs)
+        logs = log_message("[RAG] تم توليد الإجابة بنجاح.", logs)
+        logs = log_message(f"[RAG] المقاطع المستخدمة: {sources}", logs)
 
     return answer, combined_summary, logs
 
@@ -59,32 +49,17 @@ with gr.Blocks() as demo:
     gr.Markdown("# 🕌 نظام استرجاع المعرفة باللغة العربية")
 
     with gr.Row():
-        files_input = gr.File(
-            file_types=[".pdf", ".docx", ".txt"],
-            file_count="multiple",
-            label="📂 رفع الملفات"
-        )
-        upload_btn = gr.Button("🔄 رفع وبناء الفهرس")
+        files_input = gr.File(file_types=[".pdf", ".docx", ".txt"], file_count="multiple", label="📂 رفع الملفات")
+        upload_btn = gr.Button("🔄 رفع وبناء الفهرس وتلخيص")
 
-    answer_btn = gr.Button("✍️ أجب عن السؤال", visible=False)
     question_input = gr.Textbox(label="❓ اكتب سؤالك هنا", visible=False)
+    answer_btn = gr.Button("✍️ أجب عن السؤال", 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, answer_btn, question_input]
-    )
-
-    # بعد اكتمال التلخيص التلقائي، نجعل زر "أجب عن السؤال" و "اكتب سؤالك" مرئيين
-    # ومن ثم يمكن للمستخدم كتابة السؤال والضغط مباشرة على "أجب عن السؤال"
-    answer_btn.click(
-        answer_question,
-        inputs=[question_input, logs],
-        outputs=[answer_output, summary_output, logs_output]
-    )
+    upload_btn.click(upload_and_index, inputs=[files_input, logs], outputs=[logs_output, question_input, answer_btn])
+    answer_btn.click(answer_question, inputs=[question_input, logs], outputs=[answer_output, summary_output, logs_output])
 
 demo.launch()