Update app.py

app.py

@@ -1,59 +1,50 @@
 import gradio as gr
-import time
-from rag_pipeline import RAGPipeline
 from utils import process_documents
+from rag_pipeline import RAGPipeline
 
 rag = RAGPipeline()
 
 log_messages = []
 
-def log(message):
+def log_callback(message):
     log_messages.append(message)
-    return "\n".join(log_messages)
-
-def process_files(files):
-    global log_messages
-    log_messages = []
-    log("🚀 [RAG] جاري تحميل الملفات...")
-    start_time = time.time()
-    all_chunks = process_documents(files, log)
-    log(f"📚 [RAG] تم استخراج {len(all_chunks)} مقاطع نصية.")
-    rag.build_index(all_chunks, log)
-    end_time = time.time()
-    log(f"✅ [RAG] تم بناء الفهرس في {end_time - start_time:.2f} ثانية.")
-    return "\n".join(log_messages)
+    print(message)
+
+def upload_and_index(files):
+    log_messages.clear()
+    if not files:
+        return "⚠️ يرجى رفع ملفات.", "", ""
+
+    chunks = []
+    for file in files:
+        content_chunks = process_documents(file.name)
+        chunks.extend(content_chunks)
+
+    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(f"❓ [RAG] السؤال: {question}")
-    start_time = time.time()
-    answer, passages = rag.answer_question(question, log)
-    end_time = time.time()
-    log(f"🕒 تم توليد الإجابة في {end_time - start_time:.2f} ثانية.")
-    log(f"✅ الإجابة: {answer}")
-    sources = "\n\n".join([f"📄 من الفقرة:\n---\n{p}" for p in passages])
-    return answer, sources, "\n".join(log_messages)
-
-with gr.Blocks() as demo:
-    gr.Markdown("# 🤖 مساعد بحث بالذكاء الاصطناعي للكتب العربية")
+        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"], file_count="multiple", label="📂 رفع الكتب")
-        upload_btn = gr.Button("🔄 بناء الفهرس")
-    
-    log_output = gr.Textbox(label="🧭 سجل التشغيل", lines=15)
-
-    upload_btn.click(fn=process_files, inputs=[file_input], outputs=[log_output])
+        file_input = gr.File(file_types=[".pdf", ".docx", ".txt"], label="📁 حمّل ملفاتك", file_count="multiple")
+        upload_button = gr.Button("🔍 رفع وبناء الفهرس")
+    upload_output = gr.Textbox(label="🔔 الحالة", lines=2)
 
     with gr.Row():
-        question_input = gr.Textbox(label="📝 اطرح سؤالك هنا")
-        ask_btn = gr.Button("📤 إرسال السؤال")
+        question_input = gr.Textbox(placeholder="اكتب سؤالك هنا...", label="❓ السؤال")
+        submit_button = gr.Button("🚀 أرسل السؤال")
 
-    answer_output = gr.Textbox(label="📢 الإجابة", lines=5)
-    sources_output = gr.Textbox(label="📌 مصدر الإجابة", lines=10)
-    log_box = gr.Textbox(label="🔍 سجل التشغيل الكامل", lines=10)
-
-    ask_btn.click(fn=answer_question, inputs=[question_input], outputs=[answer_output, sources_output, log_box])
+    answer_output = gr.Textbox(label="🧠 الإجابة", lines=6)
+    logs_output = gr.Textbox(label="📜 سجل العمليات", lines=12)
 
+    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()