Made OpenRouter more robust

api_clients.py

@@ -13,6 +13,11 @@ class OpenRouterClient:
     def __init__(self, api_key: str):
         logger.info("Initializing OpenRouter client")
         self.api_key = api_key
+        self.base_url = "https://openrouter.ai/api/v1"
+        self.headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json"
+        }
         logger.debug("OpenRouter client initialized successfully")
     
     @property
@@ -50,18 +55,14 @@ class OpenRouterClient:
             ValueError: If API request fails
         """
         logger.info("Fetching available models from OpenRouter")
-        async with self.get_session() as session:
-            async with session.get(f"{self.base_url}/models") as response:
-                if response.status != 200:
-                    error_msg = await response.text()
-                    logger.error(f"Failed to fetch models: {error_msg}")
-                    raise ValueError(f"Failed to fetch models: {error_msg}")
-                
-                models = await response.json()
-                logger.info(f"Successfully fetched {len(models)} models")
-                logger.debug(f"Available models: {[model['name'] for model in models]}")
-                return [(model['id'], f"{model['name']} ({model['context_length']} tokens)") 
-                        for model in models]
+        response = requests.get(
+            f"{self.base_url}/models",
+            headers=self.headers
+        )
+        response.raise_for_status()
+        models = response.json()["data"]
+        # Return tuple of (id, display_name) for each model
+        return [(model["id"], model["name"]) for model in models]
 
     @log_async_execution_time(logger)
     async def generate_script(self, content: str, prompt: str, model_id: str) -> str:
@@ -83,48 +84,23 @@ class OpenRouterClient:
             raise ValueError("Please provide a more detailed prompt")
         
         try:
-            async with self.get_session() as session:
-                logger.debug("Preparing script generation request")
-                response = await self._make_script_request(session, content, prompt, model_id)
-                
-                script = response['choices'][0]['message']['content']
-                logger.info(f"Script generated successfully: {len(script)} chars")
-                logger.debug(f"Script preview: {script[:200]}...")
-                
-                return script
+            response = requests.post(
+                f"{self.base_url}/chat/completions",
+                headers=self.headers,
+                json={
+                    "model": model_id,
+                    "messages": [
+                        {"role": "system", "content": "You are a podcast script writer."},
+                        {"role": "user", "content": f"Create a podcast script from this content: {content}"}
+                    ]
+                }
+            )
+            response.raise_for_status()
+            return response.json()["choices"][0]["message"]["content"]
         except Exception as e:
             logger.error(f"Script generation failed", exc_info=True)
             raise
 
-    async def _make_script_request(self, session, content, prompt, model_id):
-        async with session.post(
-            f"{self.base_url}/chat/completions",
-            json={
-                "model": model_id,
-                "messages": [
-                    {
-                        "role": "system",
-                        "content": "You are an expert podcast script writer. Create engaging, conversational content."
-                    },
-                    {
-                        "role": "user",
-                        "content": f"""Based on this content: {content}
-                        Create a 3-minute podcast script focusing on: {prompt}
-                        Format as a natural conversation with clear speaker parts.
-                        Include [HOST] and [GUEST] markers for different voices."""
-                    }
-                ]
-            }
-        ) as response:
-            logger.debug("Sending script generation request")
-            
-            if response.status != 200:
-                error_msg = await response.text()
-                logger.error(f"Script generation failed: {error_msg}")
-                raise ValueError(f"Script generation failed: {error_msg}")
-            
-            return await response.json()
-
 class ElevenLabsClient:
     """Handles ElevenLabs API interactions with detailed performance tracking"""
     

app.py

@@ -89,11 +89,9 @@ def create_ui():
             try:
                 client = OpenRouterClient(key)
                 models = await client.get_models()
-                return gr.Dropdown(
-                    choices={name: id for id, name in models}
-                )
+                return gr.Dropdown(choices=models)
             except Exception as e:
-                return gr.Dropdown(choices={"Error": None})
+                return gr.Dropdown(choices=[("error", f"Error: {str(e)}")])
 
         def update_elevenlabs_voices(key):
             if not key: