Update app.py

app.py

@@ -1,7 +1,6 @@
 import gradio as gr
 from rag_pipeline import RAGPipeline
 from utils import process_documents
-import time
 
 rag = RAGPipeline()
 
@@ -18,37 +17,40 @@ def upload_and_index(files, logs):
         all_chunks.extend(chunks)
         logs = log_message(f"[RAG] تم استخراج {len(chunks)} مقطع من {file.name}", logs)
 
-    logs = log_message(f"[RAG] بناء الفهرس لـ {len(all_chunks)} مقطع...", logs)
-    start = time.time()
     rag.build_index(all_chunks)
-    duration = time.time() - start
-    logs = log_message(f"[RAG] تم بناء الفهرس في {duration:.2f} ثانية.", logs)
-    return logs, gr.update(visible=True), gr.update(visible=True)
+    logs = log_message("[RAG] تم بناء الفهرس.")
+    return logs, gr.update(visible=True), gr.update(visible=True), gr.update(visible=True)
+
+def summarize_chunks(logs):
+    logs = log_message("[RAG] بدأ التلخيص للمقاطع...")
+    rag.summarize_all_chunks()
+    logs = log_message("[RAG] تم التلخيص لجميع المقاطع.")
+    return logs
 
 def answer_question(question, logs):
     logs = log_message(f"[RAG] استلام السؤال: {question}", logs)
-    start = time.time()
     answer, sources, summary = rag.answer(question)
-    duration = time.time() - start
-    logs = log_message(f"[RAG] تم الإجابة في {duration:.2f} ثانية.", logs)
-    logs = log_message(f"[RAG] المصادر: {sources}", logs)
+    logs = log_message(f"[RAG] تم توليد الإجابة.\n[RAG] المصادر: {sources}")
     return answer, summary, logs
 
 with gr.Blocks() as demo:
     logs = gr.State("")
-    gr.Markdown("# 🕌 نظام استرجاع المعرفة باللغة العربية مع التلخيص والإجابة")
+    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("🔄 رفع وبناء الفهرس")
+
+    summarize_btn = gr.Button("🧠 تلخيص المقاطع", visible=False)
+    question_input = gr.Textbox(label="❓ اكتب سؤالك هنا", visible=False)
+    answer_btn = gr.Button("✍️ أجب عن السؤال", visible=False)
 
-    logs_output = gr.Textbox(label="سجل العمليات", lines=12, interactive=False, value="")
-    question_input = gr.Textbox(label="اكتب سؤالك هنا", visible=False)
-    ask_btn = gr.Button("إرسال السؤال", visible=False)
-    answer_output = gr.Textbox(label="الإجابة", lines=5)
-    summary_output = gr.Textbox(label="الملخص", lines=5)
+    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, question_input, ask_btn])
-    ask_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, 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], outputs=[answer_output, summary_output, logs_output])
 
 demo.launch()