Update main.py

main.py

@@ -1,82 +1,58 @@
-import os
-import sys
 import warnings
 
-# 🟢 CRITICAL: Disable SSR at the environment level so Hugging Face's internal server respects it
-os.environ["GRADIO_SSR_MODE"] = "False"
-
-# 🟢 CRITICAL: Must import gradio AFTER setting the env var, but BEFORE using it
-import gradio as gr 
-from huggingface_hub import HfApi
-
 # Filter warnings immediately
 warnings.filterwarnings("ignore", category=SyntaxWarning, module="pydub")
 
-# Add the 'src' directory to Python path
-sys.path.append(os.path.join(os.path.dirname(__file__), "src"))
-
 # --- INTERNAL IMPORTS ---
 from config import AppConfig
 from src.utils import create_warning_beep
+
+# Import your refactored classes
 from src.managers import GeminiManager, HFManager
 from src.input_agent import AgentInput
 from src.brain_agent import AgentInterpretation
 from src.trust_agent import AgentTrust
 from src.ux_agent import AgentUX
-from src.bridge_agent import AgentBridge
-from src.watcher_agent import AgentWatcher
-
-# ==========================================
-# 🚀 GLOBAL INITIALIZATION
-# ==========================================
-print("🚀 Initializing Modular Architecture IEDI (MA-IEDI)...")
-
-# 1. Setup Configuration & Directories
-AppConfig.setup_directories()
-
-# 2. Initialize Managers
-hf_manager = HFManager(AppConfig)
-hf_manager.pull_datasets()
-
-gemini_manager = GeminiManager(AppConfig.GOOGLE_API_KEY) if AppConfig.GOOGLE_API_KEY else None
-
-# 3. Initialize Agents
-agent1 = AgentInput()
-agent2 = AgentInterpretation(AppConfig, gemini_manager_instance=gemini_manager)
-agent4 = AgentTrust(AppConfig)
-agent4.hf_manager_ref = hf_manager 
-
-# 4. Initialize The Watcher (Background Logic)
-watcher_agent = AgentWatcher(AppConfig)
-
-# 5. Initialize Interfaces
-agent3 = AgentUX(agent1, agent2, agent4)
-agent_bridge = AgentBridge(agent1, agent2, agent4)
-agent2.ux = agent3 # Circular link for updates
-agent3.WARNING_BEEP_PATH = create_warning_beep()
-
-print("✅ System Assembled. Launching Interface...")
-
-# 6. Create UIs
-admin_ui = agent3.create_ui()
-api_ui = agent_bridge.launch()
-
-# 🟢 Mount globally as 'app' so Hugging Face can find it
-app = gr.TabbedInterface(
-    [admin_ui, api_ui], 
-    ["👮 Admin Dashboard", "🔗 React Bridge API"]
-)
 
-# 7. Start the Watcher Thread
-print("🕵️‍♂️ Starting Background Watcher...")
-watcher_agent.start()
+def main():
+    print("🚀 Initializing Modular Architecture IEDI (MA-IEDI)...")
+
+    # 1. Setup Configuration & Directories
+    AppConfig.setup_directories()
+    
+    # 2. Initialize Managers (Inject Config)
+    # We pass AppConfig to HFManager so it knows where to download files
+    hf_manager = HFManager(AppConfig)
+    hf_manager.pull_datasets()
+    
+    # Initialize Gemini (Brain Stem)
+    gemini_manager = GeminiManager(AppConfig.GOOGLE_API_KEY) if AppConfig.GOOGLE_API_KEY else None
+
+    # 3. Initialize Agents (Dependency Injection)
+    
+    # Agent 1: Input (Whisper)
+    agent1 = AgentInput()
+    
+    # Agent 2: Brain (Interpretation)
+    # We pass AppConfig so it knows where PROFILES_DIR is
+    agent2 = AgentInterpretation(AppConfig, gemini_manager_instance=gemini_manager)
+    
+    # Agent 4: Trust (Blockchain/IPFS)
+    # We pass AppConfig so it has keys for Web3 and Pinata
+    agent4 = AgentTrust(AppConfig)
+    # Inject hf_manager into Trust so it can save audio files to HF
+    agent4.hf_manager_ref = hf_manager 
+
+    # Agent 3: UX (Interface)
+    agent3 = AgentUX(agent1, agent2, agent4)
+    
+    # 4. Launch
+    print("✅ System Assembled. Launching Interface...")
+    
+    # Pass the warning beep path
+    agent3.WARNING_BEEP_PATH = create_warning_beep() 
+    
+    agent3.launch()
 
-# 8. Local Launch Fallback
 if __name__ == "__main__":
-    app.launch(
-        server_name="0.0.0.0", 
-        server_port=7860,
-        show_error=True,
-        allowed_paths=["/app/saved_audio", "/app/iuuy_datasets", "."],
-        ssr_mode=False
-    )
+    main()