Update app.py

app.py

@@ -1,50 +1,57 @@
 import gradio as gr
-from utils import process_documents
 from rag_pipeline import RAGPipeline
+from utils import process_documents
+import asyncio
+import time
 
 rag = RAGPipeline()
 
-log_messages = []
-
-def log_callback(message):
-    log_messages.append(message)
-    print(message)
-
-def upload_and_index(files):
-    log_messages.clear()
-    if not files:
-        return "⚠️ يرجى رفع ملفات.", "", ""
+def log_message(msg, logs):
+    logs.append(msg)
+    return logs
 
-    chunks = []
+def upload_and_index(files, logs):
+    logs = log_message("[RAG] بدء معالجة الملفات...", logs)
+    all_chunks = []
     for file in files:
-        content_chunks = process_documents(file.name)
-        chunks.extend(content_chunks)
+        logs = log_message(f"[RAG] معالجة الملف: {file.name}", logs)
+        chunks = process_documents(file.name)
+        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, logs)
+    duration = time.time() - start
+    logs = log_message(f"[RAG] تم بناء الفهرس في {duration:.2f} ثانية.", logs)
+    return logs, True  # True لتمكين مربع السؤال
+
+def answer_question(question, logs):
+    logs = log_message(f"[RAG] استلام السؤال: {question}", logs)
+    start = time.time()
+    answer, sources = rag.answer(question)
+    duration = time.time() - start
+    logs = log_message(f"[RAG] تم الإجابة في {duration:.2f} ثانية.", logs)
+    logs = log_message(f"[RAG] المصادر: {sources}", logs)
+    return answer, logs
+
+with gr.Blocks() as demo:
+    logs = gr.State([])
+    gr.Markdown("# نظام استرجاع المعرفة من الملفات (RAG)")
 
-    index_status = rag.build_index(chunks, log_callback=log_callback)
-    return index_status, "\n".join(log_messages), ""
-
-def answer_question(question):
-    if not question.strip():
-        return "⚠️ يرجى إدخال سؤال."
-    log_messages.clear()
-    answer = rag.answer_question(question, log_callback=log_callback)
-    return answer
-
-with gr.Blocks(title="مساعد عربي للكتب") as demo:
-    gr.Markdown("## 📚 مساعد ذكي للإجابة من كتبك باللغة العربية")
     with gr.Row():
-        file_input = gr.File(file_types=[".pdf", ".docx", ".txt"], label="📁 حمّل ملفاتك", file_count="multiple")
-        upload_button = gr.Button("🔍 رفع وبناء الفهرس")
-    upload_output = gr.Textbox(label="🔔 الحالة", lines=2)
+        files_input = gr.File(file_types=['.pdf', '.docx', '.txt'], file_count="multiple", label="رفع الملفات")
+        upload_btn = gr.Button("رفع وبناء الفهرس")
+    logs_output = gr.Textbox(label="سجل العمليات", lines=10, interactive=False)
 
-    with gr.Row():
-        question_input = gr.Textbox(placeholder="اكتب سؤالك هنا...", label="❓ السؤال")
-        submit_button = gr.Button("🚀 أرسل السؤال")
+    question_input = gr.Textbox(label="اكتب سؤالك هنا", visible=False)
+    ask_btn = gr.Button("إرسال السؤال", visible=False)
+    answer_output = gr.Textbox(label="الإجابة", lines=5)
+
+    upload_btn.click(upload_and_index, inputs=[files_input, logs], outputs=[logs_output, question_input])
+    upload_btn.click(lambda: True, None, question_input)  # اظهار مربع السؤال
+    upload_btn.click(lambda: True, None, ask_btn)         # اظهار زر السؤال
 
-    answer_output = gr.Textbox(label="🧠 الإجابة", lines=6)
-    logs_output = gr.Textbox(label="📜 سجل العمليات", lines=12)
+    ask_btn.click(answer_question, inputs=[question_input, logs], outputs=[answer_output, logs_output])
 
-    upload_button.click(fn=upload_and_index, inputs=[file_input], outputs=[upload_output, logs_output, answer_output])
-    submit_button.click(fn=answer_question, inputs=[question_input], outputs=[answer_output])
-    
 demo.launch()