Update app.py

app.py

@@ -44,8 +44,10 @@ logger.info(f"App Config: WebSearch={WEB_SEARCH_ENABLED}, ToolDecisionProvider={
 
 # --- Helper Functions (format_insights_for_prompt, generate_interaction_metrics, etc.) ---
 # These functions (format_insights_for_prompt, generate_interaction_metrics, 
-# process_user_interaction_gradio, deferred_learning_and_memory_task) remain the same as in the previous "full working file".
-# For brevity here, I will not repeat them. Ensure they are present in your actual app.py.
+# process_user_interaction_gradio, deferred_learning_and_memory_task, handle_gradio_chat_submit,
+# ui_view_rules_action_fn, ui_upload_rules_action_fn, ui_view_memories_action_fn, ui_upload_memories_action_fn) 
+# remain the same as in the previous "full working file".
+# For brevity here, I will not repeat them fully. Ensure they are present in your actual app.py.
 def format_insights_for_prompt(retrieved_insights_list: list[str]) -> tuple[str, list[dict]]:
     if not retrieved_insights_list:
         return "No specific guiding principles or learned insights retrieved.", []
@@ -190,17 +192,14 @@ def deferred_learning_and_memory_task(user_input: str, bot_response: str, provid
         summary = f"User:\"{user_input}\"\nAI:\"{bot_response}\"\nMetrics(takeaway):{metrics.get('takeaway','N/A')},Success:{metrics.get('response_success_score','N/A')}"
         existing_rules_ctx = "\n".join([f"- \"{r}\"" for r in retrieve_rules_semantic(f"{summary}\n{user_input}", k=10)]) or "No existing rules context."
         insight_sys_prompt = """You are an expert AI knowledge base curator. Your primary function is to meticulously analyze an interaction and update the AI's guiding principles (insights/rules) to improve its future performance and self-understanding.
-
 **CRITICAL OUTPUT REQUIREMENT: You MUST output a single, valid JSON list of operation objects.**
 This list can and SHOULD contain MULTIPLE distinct operations if various learnings occurred.
 If no operations are warranted, output an empty JSON list: `[]`.
 ABSOLUTELY NO other text, explanations, or markdown should precede or follow this JSON list.
-
 Each operation object in the JSON list must have these keys and string values:
 1.  `"action"`: A string, either `"add"` (for entirely new rules) or `"update"` (to replace an existing rule with a better one).
 2.  `"insight"`: A string, the full, refined insight text including its `[TYPE|SCORE]` prefix (e.g., `"[CORE_RULE|1.0] My name is Lumina, an AI assistant."`).
 3.  `"old_insight_to_replace"`: (ONLY for `"update"` action) A string, the *exact, full text* of an existing insight that the new `"insight"` should replace. If action is `"add"`, this key should be omitted or its value should be `null` or an empty string.
-
 **CRITICAL JSON STRING FORMATTING RULES (for values of "insight" and "old_insight_to_replace"):**
 -   All string values MUST be enclosed in double quotes (`"`).
 -   Any literal double quote (`"`) character *within* the string content MUST be escaped as `\\"`.
@@ -208,7 +207,6 @@ Each operation object in the JSON list must have these keys and string values:
 -   Any newline characters *within* the string content MUST be escaped as `\\n`. Avoid literal newlines in JSON string values; use `\\n` instead.
     *Example of correctly escaped insight string in JSON:*
     `"insight": "[RESPONSE_PRINCIPLE|0.8] User prefers concise answers, stating: \\"Just the facts!\\". Avoid verbose explanations unless asked.\\nFollow up with a question if appropriate."`
-
 **Your Reflection Process (Consider each step and generate operations accordingly):**
 **STEP 1: Core Identity & Purpose Review (Result: Primarily 'update' operations)**
    - Examine all `CORE_RULE`s related to my identity (name, fundamental purpose, core unchanging capabilities, origin) from the "Potentially Relevant Existing Rules".
@@ -224,12 +222,10 @@ Each operation object in the JSON list must have these keys and string values:
    - Did I learn to modify or improve an existing behavior, response style, or operational guideline (that is NOT part of core identity)?
    - For example, if an existing `RESPONSE_PRINCIPLE` was "Be formal," and the interaction showed the user prefers informality, update that principle.
    - Propose 'update' operations for the relevant `RESPONSE_PRINCIPLE` or `BEHAVIORAL_ADJUSTMENT`. Only update if the change is significant.
-
 **General Guidelines for Insight Content and Actions:**
 - Ensure the "insight" field (for both add/update) always contains the properly formatted insight string: `[TYPE|SCORE] Text`. `TYPE` can be `CORE_RULE`, `RESPONSE_PRINCIPLE`, `BEHAVIORAL_ADJUSTMENT`. Scores should reflect confidence/importance (0.0-1.0).
 - Be precise with "old_insight_to_replace" – it must *exactly* match an existing rule string from the "Potentially Relevant Existing Rules" context.
 - Aim for a comprehensive set of operations that reflects ALL key learnings from the interaction.
-
 **Example of a comprehensive JSON output with MULTIPLE operations (This is how your output should look):**
 [
   {"action": "update", "old_insight_to_replace": "[CORE_RULE|1.0] My designated name is 'LearnerAI'.", "insight": "[CORE_RULE|1.0] I am Lumina, an AI assistant designed to chat, provide information, and remember context like the secret word 'rocksyrup'."},
@@ -261,7 +257,7 @@ Combine all findings into a single JSON list of operations. If there are multipl
         json_match_ops = re.search(r"```json\s*(\[.*?\])\s*```", raw_ops_json, re.DOTALL|re.I) or re.search(r"(\[.*?\])", raw_ops_json, re.DOTALL)
         if json_match_ops:
             try: ops = json.loads(json_match_ops.group(1))
-            except Exception as e: logger.error(f"DEFERRED [{task_id}]: JSON ops parse error: {e}. Raw: {json_match_ops.group(1)[:500]}") # Log problematic part
+            except Exception as e: logger.error(f"DEFERRED [{task_id}]: JSON ops parse error: {e}. Raw: {json_match_ops.group(1)[:500]}")
         if isinstance(ops, list) and ops:
             logger.info(f"DEFERRED [{task_id}]: LLM provided {len(ops)} insight ops.")
             for op in ops:
@@ -399,238 +395,118 @@ def ui_upload_memories_action_fn(uploaded_file_obj, progress=gr.Progress()):
     logger.info(msg); return msg
 
 # --- UI Definition ---
-custom_theme = gr.themes.Soft(
+custom_theme = gr.themes.Base(
     primary_hue="teal",
     secondary_hue="purple",
     neutral_hue="zinc",
-    text_size="md", 
-    spacing_size="md",
+    text_size="sm", 
+    spacing_size="md", 
     radius_size="sm",
-    font=[gr.themes.GoogleFont("Inter"), "System UI", "sans-serif"] # Added Inter as primary
+    font=[gr.themes.GoogleFont("Inter"), "System UI", "sans-serif"]
 )
 
+# Adjusted CSS for better layout control and scrolling
 custom_css = """
-body, html {
-    margin: 0 !important;
-    padding: 0.5em !important;
-    height: 100%;
-    background: linear-gradient(to bottom right, #2c3e50, #34495e);
-    color: #ecf0f1;
-    font-family: 'Inter', 'System UI', sans-serif;
-}
-.gradio-container {
-    background: transparent !important;
-    max-width: 100% !important;
-    padding: 2em 1em 1em 1em !important;
-    box-sizing: border-box;
-    min-height: 100vh;
-    display: flex;
-    flex-direction: column;
-    overflow-y: auto !important;
-}
-.gr-block.gr-group, .gr-tabs, .gr-accordion {
-    background-color: rgba(44, 62, 80, 0.85) !important;
-    border: 1px solid rgba(189, 195, 199, 0.2) !important;
-    border-radius: 8px !important;
-    box-shadow: 0 3px 5px rgba(0,0,0,0.15);
-    margin-bottom: 1em;
-}
-.gr-tabitem {
-    background-color: rgba(52, 73, 94, 0.8) !important;
-    border-radius: 6px !important;
-    padding: 1em !important;
-    border: 1px solid rgba(189, 195, 199, 0.1) !important;
-}
-.gr-textbox, .gr-dropdown, .gr-button, .gr-code, .gr-chat-message, .gr-json, .gr-file input[type="file"], .gr-file .OgniCn {
-    border: 1px solid rgba(189, 195, 199, 0.3) !important;
-    background-color: rgba(52, 73, 94, 0.9) !important;
-    color: #ecf0f1 !important;
-    border-radius: 6px !important;
-}
-.gr-file {
-    background-color: transparent !important;
-    border: none !important;
-    padding: 0 !important;
-}
-.gr-file > .label-text {
-    color: #bdc3c7 !important;
-}
-.gr-textarea textarea, .gr-textbox input {
-    color: #ecf0f1 !important;
-}
-.gr-button.gr-button-primary {
-    background-color: #1abc9c !important;
-    color: white !important;
-    border-color: #16a085 !important;
-}
-.gr-button.gr-button-secondary {
-    background-color: #9b59b6 !important;
-    color: white !important;
-    border-color: #8e44ad !important;
-}
-.gr-button.gr-button-stop {
-    background-color: #e74c3c !important;
-    color: white !important;
-    border-color: #c0392b !important;
-}
-.gr-markdown {
-    padding: 0px;
-}
-.gr-markdown h1, .gr-markdown h2, .gr-markdown h3 {
-    color: #1abc9c !important;
-    border-bottom: 1px solid rgba(189, 195, 199, 0.2) !important;
-    padding-bottom: 0.3em;
-    margin-top: 0.8em;
-    margin-bottom: 0.6em;
-}
-.gr-markdown p, .gr-markdown li {
-    color: #ecf0f1 !important;
-    line-height: 1.6;
-}
-.gr-markdown pre code {
-    background-color: rgba(30, 40, 50, 0.95) !important;
-    border: 1px solid rgba(189, 195, 199, 0.3) !important;
-    color: #ecf0f1;
-    border-radius: 4px;
-    padding: 0.8em;
-}
-.gr-chatbot {
-    background-color: rgba(44, 62, 80, 0.7) !important;
-    border: 1px solid rgba(189, 195, 199, 0.2) !important;
-}
-.gr-chatbot .message {
-    background-color: rgba(52, 73, 94, 0.9) !important;
-    color: #ecf0f1 !important;
-    border: 1px solid rgba(189, 195, 199, 0.2) !important;
-    box-shadow: 0 1px 2px rgba(0,0,0,0.1) !important;
-}
-.gr-chatbot .message.user {
-    background-color: rgba(46, 204, 113, 0.8) !important;
-    color: #2c3e50 !important;
-}
-.gr-input-label > .label-text, .gr-dropdown-label > .label-text, .gr-checkbox-label > .label-text {
-    color: #bdc3c7 !important;
-    font-size: 0.9rem !important;
-    margin-bottom: 0.2rem !important;
-}
-.gr-info {
-    color: #bdc3c7 !important;
-    font-size: 0.85rem !important;
-}
-.status-bar {
-    background-color: rgba(44, 62, 80, 0.8) !important;
-    padding: 0.5rem 1rem !important;
-    border-radius: 6px;
-    margin: 1rem 0 !important;
-}
-.tabnav button {
-    background-color: rgba(52, 73, 94, 0.8) !important;
-    color: #ecf0f1 !important;
-    border-bottom: 2px solid transparent !important;
-    border-top-left-radius: 6px !important;
-    border-top-right-radius: 6px !important;
-    margin-right: 2px !important;
-    padding: 0.5em 1em !important;
-}
-.tabnav button.selected {
-    background-color: rgba(44, 62, 80, 0.95) !important;
-    color: #1abc9c !important;
-    border-bottom: 2px solid #1abc9c !important;
-}
-.main-title {
-    text-align: center;
-    margin: 1.5rem 0 0.5rem 0 !important;
-}
-.main-subtitle {
-    text-align: center;
-    color: #bdc3c7;
-    font-size: 0.95rem;
-    margin: 0 0 1.5rem 0 !important;
-}
-@media (max-width: 768px) {
-    .gradio-container {
-        padding: 1.5em 0.5em !important;
-    }
-    .main-title {
-        font-size: 1.5rem !important;
-        margin-top: 1rem !important;
-    }
-    .main-subtitle {
-        font-size: 0.85rem !important;
-    }
-    .status-bar {
-        margin: 0.5rem 0 !important;
-    }
-}
+body, html { margin: 0; padding: 0; height: 100%; overflow-x: hidden; /* Prevent horizontal scroll on body */ }
+body { background: linear-gradient(to bottom right, #2c3e50, #34495e); color: #ecf0f1; font-family: 'Inter', 'System UI', sans-serif;}
+.gradio-container { background: transparent !important; max-width: 100% !important; padding: 0 !important; /* Remove container padding */ box-sizing: border-box; min-height: 100vh; display: flex; flex-direction: column; overflow: hidden !important; /* Gradio container should not scroll itself */ }
+.main-app-content-wrapper { padding: 1rem; overflow-y: auto; flex-grow: 1; /* This makes the content area scrollable */ }
+.gr-block.gr-group, .gr-tabs, .gr-accordion { background-color: rgba(44, 62, 80, 0.85) !important; border: 1px solid rgba(189, 195, 199, 0.2) !important; border-radius: 8px !important; box-shadow: 0 3px 5px rgba(0,0,0,0.15); margin-bottom: 1em; padding: 1em; }
+.gr-tabitem { background-color: rgba(52, 73, 94, 0.8) !important; border-radius: 6px !important; padding: 1em !important; border: 1px solid rgba(189, 195, 199, 0.1) !important;}
+.gr-textbox, .gr-dropdown, .gr-button, .gr-code, .gr-chat-message, .gr-json, .gr-file input[type="file"], .gr-file button { border: 1px solid rgba(189, 195, 199, 0.3) !important; background-color: rgba(52, 73, 94, 0.9) !important; color: #ecf0f1 !important; border-radius: 6px !important;}
+.gr-file { background-color: transparent !important; border: none !important; padding:0 !important;}
+.gr-file > .label-text { color: #bdc3c7 !important;}
+.gr-textarea textarea, .gr-textbox input { color: #ecf0f1 !important; background-color: transparent !important; }
+.gr-button.gr-button-primary { background-color: #1abc9c !important; color: white !important; border-color: #16a085 !important; }
+.gr-button.gr-button-secondary { background-color: #9b59b6 !important; color: white !important; border-color: #8e44ad !important; }
+.gr-button.gr-button-stop { background-color: #e74c3c !important; color: white !important; border-color: #c0392b !important; }
+.gr-markdown { padding: 5px; } 
+.gr-markdown h1, .gr-markdown h2, .gr-markdown h3 { color: #1abc9c !important; border-bottom: 1px solid rgba(189, 195, 199, 0.2) !important; padding-bottom: 0.3em; margin-top:0.5em; margin-bottom: 0.5em; }
+.gr-markdown p, .gr-markdown li { color: #ecf0f1 !important; line-height:1.6;}
+.gr-markdown pre code { background-color: rgba(30, 40, 50, 0.95) !important; border: 1px solid rgba(189, 195, 199, 0.3) !important; color: #ecf0f1; border-radius: 4px; padding: 0.8em; }
+.gr-chatbot { background-color: rgba(44, 62, 80, 0.7) !important; border: 1px solid rgba(189, 195, 199, 0.2) !important; }
+.gr-chatbot .message { background-color: rgba(52, 73, 94, 0.9) !important; color: #ecf0f1 !important; border: 1px solid rgba(189, 195, 199, 0.2) !important; box-shadow: 0 1px 2px rgba(0,0,0,0.1) !important; }
+.gr-chatbot .message.user { background-color: rgba(46, 204, 113, 0.8) !important; color: #2c3e50 !important; }
+.gr-input-label > .label-text, .gr-dropdown-label > .label-text, .gr-checkbox-label > .label-text { color: #bdc3c7 !important; font-size: 0.9rem !important; margin-bottom: 0.2rem !important; }
+.gr-info { color: #bdc3c7 !important; font-size: 0.85rem !important; }
+.status-bar { background-color: rgba(44, 62, 80, 0.8) !important; padding: 0.75rem 1rem !important; border-radius: 8px; margin-bottom: 1rem; }
+.tabnav button { background-color: rgba(52, 73, 94, 0.8) !important; color: #ecf0f1 !important; border-bottom: 3px solid transparent !important; border-top-left-radius: 6px !important; border-top-right-radius: 6px !important; margin-right: 3px !important; padding: 0.6em 1.2em !important; font-weight: 500;}
+.tabnav button.selected { background-color: rgba(44, 62, 80, 0.95) !important; color: #1abc9c !important; border-bottom: 3px solid #1abc9c !important;}
+.main-title { text-align: center; margin: 0 0 0.25rem 0 !important; padding-top: 0.5rem !important; color: #ecf0f1 !important; font-size: 2rem !important;} /* Added padding-top */
+.main-subtitle { text-align: center; color: #bdc3c7; font-size: 1rem; margin: 0 0 1rem 0 !important; }
+.sidebar-column { max-height: calc(100vh - 10rem); overflow-y: auto; /* Make sidebar scrollable if content exceeds height */ }
+.main-content-column { max-height: calc(100vh - 10rem); overflow-y: auto; /* Make main content scrollable */ }
 """
 
-with gr.Blocks(theme=custom_theme, css=custom_css, title="AI Research Agent") as demo:
-    gr.HTML("<style>" + custom_css + "</style>") # Ensure CSS is applied
+with gr.Blocks(theme=custom_theme, css=custom_css, title="AI Research Agent v5.6") as demo:
+    gr.HTML("<style>" + custom_css + "</style>")
 
-    gr.Markdown("# 🤖 AI Research Agent", elem_classes="main-title")
-    gr.Markdown("Configure AI, chat for research, and manage its knowledge base.", elem_classes="main-subtitle")
-    
-    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 []
-    def_model_disp = get_default_model_display_name_for_provider(def_prov) if def_prov else None
-    
-    with gr.Row(elem_classes="status-bar"):
-        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)
-        if MEMORY_STORAGE_BACKEND == "SQLITE":
-            with gr.Column(scale=2): gr.Textbox(label="SQLite Path", value=f"{MEMORY_SQLITE_PATH}", interactive=False)
-        elif MEMORY_STORAGE_BACKEND == "HF_DATASET":
-            with gr.Column(scale=2): gr.Textbox(label="HF Repos", value=f"M: {MEMORY_HF_MEM_REPO}, R: {MEMORY_HF_RULES_REPO}", interactive=False)
+    with gr.Column(elem_classes="main-app-content-wrapper"): # This outer column will handle overall page scroll
+        gr.Markdown("# 🤖 AI Research Agent", elem_classes="main-title")
+        gr.Markdown("Configure AI, chat for research, and manage its knowledge base.", elem_classes="main-subtitle")
+        
+        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 []
+        def_model_disp = get_default_model_display_name_for_provider(def_prov) if def_prov else None
+        
+        with gr.Row(elem_classes="status-bar"):
+            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)
+            if MEMORY_STORAGE_BACKEND == "SQLITE":
+                with gr.Column(scale=2): gr.Textbox(label="SQLite Path", value=f"{MEMORY_SQLITE_PATH}", interactive=False)
+            elif MEMORY_STORAGE_BACKEND == "HF_DATASET":
+                with gr.Column(scale=2): gr.Textbox(label="HF Repos", value=f"M: {MEMORY_HF_MEM_REPO}, R: {MEMORY_HF_RULES_REPO}", interactive=False)
 
-    with gr.Row(equal_height=False):
-        with gr.Column(scale=1, min_width=360): # Sidebar for configuration
-            gr.Markdown("## ⚙️ Configuration")
-            with gr.Group():
-                gr.Markdown("### AI Model Settings")
-                api_key_tb = gr.Textbox(label="AI Provider API Key (Optional Override)", type="password", placeholder="Paste key or use .env")
-                prov_sel_dd = gr.Dropdown(label="AI Provider", choices=avail_provs, value=def_prov, interactive=True)
-                model_sel_dd = gr.Dropdown(label="AI Model", choices=def_models, value=def_model_disp, interactive=True)
-            with gr.Group():
-                gr.Markdown("### System Prompt")
-                sys_prompt_tb = gr.Textbox(label="System Prompt Base", lines=12, value=DEFAULT_SYSTEM_PROMPT, interactive=True)
+        with gr.Row(equal_height=False): 
+            with gr.Column(scale=1, min_width=360, elem_classes="sidebar-column"): 
+                gr.Markdown("## ⚙️ Configuration")
+                with gr.Group():
+                    gr.Markdown("### AI Model Settings")
+                    api_key_tb = gr.Textbox(label="AI Provider API Key (Optional Override)", type="password", placeholder="Paste key or use .env")
+                    prov_sel_dd = gr.Dropdown(label="AI Provider", choices=avail_provs, value=def_prov, interactive=True)
+                    model_sel_dd = gr.Dropdown(label="AI Model", choices=def_models, value=def_model_disp, interactive=True)
+                with gr.Group():
+                    gr.Markdown("### System Prompt")
+                    sys_prompt_tb = gr.Textbox(label="System Prompt Base", lines=12, value=DEFAULT_SYSTEM_PROMPT, interactive=True)
+                gr.Markdown(f"**Memory Backend:** {MEMORY_STORAGE_BACKEND}")
+                if MEMORY_STORAGE_BACKEND == "RAM":
+                    save_faiss_sidebar_btn = gr.Button("Save FAISS Indices", variant="secondary")
 
-        with gr.Column(scale=2): # Main content area with Tabs
-            with gr.Tabs():
-                with gr.TabItem("💬 AI Chat & Research Output"):
-                    gr.Markdown("## 💬 AI Chat Interface")
-                    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 or give an instruction...", 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 / Output Details", open=True): # Open by default
-                        fmt_report_tb = gr.Textbox(label="Full AI Response", lines=10, interactive=True, show_copy_button=True, value="*AI responses will appear here...*")
-                        dl_report_btn = gr.DownloadButton(value="Download Report", interactive=False, visible=False) 
-                    detect_out_md = gr.Markdown("*Insights used or other intermediate details will show here...*")
+            with gr.Column(scale=2, elem_classes="main-content-column"):
+                with gr.Tabs():
+                    with gr.TabItem("💬 AI Chat & Research Output"):
+                        gr.Markdown("## 💬 AI Chat Interface")
+                        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 or give an instruction...", 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 / Output Details", open=True):
+                            fmt_report_tb = gr.Textbox(label="Full AI Response", lines=10, interactive=True, show_copy_button=True, value="*AI responses will appear here...*")
+                            dl_report_btn = gr.DownloadButton("Download Report", interactive=False, visible=False) 
+                        detect_out_md = gr.Markdown("*Insights used or other intermediate details will show here...*")
 
-                with gr.TabItem("🧠 Knowledge Base Management"):
-                    gr.Markdown("## Manage Stored Knowledge")
-                    with gr.Row(equal_height=False):
-                        with gr.Column(scale=1):
-                            with gr.Group():
-                                gr.Markdown("### 📜 Rules (Learned Insights)")
-                                rules_disp_ta = gr.TextArea(label="View/Edit Rules", lines=15, interactive=True, placeholder="Load or type rules here (one per line or '---' separated for multiple).")
-                                with gr.Row():
-                                    view_rules_btn = gr.Button("🔄 Load/Refresh Rules"); save_edited_rules_btn = gr.Button("💾 Save Edited Rules", variant="primary")
-                                upload_rules_fobj = gr.File(label="Upload Rules File (.txt/.jsonl)", file_types=[".txt", ".jsonl"])
-                                rules_stat_tb = gr.Textbox(label="Rules Operation Status", interactive=False, lines=2, placeholder="Status...")
-                                with gr.Row():
-                                    clear_rules_btn = gr.Button("⚠️ Clear All Rules", variant="stop")
-                                    save_faiss_ram_btn_kb = gr.Button("Save FAISS Indices", visible=(MEMORY_STORAGE_BACKEND == "RAM"))
-                        
-                        with gr.Column(scale=1):
-                            with gr.Group():
-                                gr.Markdown("### 📚 Memories (Past Interactions)")
-                                mems_disp_json = gr.JSON(label="View Memories (JSON format)") 
-                                with gr.Row(): view_mems_btn = gr.Button("🔄 Load/Refresh Memories")
-                                upload_mems_fobj = gr.File(label="Upload Memories File (.jsonl)", file_types=[".jsonl"])
-                                mems_stat_tb = gr.Textbox(label="Memories Operation Status", interactive=False, lines=2, placeholder="Status...")
-                                clear_mems_btn = gr.Button("⚠️ Clear All Memories", variant="stop")
+                    with gr.TabItem("🧠 Knowledge Base Management"):
+                        gr.Markdown("## Manage Stored Knowledge")
+                        with gr.Row(equal_height=False):
+                            with gr.Column(scale=1):
+                                with gr.Group():
+                                    gr.Markdown("### 📜 Rules (Learned Insights)")
+                                    rules_disp_ta = gr.TextArea(label="View/Edit Rules", lines=15, interactive=True, placeholder="Load or type rules here (one per line or '---' separated for multiple).")
+                                    with gr.Row():
+                                        view_rules_btn = gr.Button("🔄 Load/Refresh Rules"); save_edited_rules_btn = gr.Button("💾 Save Edited Rules", variant="primary")
+                                    upload_rules_fobj = gr.File(label="Upload Rules File (.txt/.jsonl)", file_types=[".txt", ".jsonl"])
+                                    rules_stat_tb = gr.Textbox(label="Rules Operation Status", interactive=False, lines=2, placeholder="Status...")
+                                    with gr.Row(): clear_rules_btn = gr.Button("⚠️ Clear All Rules", variant="stop")
+                                    
+                            with gr.Column(scale=1): # This was nested incorrectly before
+                                with gr.Group():
+                                    gr.Markdown("### 📚 Memories (Past Interactions)")
+                                    mems_disp_json = gr.JSON(label="View Memories (JSON format)") 
+                                    with gr.Row(): view_mems_btn = gr.Button("🔄 Load/Refresh Memories")
+                                    upload_mems_fobj = gr.File(label="Upload Memories File (.jsonl)", file_types=[".jsonl"])
+                                    mems_stat_tb = gr.Textbox(label="Memories Operation Status", interactive=False, lines=2, placeholder="Status...")
+                                    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)
@@ -659,9 +535,9 @@ with gr.Blocks(theme=custom_theme, css=custom_css, title="AI Research Agent") as
     upload_rules_fobj.upload(fn=ui_upload_rules_action_fn, inputs=[upload_rules_fobj], outputs=[rules_stat_tb], show_progress="full").then(fn=ui_view_rules_action_fn, outputs=rules_disp_ta)
     clear_rules_btn.click(fn=lambda: "All rules cleared." if clear_all_rules_data_backend() else "Error clearing rules.", outputs=rules_stat_tb).then(fn=ui_view_rules_action_fn, outputs=rules_disp_ta)
     
-    if MEMORY_STORAGE_BACKEND == "RAM" and 'save_faiss_ram_btn_kb' in locals() and save_faiss_ram_btn_kb is not None : 
-        def save_faiss_action_with_feedback_fn_kb(): save_faiss_indices_to_disk(); gr.Info("Attempted to save FAISS indices to disk.")
-        save_faiss_ram_btn_kb.click(fn=save_faiss_action_with_feedback_fn_kb, inputs=None, outputs=None)
+    if MEMORY_STORAGE_BACKEND == "RAM" and 'save_faiss_sidebar_btn' in locals() and save_faiss_sidebar_btn is not None: # Check button in sidebar
+        def save_faiss_action_with_feedback_sidebar_fn(): save_faiss_indices_to_disk(); gr.Info("Attempted to save FAISS indices to disk (Sidebar).")
+        save_faiss_sidebar_btn.click(fn=save_faiss_action_with_feedback_sidebar_fn, inputs=None, outputs=None)
     
     view_mems_btn.click(fn=ui_view_memories_action_fn, outputs=mems_disp_json)
     upload_mems_fobj.upload(fn=ui_upload_memories_action_fn, inputs=[upload_mems_fobj], outputs=[mems_stat_tb], show_progress="full").then(fn=ui_view_memories_action_fn, outputs=mems_disp_json)
@@ -678,7 +554,7 @@ with gr.Blocks(theme=custom_theme, css=custom_css, title="AI Research Agent") as
     demo.load(fn=app_load_fn, inputs=None, outputs=[agent_stat_tb, rules_disp_ta, mems_disp_json])
 
 if __name__ == "__main__":
-    logger.info(f"Starting Gradio AI Research Mega Agent (v5.5 - UI Style Update, Memory: {MEMORY_STORAGE_BACKEND})...")
+    logger.info(f"Starting Gradio AI Research Mega Agent (v5.6 - UI Layout Fix, Memory: {MEMORY_STORAGE_BACKEND})...")
     app_port, app_server = int(os.getenv("GRADIO_PORT", 7860)), os.getenv("GRADIO_SERVER_NAME", "127.0.0.1")
     app_debug, app_share = os.getenv("GRADIO_DEBUG", "False").lower()=="true", os.getenv("GRADIO_SHARE", "False").lower()=="true"
     logger.info(f"Launching Gradio server: http://{app_server}:{app_port}. Debug: {app_debug}, Share: {app_share}")