Upload 9 files

Patch to add configurable API keys

ui/handlers.py

@@ -115,6 +115,28 @@ class UIHandlers:
             return (f"ℹ️ Using Mock AI for demo (Error: {str(e)}) 🎭",
                    f"**AI Provider:** `MOCK AI (Error Fallback)`")
 
+    def reconfigure_agent(self, provider: str, api_key: str, eleven_key: str) -> tuple:
+        """
+        Re-configure the agent with user-provided keys (Demo Mode).
+        """
+        # Update Environment Variables
+        if api_key.strip():
+            if provider == "openai":
+                os.environ["OPENAI_API_KEY"] = api_key
+            elif provider == "anthropic":
+                os.environ["ANTHROPIC_API_KEY"] = api_key
+            elif provider == "gemini":
+                os.environ["GEMINI_API_KEY"] = api_key
+
+        if eleven_key.strip():
+            os.environ["ELEVEN_API_KEY"] = eleven_key
+            # Re-init voice
+            from voice import voice_generator
+            voice_generator.initialize()
+
+        # Re-initialize Agent
+        return self.initialize_agent(provider)
+
     def process_onboarding(self, project_description: str) -> tuple:
         """Process onboarding and generate tasks."""
         # Default UI updates for failure cases (no change to timer/monitoring)

ui/layout.py

