Update app.py

app.py

@@ -33,13 +33,18 @@ def retrieve_passages(question, chunks, logs):
     logs = log_message(f"[RAG] تم العثور على {len(passages)} مقطع مرتبط بالسؤال.", logs)
     return passages, passages_text, logs
 
-def summarize_answer(question, passages_text, logs):
-    answer, summary = rag.generate_answer_from_passages(question, passages_text)
+def generate_summary_only(passages_text, logs):
+    summary = rag.summarize_text(passages_text)
+    logs = log_message("[RAG] تم توليد الملخص بنجاح.", logs)
+    return summary, logs
+
+def generate_final_answer(question, summary_text, logs):
+    answer, _ = rag.generate_answer_from_passages(question, summary_text)
     if not answer.strip():
         logs = log_message("[RAG] ⚠️ لم يتم توليد إجابة. ربما النص طويل أو النموذج فشل.", logs)
     else:
         logs = log_message("[RAG] ✅ تم توليد الإجابة النهائية.", logs)
-    return summary, answer, logs
+    return answer, logs
 
 with gr.Blocks() as demo:
     logs = gr.State("")
@@ -58,16 +63,18 @@ with gr.Blocks() as demo:
         store_question_btn = gr.Button("📥 تخزين السؤال")
 
     find_btn = gr.Button("🔍 بحث عن المقاطع المرتبطة")
-    answer_btn = gr.Button("✍️ تلخيص وتوليد الإجابة")
+    summarize_btn = gr.Button("📝 توليد الملخص")
+    answer_btn = gr.Button("✍️ توليد الإجابة النهائية")
 
     passage_output = gr.Textbox(label="📄 المقاطع المرتبطة", lines=8)
     summary_output = gr.Textbox(label="📌 الملخص المستخدم", lines=5)
     answer_output = gr.Textbox(label="✅ الإجابة النهائية", lines=5)
     logs_output = gr.Textbox(label="📜 سجل العمليات", lines=10, interactive=False)
 
-    upload_btn.click(upload_and_prepare, inputs=[files_input, logs], outputs=[chunks, logs_output, store_question_btn, find_btn, answer_btn])
+    upload_btn.click(upload_and_prepare, inputs=[files_input, logs], outputs=[chunks, logs_output, store_question_btn, find_btn, summarize_btn])
     store_question_btn.click(store_question, inputs=question_input, outputs=question_state)
     find_btn.click(retrieve_passages, inputs=[question_state, chunks, logs], outputs=[retrieved_passages, passage_output, logs_output])
-    answer_btn.click(summarize_answer, inputs=[question_state, passage_output, logs], outputs=[summary_output, answer_output, logs_output])
+    summarize_btn.click(generate_summary_only, inputs=[passage_output, logs], outputs=[summary_output, logs_output])
+    answer_btn.click(generate_final_answer, inputs=[question_state, summary_output, logs], outputs=[answer_output, logs_output])
 
 demo.launch()