Update notification service

AppImage/scripts/ai_providers/gemini_provider.py

@@ -1,9 +1,12 @@
 """Google Gemini provider implementation.
 
 Google's Gemini models offer a free tier and excellent quality/price ratio.
-Gemini 2.0 Flash is fast and cost-effective with improved capabilities.
+Models are loaded dynamically from the API - no hardcoded model names.
 """
-from typing import Optional
+from typing import Optional, List
+import json
+import urllib.request
+import urllib.error
 from .base import AIProvider, AIProviderError
 
 
@@ -11,10 +14,44 @@ class GeminiProvider(AIProvider):
     """Google Gemini provider using the Generative Language API."""
     
     NAME = "gemini"
-    DEFAULT_MODEL = "gemini-2.0-flash"
     REQUIRES_API_KEY = True
     API_BASE = "https://generativelanguage.googleapis.com/v1beta/models"
     
+    def list_models(self) -> List[str]:
+        """List available Gemini models that support generateContent.
+        
+        Returns:
+            List of model IDs available for text generation.
+        """
+        if not self.api_key:
+            return []
+        
+        try:
+            url = f"{self.API_BASE}?key={self.api_key}"
+            req = urllib.request.Request(url, method='GET')
+            
+            with urllib.request.urlopen(req, timeout=10) as resp:
+                data = json.loads(resp.read().decode('utf-8'))
+            
+            models = []
+            for model in data.get('models', []):
+                model_name = model.get('name', '')
+                # Extract just the model ID (e.g., "models/gemini-pro" -> "gemini-pro")
+                if model_name.startswith('models/'):
+                    model_id = model_name[7:]
+                else:
+                    model_id = model_name
+                
+                # Only include models that support generateContent
+                supported_methods = model.get('supportedGenerationMethods', [])
+                if 'generateContent' in supported_methods:
+                    models.append(model_id)
+            
+            return models
+        except Exception as e:
+            print(f"[GeminiProvider] Failed to list models: {e}")
+            return []
+    
     def generate(self, system_prompt: str, user_message: str,
                  max_tokens: int = 200) -> Optional[str]:
         """Generate a response using Google's Gemini API.