Update app.py

app.py

@@ -402,37 +402,49 @@ def update_history(q, ans, src, hist):
 
 
 # ================= UI =================
-with gr.Row():
-    # ---------- LEFT (History) ----------
-    with gr.Column(scale=1):
-        gr.Markdown("## 📝 السجل")
-        history_md = gr.Markdown("لا يوجد سجل بعد")
-        history_index = gr.Number(label="اختر رقم السؤال", precision=0, value=1)
+with gr.Blocks(title="⚖️ محامي افتراضي مغربي") as demo:
+    gr.Markdown("# ⚖️ محامي افتراضي\n### القانون المغربي بين يديك")
 
-    # ---------- MAIN ----------
-    with gr.Column(scale=3):
-        query = gr.Textbox(
-            placeholder="اكتب هنا / Écrivez ici",
-            label="❓ سؤالك القانوني",
-            lines=2
-        )
+    history_state = gr.State([])
+    retrieved_state = gr.State()
+    prompt_state = gr.State()
 
-        ask = gr.Button("🔍 اسأل", variant="primary")
+    with gr.Row():
+        # ---------- LEFT (History) ----------
+        with gr.Column(scale=1):
+            gr.Markdown("## 📝 السجل")
+            history_md = gr.Markdown("لا يوجد سجل بعد")
+            history_index = gr.Number(
+                label="اختر رقم السؤال",
+                precision=0,
+                value=1
+            )
 
-        loading_md = gr.Markdown("", visible=False)
+        # ---------- MAIN ----------
+        with gr.Column(scale=3):
+            query = gr.Textbox(
+                placeholder="اكتب هنا / Écrivez ici",
+                label="❓ سؤالك القانوني",
+                lines=2
+            )
 
-        answer = gr.Markdown(label="📜 الجواب")
+            ask = gr.Button("🔍 اسأل", variant="primary")
 
-        with gr.Accordion("📚 المصادر المستخدمة", open=False):
-            sources = gr.Markdown("—")
+            loading_md = gr.Markdown("🔄 **جارٍ البحث والتوليد…**\n\n⏳", visible=False)
 
-        stats = gr.Markdown("—")
+            answer = gr.Markdown(label="📜 الجواب")
+
+            with gr.Accordion("📚 المصادر المستخدمة", open=False):
+                sources = gr.Markdown("—")
+
+            stats = gr.Markdown("—")
+
+        # ---------- SETTINGS ----------
+        with gr.Column(scale=1):
+            gr.Markdown("## ⚙️ الإعدادات")
+            top_k = gr.Slider(1, 8, value=3, step=1, label="عدد المقتطفات")
+            temperature = gr.Slider(0, 1, value=0.0, step=0.05, label="العشوائية")
 
-    # ---------- SETTINGS ----------
-    with gr.Column(scale=1):
-        gr.Markdown("## ⚙️ الإعدادات")
-        top_k = gr.Slider(1, 8, value=3, step=1, label="عدد المقتطفات")
-        temperature = gr.Slider(0, 1, value=0.0, step=0.05, label="العشوائية")
 
     # ---------- EVENTS ----------
     ask.click(