@@ -95,6 +95,33 @@ def create_app(ui_handlers, pomodoro_timer: PomodoroTimer, launch_mode: str, ai_
                     gr.Markdown("""
                     > ℹ️ **Demo Mode**: Use the text area in Monitor tab to simulate your workspace.
                     """)
+
+                    with gr.Accordion("⚙️ Configure Demo (Add your keys)", open=False):
+                        gr.Markdown("If the default demo keys are expired, enter your own here to enable AI features.")
+                        with gr.Row():
+                            config_provider = gr.Dropdown(
+                                label="LLM Provider",
+                                choices=["openai", "anthropic", "gemini"],
+                                value=ai_provider
+                            )
+                            config_api_key = gr.Textbox(
+                                label="LLM API Key",
+                                placeholder="sk-...",
+                                type="password"
+                            )
+                            config_eleven_key = gr.Textbox(
+                                label="ElevenLabs Key (Optional)",
+                                placeholder="sk-...",
+                                type="password"
+                            )
+                        config_save_btn = gr.Button("💾 Save & Re-Initialize", variant="primary")
+
+                        config_save_btn.click(
+                            fn=ui_handlers.reconfigure_agent,
+                            inputs=[config_provider, config_api_key, config_eleven_key],
+                            outputs=[init_status, ai_provider_display],
+                            api_name=False
+                        )
                 else:
                     gr.Markdown("""
                     > ℹ️ **Local Mode**: Monitor your actual project directory.

voice.py

@@ -14,53 +14,57 @@ class VoiceGenerator:
     Handles text-to-speech generation using ElevenLabs API.
     Designed for graceful degradation - never crashes if voice unavailable.
     """
-    
+
     def __init__(self):
         """Initialize ElevenLabs client if API key available."""
+        self.initialize()
+
+    def initialize(self):
+        """Initialize or re-initialize the client."""
         self.client = None
         self.available = False
         self.voice_id = "JBFqnCBsd6RMkjVDRZzb"  # George - friendly, clear voice
         self.model_id = "eleven_turbo_v2_5"  # Fast, low-latency model
-        
+
         try:
             # Check for API key (demo key first, then user key)
             api_key = os.getenv("DEMO_ELEVEN_API_KEY") or os.getenv("ELEVEN_API_KEY")
-            
+
             if not api_key:
                 print("ℹ️ ElevenLabs: No API key found. Voice feedback disabled (text-only mode).")
                 return
-            
+
             # Try to initialize client
             from elevenlabs.client import ElevenLabs
             self.client = ElevenLabs(api_key=api_key)
             self.available = True
-            
+
             key_type = "demo" if os.getenv("DEMO_ELEVEN_API_KEY") else "user"
             print(f"✅ ElevenLabs voice initialized ({key_type} key)")
-            
+
         except ImportError:
             print("⚠️ ElevenLabs: Package not installed. Run: pip install elevenlabs")
         except Exception as e:
             print(f"⚠️ ElevenLabs: Initialization failed: {e}")
-    
+
     def text_to_speech(self, text: str, emotion: str = "neutral") -> Optional[str]:
         """
         Convert text to speech and return path to temporary audio file.
-        
+
         Args:
             text: Text to convert to speech
             emotion: Emotion hint (not used in current implementation)
-        
+
         Returns:
             Path to temporary MP3 file, or None if voice unavailable
         """
         # Check if voice is enabled globally
         if os.getenv("VOICE_ENABLED", "true").lower() == "false":
             return None
-        
+
         if not self.available or not self.client:
             return None
-        
+
         try:
             # Generate audio using ElevenLabs API
             audio = self.client.text_to_speech.convert(
@@ -69,10 +73,10 @@ class VoiceGenerator:
                 model_id=self.model_id,
                 output_format="mp3_44100_128"
             )
-            
+
             # Convert generator/stream to bytes
             audio_bytes = b"".join(audio)
-            
+
             # Save to temporary file (Gradio expects file path, not data URL)
             temp_file = tempfile.NamedTemporaryFile(
                 delete=False,
@@ -81,63 +85,63 @@ class VoiceGenerator:
             )
             temp_file.write(audio_bytes)
             temp_file.close()
-            
+
             return temp_file.name
-            
+
         except Exception as e:
             # Graceful degradation - log error but don't crash
             print(f"⚠️ ElevenLabs: TTS failed: {e}")
             return None
-    
+
     def get_focus_message_audio(self, verdict: str, message: str) -> Optional[str]:
         """
         Generate voice feedback for focus check results.
-        
+
         Args:
             verdict: "On Track", "Distracted", or "Idle"
             message: Text message to speak
-        
+
         Returns:
             Path to temporary audio file or None
         """
         if not self.available:
             return None
-        
+
         # Add emotion/tone based on verdict (for future voice modulation)
         emotion_map = {
             "On Track": "cheerful",
             "Distracted": "concerned",
             "Idle": "motivating"
         }
-        
+
         emotion = emotion_map.get(verdict, "neutral")
         return self.text_to_speech(message, emotion=emotion)
-    
+
     def get_pomodoro_audio(self, event_type: str) -> Optional[str]:
         """
         Generate voice alerts for Pomodoro timer events.
-        
+
         Args:
             event_type: "work_complete" or "break_complete"
-        
+
         Returns:
             Path to temporary audio file or None
         """
         if not self.available:
             return None
-        
+
         messages = {
             "work_complete": "Great work! Time for a 5-minute break. You've earned it!",
             "break_complete": "Break's over! Let's get back to work and stay focused!"
         }
-        
+
         message = messages.get(event_type, "Timer complete!")
         return self.text_to_speech(message, emotion="cheerful")
-    
+
     def test_voice(self) -> Dict[str, any]:
         """
         Test voice generation (for setup/debugging).
-        
+
         Returns:
             Dict with status, message, and optional audio data
         """
@@ -147,11 +151,11 @@ class VoiceGenerator:
                 "message": "Voice not available (no API key or initialization failed)",
                 "audio": None
             }
-        
+
         try:
             test_message = "Hello! FocusFlow voice is working perfectly!"
             audio = self.text_to_speech(test_message)
-            
+
             if audio:
                 return {
                     "status": "success",
@@ -179,7 +183,7 @@ voice_generator = VoiceGenerator()
 def get_voice_status() -> str:
     """
     Get human-readable voice status for UI display.
-    
+
     Returns:
         Status string like "✅ ElevenLabs Voice Enabled" or "ℹ️ Voice Disabled"
     """