Update app.py

app.py

@@ -13,7 +13,7 @@ import xml.etree.ElementTree as ET
 load_dotenv()
 
 # --- DEMO MODE TOGGLE ---
-# If True, disables all database writing functions and locks the model to the default groq provider.
+# If True, prevents the "Clear All" buttons from being displayed in the UI.
 DEMO_MODE = True
 
 from model_logic import (
@@ -189,11 +189,6 @@ def process_user_interaction_gradio(user_input: str, provider_name: str, model_d
     yield "final_response_and_insights", {"response": final_bot_text, "insights_used": parsed_initial_insights_list}
 
 def perform_post_interaction_learning(user_input: str, bot_response: str, provider: str, model_disp_name: str, insights_reflected: list[dict], api_key_override: str = None):
-    # --- DEMO MODE CHANGE ---
-    if DEMO_MODE:
-        logger.info("POST_INTERACTION_LEARNING: Skipping database writes in Demo Mode.")
-        return
-
     task_id = os.urandom(4).hex()
     logger.info(f"POST_INTERACTION_LEARNING [{task_id}]: START User='{user_input[:40]}...', Bot='{bot_response[:40]}...'")
     learning_start_time = time.time()
@@ -363,15 +358,6 @@ Combine all findings into a single, valid XML structure as specified in the syst
 
 
 def handle_gradio_chat_submit(user_msg_txt: str, gr_hist_list: list, sel_prov_name: str, sel_model_disp_name: str, ui_api_key: str|None, cust_sys_prompt: str):
-    # --- DEMO MODE CHANGE ---
-    if DEMO_MODE:
-        sel_prov_name = "groq"
-        sel_model_disp_name = get_default_model_display_name_for_provider("groq")
-        if not sel_model_disp_name:
-            # Fallback in case the function fails
-            sel_model_disp_name = "Llama3 8B 8192"
-        logger.info(f"DEMO MODE: Forcing model to {sel_prov_name}/{sel_model_disp_name}")
-
     global current_chat_session_history
     cleared_input, updated_gr_hist, status_txt = "", list(gr_hist_list), "Initializing..."
     # Initialize all potential output components with default/current values
@@ -498,8 +484,6 @@ def handle_gradio_chat_submit(user_msg_txt: str, gr_hist_list: list, sel_prov_na
                 api_key_override=ui_api_key.strip() if ui_api_key else None
             )
             status_txt = "Response & Learning Complete."
-            if DEMO_MODE:
-                status_txt = "Response Complete. (Learning disabled in Demo Mode)"
         except Exception as e_learn:
             logger.error(f"Error during post-interaction learning: {e_learn}", exc_info=True)
             status_txt = "Response complete. Error during learning."
@@ -706,9 +690,6 @@ def ui_download_rules_action_fn():
         return gr.DownloadButton(value=None, interactive=False, label="Error")
 
 def ui_upload_rules_action_fn(uploaded_file_obj, progress=gr.Progress()):
-    # --- DEMO MODE CHANGE ---
-    if DEMO_MODE:
-        return "Uploading is disabled in Demo Mode."
     if not uploaded_file_obj: return "No file provided for rules upload."
     try:
         with open(uploaded_file_obj.name, 'r', encoding='utf-8') as f: content = f.read()
@@ -785,9 +766,6 @@ def ui_download_memories_action_fn():
         return gr.DownloadButton(value=None, interactive=False, label="Error")
 
 def ui_upload_memories_action_fn(uploaded_file_obj, progress=gr.Progress()):
-    # --- DEMO MODE CHANGE ---
-    if DEMO_MODE:
-        return "Uploading is disabled in Demo Mode."
     if not uploaded_file_obj: return "No file provided for memories upload."
     try:
         with open(uploaded_file_obj.name, 'r', encoding='utf-8') as f: content = f.read()
@@ -846,9 +824,6 @@ def ui_upload_memories_action_fn(uploaded_file_obj, progress=gr.Progress()):
     logger.info(msg); return msg
 
 def save_edited_rules_action_fn(edited_rules_text: str, progress=gr.Progress()):
-    # --- DEMO MODE CHANGE ---
-    if DEMO_MODE:
-        return "Saving is disabled in Demo Mode."
     if not edited_rules_text.strip():
         return "No rules text to save."
 
@@ -890,9 +865,8 @@ def app_load_fn():
     mems_added, mems_format_errors, mems_save_errors = load_memories_from_file(LOAD_MEMORIES_FILE)
     mems_load_msg = f"Memories: Added {mems_added}, Format Errors {mems_format_errors}, Save Errors {mems_save_errors} from {LOAD_MEMORIES_FILE or 'None'}."
     logger.info(mems_load_msg)
-    
-    status_prefix = "DEMO MODE" if DEMO_MODE else "AI Systems Initialized"
-    final_status = f"{status_prefix}. {rules_load_msg} {mems_load_msg} Ready."
+
+    final_status = f"AI Systems Initialized. {rules_load_msg} {mems_load_msg} Ready."
 
     # Initial population of all relevant UI components AFTER loading
     rules_on_load = ui_refresh_rules_display_fn()
@@ -923,8 +897,8 @@ with gr.Blocks(
     """
 ) as demo:
     gr.Markdown(
-        f"""
-        # 🤖 AI Research Agent {'(DEMO MODE)' if DEMO_MODE else ''}
+        """
+        # 🤖 AI Research Agent
         Your intelligent assistant for research and knowledge management
         """,
         elem_classes=["header"]
@@ -962,24 +936,16 @@ with gr.Blocks(
                 )
                 available_providers = get_available_providers()
                 default_provider = available_providers[0] if available_providers else None
-
-                # --- DEMO MODE CHANGES to Dropdowns ---
                 prov_sel_dd = gr.Dropdown(
-                    label="AI Provider",
-                    choices=["groq"] if DEMO_MODE else available_providers,
-                    value="groq" if DEMO_MODE else default_provider,
-                    interactive=not DEMO_MODE
+                    label="AI Provider", choices=available_providers,
+                    value=default_provider, interactive=True
                 )
-                
-                initial_provider = prov_sel_dd.value
-                initial_models = get_model_display_names_for_provider(initial_provider) if initial_provider else []
-                initial_model_val = get_default_model_display_name_for_provider(initial_provider) if initial_provider else None
-
+                default_model_display = get_default_model_display_name_for_provider(default_provider) if default_provider else None
                 model_sel_dd = gr.Dropdown(
                     label="AI Model",
-                    choices=initial_models,
-                    value=initial_model_val,
-                    interactive=not DEMO_MODE
+                    choices=get_model_display_names_for_provider(default_provider) if default_provider else [],
+                    value=default_model_display,
+                    interactive=True
                 )
             with gr.Group():
                 gr.Markdown("### System Prompt")
@@ -1024,17 +990,14 @@ with gr.Blocks(
                                 interactive=True
                             )
                             gr.Markdown("To edit rules, modify the text above and click 'Save Edited Text', or upload a new file.")
-                            # --- DEMO MODE CHANGE ---
-                            save_edited_rules_btn = gr.Button("💾 Save Edited Text", variant="primary", interactive=not DEMO_MODE)
+                            save_edited_rules_btn = gr.Button("💾 Save Edited Text", variant="primary")
                             with gr.Row(variant="compact"):
                                 dl_rules_btn = gr.DownloadButton("⬇️ Download Rules", value=None)
-                                # --- DEMO MODE CHANGE ---
-                                clear_rules_btn = gr.Button("🗑️ Clear All Rules", variant="stop", interactive=not DEMO_MODE)
-                            # --- DEMO MODE CHANGE ---
+                                # --- DEMO MODE CHANGE: Button is hidden in demo mode ---
+                                clear_rules_btn = gr.Button("🗑️ Clear All Rules", variant="stop", visible=not DEMO_MODE)
                             upload_rules_fobj = gr.File(
                                 label="Upload Rules File (.txt with '---' separators, or .jsonl of rule strings)",
-                                file_types=[".txt", ".jsonl"],
-                                interactive=not DEMO_MODE
+                                file_types=[".txt", ".jsonl"]
                             )
                             rules_stat_tb = gr.Textbox(
                                 label="Rules Status", interactive=False, lines=1, elem_classes=["status-text"]
@@ -1048,13 +1011,11 @@ with gr.Blocks(
                             gr.Markdown("To add memories, upload a .jsonl or .json file.")
                             with gr.Row(variant="compact"):
                                 dl_mems_btn = gr.DownloadButton("⬇️ Download Memories", value=None)
-                                # --- DEMO MODE CHANGE ---
-                                clear_mems_btn = gr.Button("🗑️ Clear All Memories", variant="stop", interactive=not DEMO_MODE)
-                            # --- DEMO MODE CHANGE ---
+                                # --- DEMO MODE CHANGE: Button is hidden in demo mode ---
+                                clear_mems_btn = gr.Button("🗑️ Clear All Memories", variant="stop", visible=not DEMO_MODE)
                             upload_mems_fobj = gr.File(
                                 label="Upload Memories File (.jsonl of memory objects, or .json array of objects)",
-                                file_types=[".jsonl", ".json"],
-                                interactive=not DEMO_MODE
+                                file_types=[".jsonl", ".json"]
                             )
                             mems_stat_tb = gr.Textbox(
                                 label="Memories Status", interactive=False, lines=1, elem_classes=["status-text"]
@@ -1094,9 +1055,8 @@ with gr.Blocks(
         show_progress="full"
     ).then(fn=ui_refresh_rules_display_fn, outputs=rules_disp_ta, show_progress=False) # Refresh display after upload
 
-    # --- DEMO MODE CHANGE ---
     clear_rules_btn.click(
-        fn=lambda: "Clearing is disabled in Demo Mode." if DEMO_MODE else ("All rules cleared." if clear_all_rules_data_backend() else "Error clearing rules."),
+        fn=lambda: ("All rules cleared." if clear_all_rules_data_backend() else "Error clearing rules."),
         outputs=rules_stat_tb,
         show_progress=False
     ).then(fn=ui_refresh_rules_display_fn, outputs=rules_disp_ta, show_progress=False) # Refresh display after clear
@@ -1111,9 +1071,8 @@ with gr.Blocks(
         show_progress="full"
     ).then(fn=ui_refresh_memories_display_fn, outputs=mems_disp_json, show_progress=False) # Refresh display after upload
 
-    # --- DEMO MODE CHANGE ---
     clear_mems_btn.click(
-        fn=lambda: "Clearing is disabled in Demo Mode." if DEMO_MODE else ("All memories cleared." if clear_all_memory_data_backend() else "Error clearing memories."),
+        fn=lambda: ("All memories cleared." if clear_all_memory_data_backend() else "Error clearing memories."),
         outputs=mems_stat_tb,
         show_progress=False
     ).then(fn=ui_refresh_memories_display_fn, outputs=mems_disp_json, show_progress=False) # Refresh display after clear