Update app.py

app.py

@@ -431,6 +431,13 @@ body { background: linear-gradient(to bottom right, #2c3e50, #34495e); color: #e
 """
 
 with gr.Blocks(title="AI Research Agent v5.8") as demo:
+    gr.Markdown("# 🤖 AI Research Agent") 
+    with gr.Row():
+        with gr.Column(scale=3): agent_stat_tb = gr.Textbox(label="Agent Status", interactive=False, lines=1, value="Initializing systems...")
+        with gr.Column(scale=1): memory_backend_info_tb = gr.Textbox(label="Memory Backend", value=f"{MEMORY_STORAGE_BACKEND}", interactive=False)
+        sqlite_path_display = gr.Textbox(label="SQLite Path", value=f"{MEMORY_SQLITE_PATH}", interactive=False, scale=2) # Use visible
+        hf_repos_display = gr.Textbox(label="HF Repos", value=f"M: {MEMORY_HF_MEM_REPO}, R: {MEMORY_HF_RULES_REPO}", interactive=False, scale=2) # Use visible
+
     avail_provs = get_available_providers()
     def_prov = avail_provs[0] if avail_provs else None
     def_models = get_model_display_names_for_provider(def_prov) if def_prov else []
@@ -447,46 +454,38 @@ with gr.Blocks(title="AI Research Agent v5.8") as demo:
             sys_prompt_tb = gr.Textbox(label="System Prompt Base", lines=10, value=DEFAULT_SYSTEM_PROMPT, interactive=True)
         if MEMORY_STORAGE_BACKEND == "RAM":
             save_faiss_sidebar_btn = gr.Button("Save FAISS Indices", variant="secondary")
-    with gr.Group():
-        with gr.Row():
-            gr.Markdown("# 🤖 AI Research Agent") 
-        with gr.Row():
-            with gr.Column(scale=3): agent_stat_tb = gr.Textbox(label="Agent Status", interactive=False, lines=1, value="Initializing systems...")
-            with gr.Column(scale=1): memory_backend_info_tb = gr.Textbox(label="Memory Backend", value=f"{MEMORY_STORAGE_BACKEND}", interactive=False)
-            sqlite_path_display = gr.Textbox(label="SQLite Path", value=f"{MEMORY_SQLITE_PATH}", interactive=False, scale=2) # Use visible
-            hf_repos_display = gr.Textbox(label="HF Repos", value=f"M: {MEMORY_HF_MEM_REPO}, R: {MEMORY_HF_RULES_REPO}", interactive=False, scale=2) # Use visible
-    
-        with gr.Tabs():
-            with gr.TabItem("💬 AI Chat & Research Output"):
-                gr.Markdown("### AI Chat Interface") # Using H3 for section titles within tabs
-                main_chat_disp = gr.Chatbot(label="AI Research Chat", height=500, bubble_full_width=False, avatar_images=(None, "https://raw.githubusercontent.com/huggingface/brand-assets/main/hf-logo-with-title.png"), show_copy_button=True, render_markdown=True, sanitize_html=True)
-                with gr.Row():
-                    user_msg_tb = gr.Textbox(show_label=False, placeholder="Ask your research question...", scale=7, lines=1, max_lines=5, autofocus=True)
-                    send_btn = gr.Button("Send", variant="primary", scale=1, min_width=100)
-                with gr.Accordion("📝 Full Response / Details", open=False):
-                    fmt_report_tb = gr.Textbox(label="Full AI Response", lines=10, interactive=True, show_copy_button=True)
-                    dl_report_btn = gr.DownloadButton("Download Report", interactive=False, visible=False) 
-                detect_out_md = gr.Markdown()
-    
-            with gr.TabItem("🧠 Knowledge Base Management"):
-                gr.Markdown("### Manage Stored Knowledge")
-                with gr.Row():
-                    with gr.Column():
-                        with gr.Group():
-                            gr.Markdown("#### 📜 Rules (Insights)")
-                            rules_disp_ta = gr.TextArea(label="View/Edit Rules", lines=15, interactive=True, placeholder="Load rules or type new ones...")
-                            with gr.Row(): view_rules_btn = gr.Button("🔄 Load"); save_edited_rules_btn = gr.Button("💾 Save Edits", variant="primary")
-                            upload_rules_fobj = gr.File(label="Upload Rules File", file_types=[".txt", ".jsonl"])
-                            rules_stat_tb = gr.Textbox(label="Rules Status", interactive=False, lines=2)
-                            clear_rules_btn = gr.Button("⚠️ Clear All Rules", variant="stop")
-                    with gr.Column():
-                        with gr.Group():
-                            gr.Markdown("#### 📚 Memories")
-                            mems_disp_json = gr.JSON(label="View Memories") 
-                            view_mems_btn = gr.Button("🔄 Load Memories")
-                            upload_mems_fobj = gr.File(label="Upload Memories File", file_types=[".jsonl"])
-                            mems_stat_tb = gr.Textbox(label="Memories Status", interactive=False, lines=2)
-                            clear_mems_btn = gr.Button("⚠️ Clear All Memories", variant="stop")
+        
+    with gr.Tabs():
+        with gr.TabItem("💬 AI Chat & Research Output"):
+            gr.Markdown("### AI Chat Interface") # Using H3 for section titles within tabs
+            main_chat_disp = gr.Chatbot(label="AI Research Chat", height=500, bubble_full_width=False, avatar_images=(None, "https://raw.githubusercontent.com/huggingface/brand-assets/main/hf-logo-with-title.png"), show_copy_button=True, render_markdown=True, sanitize_html=True)
+            with gr.Row():
+                user_msg_tb = gr.Textbox(show_label=False, placeholder="Ask your research question...", scale=7, lines=1, max_lines=5, autofocus=True)
+                send_btn = gr.Button("Send", variant="primary", scale=1, min_width=100)
+            with gr.Accordion("📝 Full Response / Details", open=False):
+                fmt_report_tb = gr.Textbox(label="Full AI Response", lines=10, interactive=True, show_copy_button=True)
+                dl_report_btn = gr.DownloadButton("Download Report", interactive=False, visible=False) 
+            detect_out_md = gr.Markdown()
+
+        with gr.TabItem("🧠 Knowledge Base Management"):
+            gr.Markdown("### Manage Stored Knowledge")
+            with gr.Row():
+                with gr.Column():
+                    with gr.Group():
+                        gr.Markdown("#### 📜 Rules (Insights)")
+                        rules_disp_ta = gr.TextArea(label="View/Edit Rules", lines=15, interactive=True, placeholder="Load rules or type new ones...")
+                        with gr.Row(): view_rules_btn = gr.Button("🔄 Load"); save_edited_rules_btn = gr.Button("💾 Save Edits", variant="primary")
+                        upload_rules_fobj = gr.File(label="Upload Rules File", file_types=[".txt", ".jsonl"])
+                        rules_stat_tb = gr.Textbox(label="Rules Status", interactive=False, lines=2)
+                        clear_rules_btn = gr.Button("⚠️ Clear All Rules", variant="stop")
+                with gr.Column():
+                    with gr.Group():
+                        gr.Markdown("#### 📚 Memories")
+                        mems_disp_json = gr.JSON(label="View Memories") 
+                        view_mems_btn = gr.Button("🔄 Load Memories")
+                        upload_mems_fobj = gr.File(label="Upload Memories File", file_types=[".jsonl"])
+                        mems_stat_tb = gr.Textbox(label="Memories Status", interactive=False, lines=2)
+                        clear_mems_btn = gr.Button("⚠️ Clear All Memories", variant="stop")
         
     def dyn_upd_model_dd(sel_prov_dyn:str): models_dyn, def_model_dyn = get_model_display_names_for_provider(sel_prov_dyn), get_default_model_display_name_for_provider(sel_prov_dyn); return gr.Dropdown(choices=models_dyn, value=def_model_dyn, interactive=True)
     prov_sel_dd.change(fn=dyn_upd_model_dd, inputs=prov_sel_dd, outputs=model_sel_dd)