Update main_web.py

main_web.py

@@ -1,334 +1,97 @@
+# FILE: main_web.py (Hugging Face Demo - Final Upgraded Version)
+
 import gradio as gr
 import asyncio
 import logging
 import os
 import sys
-import time
-import random
-from typing import Optional, Dict, Any, List, Union, Tuple
-import html
-import json
-import tempfile
+from typing import Optional, List, Dict, Any
 
-# --- Constants ---
+# --- Basic Setup ---
 APP_TITLE = "ZOTHEOS - Ethical Fusion AI"
-FAVICON_FILENAME = "favicon_blackBW.ico"
-LOGO_FILENAME = "zotheos_logo.png"
-
-# --- Logger (Your existing code is perfect) ---
-def init_logger():
-    logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s", datefmt="%Y-%m-%d %H:%M:%S")
-    logger_instance = logging.getLogger("ZOTHEOS_Interface")
-    if not logger_instance.handlers:
-        handler = logging.StreamHandler(sys.stdout); handler.setFormatter(logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s", datefmt="%Y-%m-%d %H:%M:%S")); logger_instance.addHandler(handler)
-    return logger_instance
-logger = init_logger()
+logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+logger = logging.getLogger("ZOTHEOS_Interface_HF")
 
-# --- Asset Path Helper (Your existing code is perfect) ---
-def get_asset_path(filename: str) -> Optional[str]:
-    base_dir = getattr(sys, '_MEIPASS', os.path.dirname(os.path.abspath(__file__)))
-    for path in [os.path.join(base_dir, 'assets', filename), os.path.join(base_dir, filename)]:
-        if os.path.exists(path):
-            logger.info(f"Asset found: '{filename}' -> '{path}'")
-            return path
-    logger.warning(f"Asset NOT FOUND: '{filename}'.")
-    return None
+def get_asset_path(filename: str) -> str:
+    # In Hugging Face Spaces, assets are in the root or /assets
+    return filename if os.path.exists(filename) else os.path.join('assets', filename)
 
-logo_path_verified = get_asset_path(LOGO_FILENAME)
-favicon_path_verified = get_asset_path(FAVICON_FILENAME)
+logo_path_verified = get_asset_path("zotheos_logo.png")
+GUMROAD_LINK = "YOUR_GUMROAD_LINK_HERE" # <-- IMPORTANT: REPLACE WITH YOUR GUMROAD LINK
 
-# --- Core Logic Imports (Your existing code is perfect) ---
-MainFusionPublic = None; initialization_error: Optional[Exception] = None; DEFAULT_INFERENCE_PRESET_INTERFACE = "balanced"; get_user_tier = lambda token: "error (auth module failed)"
+# --- Core Logic Imports ---
 try:
-    script_dir = os.path.dirname(os.path.abspath(__file__)); project_root = script_dir
-    if project_root not in sys.path: sys.path.insert(0, project_root)
     from modules.main_fusion_public import MainFusionPublic
-    try: from modules.config_settings_public import DEFAULT_INFERENCE_PRESET as CONFIG_DEFAULT_PRESET; DEFAULT_INFERENCE_PRESET_INTERFACE = CONFIG_DEFAULT_PRESET
-    except ImportError: pass
-    from modules.user_auth import get_user_tier as imported_get_user_tier; get_user_tier = imported_get_user_tier
-except Exception as e: initialization_error = e
-
-# --- Initialize AI System (Your existing code is perfect) ---
-ai_system: Optional[MainFusionPublic] = None
-if 'MainFusionPublic' in globals() and MainFusionPublic is not None and initialization_error is None:
-    try: ai_system = MainFusionPublic()
-    except Exception as e_init: initialization_error = e_init
-elif initialization_error is None: initialization_error = ModuleNotFoundError("MainFusionPublic module could not be imported.")
-
-# --- TIER_FEATURES Dictionary (Your existing code is perfect) ---
-TIER_FEATURES = {"free": {"display_name": "Free Tier", "memory_enabled": False, "export_enabled": False}, "starter": {"display_name": "Starter Tier", "memory_enabled": True, "export_enabled": False}, "pro": {"display_name": "Pro Tier", "memory_enabled": True, "export_enabled": True}, "error (auth module failed)": {"display_name": "Error", "memory_enabled": False, "export_enabled": False}}
+except ImportError:
+    MainFusionPublic = None
+ai_system = MainFusionPublic() if MainFusionPublic else None
 
-# --- JESSICA WALSH INSPIRED REDESIGN ---
-zotheos_base_css = """
+# --- ✅ DEFINITIVE "OBSIDIAN" CSS THEME ---
+zotheos_web_css = """
 @import url('https://fonts.googleapis.com/css2?family=Bebas+Neue&family=Inter:wght@400;600;700&display=swap');
-
-:root {
-    --bg-main: #050505;
-    --bg-surface: #121212;
-    --bg-surface-raised: #1A1A1A;
-    --text-primary: #EAEAEA;
-    --text-secondary: #888888;
-    --border-subtle: #2A2A2A;
-    --accent-glow: #00BFFF; /* Electric "Truth" Blue */
-    --accent-glow-darker: #009ACD;
-
-    --font-display: 'Bebas Neue', 'Staatliches', cursive;
-    --font-body: 'Inter', sans-serif;
-
-    --radius-md: 12px;
-    --radius-sm: 8px;
-    --shadow-glow: 0 0 15px rgba(0, 191, 255, 0.2);
-}
-
-/* --- Base & Reset --- */
-footer, canvas { display: none !important; }
-html, body {
-    background-color: var(--bg-main) !important;
-    font-family: var(--font-body);
-    color: var(--text-primary);
-    line-height: 1.6;
-    scroll-behavior: smooth;
-}
-.gradio-container {
-    max-width: 800px !important;
-    margin: 0 auto !important;
-    padding: 2rem 1.5rem !important;
-    background-color: transparent !important;
-}
-
-/* --- Typographic Hierarchy --- */
-h1, h2, h3, #header_subtitle, #welcome_message {
-    font-family: var(--font-display) !important;
-    text-transform: uppercase;
-    letter-spacing: 1.5px;
-    color: var(--text-primary);
-    text-align: center;
-}
-
-/* --- Header & Branding --- */
-#header_column { margin-bottom: 2rem; }
-#header_logo img {
-    max-height: 70px;
-    filter: brightness(0) invert(1); /* Force pure white logo */
-}
-#header_subtitle { font-size: 1.2rem !important; color: var(--text-secondary); }
-#welcome_message {
-    font-size: 1.5rem !important;
-    color: var(--text-primary);
-    margin-bottom: 2.5rem;
-}
-
-/* --- Interactive Elements & Input Areas --- */
-.gradio-group {
-    background: transparent !important;
-    border: none !important;
-    padding: 0 !important;
-}
-.gradio-textbox textarea, .gradio-textbox input {
-    background-color: var(--bg-surface) !important;
-    color: var(--text-primary) !important;
-    border: 1px solid var(--border-subtle) !important;
-    border-radius: var(--radius-md) !important;
-    padding: 1rem !important;
-    transition: border-color 0.3s, box-shadow 0.3s;
-}
-.gradio-textbox textarea:focus, .gradio-textbox input:focus {
-    border-color: var(--accent-glow) !important;
-    box-shadow: var(--shadow-glow) !important;
-}
-#query_input textarea { min-height: 180px !important; }
-
-/* --- Buttons: The Core Interaction --- */
-.gradio-button {
-    border-radius: var(--radius-md) !important;
-    transition: all 0.2s ease-in-out;
-    border: none !important;
-}
-#submit_button {
-    background: var(--accent-glow) !important;
-    color: var(--bg-main) !important;
-    font-weight: 700 !important;
-    text-transform: uppercase;
-}
-#submit_button:hover {
-    background: var(--text-primary) !important;
-    transform: translateY(-2px);
-    box-shadow: 0 4px 20px rgba(0,0,0,0.2);
-}
-#clear_button {
-    background: var(--bg-surface-raised) !important;
-    color: var(--text-secondary) !important;
-}
-#clear_button:hover {
-    background: var(--border-subtle) !important;
-    color: var(--text-primary) !important;
-}
-#export_memory_button {
-    background: transparent !important;
-    color: var(--text-secondary) !important;
-    border: 1px solid var(--border-subtle) !important;
-}
-#export_memory_button:hover {
-    color: var(--text-primary) !important;
-    border-color: var(--accent-glow) !important;
-}
-
-/* --- Output & Information Display --- */
-#synthesized_summary_output, #fusion_output {
-    background-color: var(--bg-surface) !important;
-    border: 1px solid var(--border-subtle) !important;
-    border-radius: var(--radius-md) !important;
-    padding: 1.5rem !important;
-    margin-top: 1rem;
-}
-#synthesized_summary_output h2, #fusion_output h2 { font-size: 1.2rem; color: var(--accent-glow); }
-
-/* --- Accordions & Tier Status --- */
-.gradio-accordion { border: none !important; }
-.gradio-accordion > .label-wrap {
-    background: var(--bg-surface-raised) !important;
-    border-radius: var(--radius-sm) !important;
-    padding: 0.75rem 1rem !important;
-    color: var(--text-secondary) !important;
-}
-#tier_status_display { text-align: right !important; }
-#tier_status_display small { color: var(--text-secondary) !important; }
-
-/* --- Loading Indicator --- */
-.cosmic-loading { padding: 3rem; }
-.orbital-spinner {
-    border-color: var(--border-subtle);
-    border-top-color: var(--accent-glow);
-}
-.thinking-text { font-family: var(--font-body); color: var(--text-secondary); }
-
-/* --- Footer --- */
-#footer_attribution {
-    font-size: 0.8rem;
-    color: #555;
-    text-align: center;
-    margin-top: 4rem;
-    padding-bottom: 1rem;
-    line-height: 1.5;
-}
-#footer_attribution strong { color: #777; font-weight: 600; }
+body { background: linear-gradient(135deg, #000000, #1c1c1c) !important; font-family: 'Inter', sans-serif !important; }
+.gradio-container { background: transparent !important; max-width: 900px !important; }
+#header_logo img { max-height: 70px !important; filter: brightness(0) invert(1) !important; }
+#header_subtitle { font-family: 'Bebas Neue', cursive !important; font-size: 1.8rem !important; text-transform: uppercase !important; letter-spacing: 2px !important; }
+#header_tagline { color: #a0a0a0 !important; margin-top: -10px !important; margin-bottom: 1.5rem !important; }
+.gradio-textbox textarea { font-size: 1.1rem !important; }
+.result-box { background-color: rgba(18, 18, 18, 0.8) !important; border: 1px solid #333 !important; border-radius: 12px !important; padding: 1.5rem 2rem !important; }
+.result-box h2 { font-family: 'Bebas Neue', cursive !important; font-size: 1.5rem !important; border-bottom: 1px solid #333; padding-bottom: 0.75rem; margin-bottom: 1rem; }
+.cta-button { background: #FFFFFF !important; color: #000000 !important; font-weight: bold !important; }
+.cta-button:hover { background: #e0e0e0 !important; }
+footer { display: none !important; }
 """
 
-# --- Helper Functions (Your existing code is perfect) ---
-def render_memory_entries_as_html(memory_entries: List[dict]) -> str:
-    if not memory_entries: return "<div class='memory-entry'>No stored memory entries or memory disabled.</div>"
-    html_blocks = [];
-    for entry in reversed(memory_entries[-5:]):
-        query = html.escape(entry.get("query", "N/A")); ts_iso = entry.get("metadata", {}).get("timestamp_iso", "Unknown");
-        try: dt_obj = time.strptime(ts_iso, '%Y-%m-%dT%H:%M:%SZ'); formatted_ts = time.strftime('%Y-%m-%d %H:%M UTC', dt_obj)
-        except: formatted_ts = html.escape(ts_iso)
-        models_list = entry.get("metadata", {}).get("active_models_queried", []); models_str = html.escape(", ".join(models_list)) if models_list else "N/A"
-        summary_raw = entry.get("metadata", {}).get("synthesized_summary_text", "No summary."); summary = html.escape(summary_raw[:300]) + ("..." if len(summary_raw) > 300 else "")
-        html_blocks.append(f"<div class='memory-entry'><p><strong>🕒 TS:</strong> {formatted_ts}</p><p><strong>🧠 Models:</strong> {models_str}</p><div><strong>❓ Q:</strong><div class='query-text'>{query}</div></div><div><strong>📌 Sum:</strong><div class='summary-text'>{summary}</div></div></div>")
-    return "\n".join(html_blocks)
-
-def format_tier_status_for_ui(tier_name: str) -> str:
-    features = TIER_FEATURES.get(tier_name, TIER_FEATURES["error (auth module failed)"])
-    return f"<small>Tier: **{features['display_name']}** | Memory: {'✅' if features['memory_enabled'] else '❌'} | Export: {'✅' if features['export_enabled'] else '❌'}</small>"
-
-def update_tier_display_and_features_ui(user_token_str: Optional[str]) -> Tuple:
-    tier_name = get_user_tier(user_token_str or "")
-    tier_features = TIER_FEATURES.get(tier_name, TIER_FEATURES["free"])
-    return (gr.update(value=format_tier_status_for_ui(tier_name)), gr.update(interactive=tier_features["export_enabled"]))
-
-def export_user_memory_data_for_download(user_token_str: Optional[str]):
-    # This is a placeholder for your actual export logic
-    gr.Info("Export functionality would be triggered here for Pro users.")
-    return None
-
-# --- Build Gradio Interface (Main Structure is Yours, Components are Styled) ---
-def build_interface(logo_path_param: Optional[str], favicon_path_param: Optional[str]) -> gr.Blocks:
-    theme = gr.themes.Base(font=[gr.themes.GoogleFont("Inter"), "sans-serif"])
-    initial_tier_name = get_user_tier("")
-    initial_tier_features = TIER_FEATURES.get(initial_tier_name, TIER_FEATURES["free"])
-
-    with gr.Blocks(theme=theme, css=zotheos_base_css, title=APP_TITLE) as demo:
-        with gr.Column(elem_classes="centered-container"):
-            # --- Header ---
-            with gr.Column(elem_id="header_column"):
-                if logo_path_param:
-                    gr.Image(value=logo_path_param, elem_id="header_logo", show_label=False, container=False, interactive=False)
-                gr.Markdown("Ethical Fusion AI for Synthesized Intelligence", elem_id="header_subtitle")
+# --- Build Gradio Interface ---
+def build_interface():
+    zotheos_theme = gr.themes.Base(primary_hue=gr.themes.colors.neutral, secondary_hue=gr.themes.colors.neutral).set(
+        button_primary_background_fill="white", button_primary_text_color="black"
+    )
+
+    with gr.Blocks(theme=zotheos_theme, css=zotheos_web_css, title=APP_TITLE) as demo:
+        with gr.Column(elem_classes="gradio-container"):
+            # --- ✅ UPGRADED HEADER WITH CALL TO ACTION ---
+            gr.Image(value=logo_path_verified, elem_id="header_logo", show_label=False, container=False, interactive=False)
+            gr.Markdown("Ethical Fusion AI for Synthesized Intelligence", elem_id="header_subtitle")
+            gr.Markdown("Fusing perspectives for deeper truth.", elem_id="header_tagline")
             
-            gr.Markdown("Fusing perspectives for deeper truth.", elem_id="welcome_message")
-
-            # --- Authentication & Tier Status ---
-            with gr.Row(equal_height=False):
-                user_token_input = gr.Textbox(label="Access Token", placeholder="Enter token...", type="password", scale=3, container=False)
-                # --- FIX IS HERE ---
-                tier_status_display = gr.Markdown(value=format_tier_status_for_ui(initial_tier_name), elem_id="tier_status_display")
-
-            # --- Core Input ---
-            with gr.Group():
-                query_input_component = gr.Textbox(label="Your Inquiry:", placeholder="e.g., Analyze the ethical implications of AI in art...", lines=6, elem_id="query_input")
-                status_indicator_component = gr.HTML(elem_id="status_indicator", visible=False)
-                with gr.Row():
-                    clear_button_component = gr.Button("Clear", elem_id="clear_button", variant="secondary")
-                    submit_button_component = gr.Button("Process Inquiry", elem_id="submit_button", variant="primary", scale=3)
-
-            # --- Main Outputs ---
-            with gr.Group():
-                synthesized_summary_component = gr.Markdown(elem_id="synthesized_summary_output", value="Synthesized insight will appear here...", visible=True)
-                fused_response_output_component = gr.Markdown(elem_id="fusion_output", value="Detailed perspectives will appear here...", visible=True)
+            with gr.Accordion("🔥 Get the Full Offline Desktop Version", open=True):
+                gr.Markdown("This web demo uses smaller, slower models. For the full-power, GPU-accelerated, 100% private experience, download the ZOTHEOS Public Beta for your Windows PC.")
+                download_button = gr.Button("Download Full Version on Gumroad", link=GUMROAD_LINK, elem_classes="cta-button")
             
-            # --- Tools & Info ---
-            export_memory_button = gr.Button("Export Memory Log", elem_id="export_memory_button", interactive=initial_tier_features["export_enabled"])
-            with gr.Accordion("System Status & Active Models", open=False):
-                 active_models_display_component = gr.Markdown("**Status:** Initializing...")
-            with gr.Accordion("Recent Interaction Chronicle", open=False):
-                memory_display_panel_html = gr.HTML("<div class='memory-entry'>No memory entries yet.</div>")
-
-            # --- Footer ---
-            gr.Markdown("---", elem_id="footer_separator")
-            gr.Markdown("ZOTHEOS © 2025 ZOTHEOS LLC. System Architect: David A. Garcia. All Rights Reserved.", elem_id="footer_attribution")
+            gr.Markdown("---") # Separator
 
-        # --- Your original, robust backend logic should be here ---
-        # I'm using a simplified version for demonstration.
-        async def process_query_wrapper_internal(query, token):
-            tier_name = get_user_tier(token or ""); display_name = TIER_FEATURES.get(tier_name, {})["display_name"]
-            loading_html = f"<div class='cosmic-loading'><div class='orbital-spinner'></div><div class='thinking-text'>Synthesizing (Tier: {display_name})...</div></div>"
-            yield (gr.update(value=loading_html, visible=True), gr.update(interactive=False), gr.update(interactive=False), "Processing...", "Processing...", "Processing...", "Processing...")
-            if not ai_system:
-                yield (gr.update(value="<p style='color:red;'>SYSTEM OFFLINE</p>", visible=True), gr.update(interactive=True), gr.update(interactive=True), "Error", "Error", "Offline", "Error")
-                return
+            # --- Main Interaction Area ---
+            query_input = gr.Textbox(label="Your Inquiry:", placeholder="e.g., Analyze the ethical implications of AI in art...", lines=6)
+            submit_button = gr.Button("Process Inquiry (Web Demo)", variant="primary")
             
-            # Replace with your actual full response logic
-            response = "## ✨ Final Synthesized Insight ✨\nThis is a placeholder summary. ### 💬 Detailed Perspectives\nThis is a placeholder for the detailed views."
-            if hasattr(ai_system, 'process_query_with_fusion'):
-                 response = await ai_system.process_query_with_fusion(query, user_token=token, fusion_mode_override=DEFAULT_INFERENCE_PRESET_INTERFACE)
-
-            summary = response[:response.find("###")] if "###" in response else response
-            perspectives = response[response.find("###"):] if "###" in response else "Details integrated in summary."
+            with gr.Column(visible=False) as results_box:
+                gr.Markdown("## ✨ ZOTHEOS Final Synthesized Insight ✨")
+                synthesized_summary_output = gr.Markdown()
+                gr.Markdown("### 💬 Detailed Individual Perspectives")
+                fusion_output = gr.Markdown()
+
+        # --- Backend Logic ---
+        async def process_query_wrapper(query):
+            if not ai_system:
+                return gr.update(visible=True), "[SYSTEM OFFLINE] The AI engine failed to load.", ""
             
-            mem_html = "<div class='memory-entry'>Memory disabled.</div>"
-            if TIER_FEATURES[get_user_tier(token or "")]['memory_enabled'] and hasattr(ai_system, 'memory_bank'):
-                mem_data = await ai_system.memory_bank.retrieve_recent_memories_async(limit=5)
-                mem_html = render_memory_entries_as_html(mem_data)
-
-            status_md = "**Status:** `Ready`"
-            if hasattr(ai_system, 'get_status_report'):
-                status_report = await ai_system.get_status_report()
-                status_md = f"**Last Used:** {', '.join(status_report.get('models_last_queried_for_perspectives', ['N/A']))}"
+            response = await ai_system.process_query_with_fusion(query)
+            summary = response.split("###")[0].replace("## ✨ ZOTHEOS Final Synthesized Insight ✨", "").strip()
+            perspectives = "###" + response.split("###", 1)[1] if "###" in response else ""
             
-            yield (gr.update(visible=False), gr.update(interactive=True), gr.update(interactive=True), summary, perspectives, status_md, mem_html)
-
-        # Connect events
-        submit_button_component.click(fn=process_query_wrapper_internal, inputs=[query_input_component, user_token_input], outputs=[status_indicator_component, submit_button_component, clear_button_component, synthesized_summary_component, fused_response_output_component, active_models_display_component, memory_display_panel_html], show_progress="hidden")
-        clear_button_component.click(lambda: ("", "", "Synthesized insight...", "Detailed perspectives..."), outputs=[query_input_component, user_token_input, synthesized_summary_component, fused_response_output_component])
-        user_token_input.change(fn=update_tier_display_and_features_ui, inputs=[user_token_input], outputs=[tier_status_display, export_memory_button])
-        export_memory_button.click(fn=export_user_memory_data_for_download, inputs=[user_token_input], outputs=[])
+            return gr.update(visible=True), summary, perspectives
 
+        submit_button.click(
+            fn=process_query_wrapper,
+            inputs=[query_input],
+            outputs=[results_box, synthesized_summary_output, fusion_output]
+        )
     return demo
 
 # --- Main Execution Block ---
 if __name__ == "__main__":
-    logger.info("--- Initializing ZOTHEOS Gradio Interface (V13 - Walsh Redesign, Corrected) ---")
-    zotheos_interface = build_interface(logo_path_verified, favicon_path_verified)
-    logger.info("✅ ZOTHEOS Gradio UI built.")
-    launch_kwargs = {"server_name": "0.0.0.0", "server_port": int(os.getenv("PORT", 7860)), "share": os.getenv("GRADIO_SHARE", "False").lower() == "true"}
-    zotheos_interface.queue().launch(**launch_kwargs)
+    logger.info("--- Initializing ZOTHEOS Hugging Face Demo ---")
+    zotheos_interface = build_interface()
+    logger.info("✅ HF Demo UI built.")
+    zotheos_interface.queue().launch()