Update app.py

app.py

@@ -1,6 +1,3 @@
-# ✅ Arabic RAG System with Cascaded Memory Pipeline
-# Step-by-step: Chunking -> Question -> Retrieval -> Summarization
-
 import gradio as gr
 from rag_pipeline import RAGPipeline
 from utils import process_documents
@@ -29,11 +26,12 @@ def store_question(q):
 
 def retrieve_passages(question, chunks, logs):
     passages = rag.retrieve_passages(question)
+    passages_text = "\n---\n".join(passages)
     logs = log_message(f"[RAG] تم العثور على {len(passages)} مقطع مرتبط بالسؤال.", logs)
-    return passages, logs
+    return passages, passages_text, logs
 
-def summarize_answer(question, passages, logs):
-    answer, summary = rag.generate_answer_from_passages(question, passages)
+def summarize_answer(question, passages_text, logs):
+    answer, summary = rag.generate_answer_from_passages(question, passages_text)
     logs = log_message("[RAG] تم توليد الإجابة النهائية.", logs)
     return answer, summary, logs
 
@@ -56,13 +54,14 @@ with gr.Blocks() as demo:
     find_btn = gr.Button("🔍 بحث عن المقاطع المرتبطة")
     answer_btn = gr.Button("✍️ تلخيص وتوليد الإجابة")
 
-    answer_output = gr.Textbox(label="✅ الإجابة", lines=5)
+    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])
     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, logs_output])
-    answer_btn.click(summarize_answer, inputs=[question_state, retrieved_passages, logs], outputs=[answer_output, summary_output, logs_output])
+    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=[answer_output, summary_output, logs_output])
 
 demo.launch()