Using API instead of pydantic

api_clients.py

@@ -5,7 +5,7 @@ import elevenlabs
 import time
 from contextlib import asynccontextmanager
 from logger import setup_logger, log_execution_time, log_async_execution_time
-from models import OpenRouterRequest, OpenRouterResponse, Message, OpenRouterModel
+from models import OpenRouterModel
 
 logger = setup_logger("api_clients")
 
@@ -18,7 +18,7 @@ class OpenRouterClient:
         self.base_url = "https://openrouter.ai/api/v1"
         self.headers = {
             "Authorization": f"Bearer {api_key}",
-            "HTTP-Referer": "https://github.com/your-repo",  # Required by OpenRouter
+            "HTTP-Referer": "https://localhost:7860",  # Required by OpenRouter
             "X-Title": "URL to Podcast Generator",  # Required by OpenRouter
             "Content-Type": "application/json"
         }
@@ -37,8 +37,8 @@ class OpenRouterClient:
         # Update headers when API key changes
         self.headers = {
             "Authorization": f"Bearer {value}",
-            "HTTP-Referer": "https://github.com/your-repo",  # Required by OpenRouter
-            "X-Title": "URL to Podcast Generator",  # Required by OpenRouter
+            "HTTP-Referer": "https://localhost:7860",
+            "X-Title": "URL to Podcast Generator",
             "Content-Type": "application/json",
         }
         logger.info("OpenRouter API key updated successfully")
@@ -88,25 +88,25 @@ class OpenRouterClient:
             logger.error("Prompt too short or missing")
             raise ValueError("Please provide a more detailed prompt")
         
-        # Combine prompt and content into a clear instruction
-        full_prompt = f"""
-Based on the following content, create an engaging podcast script:
+        system_prompt = "You are a professional podcast script writer. Create engaging, conversational content that sounds natural when read aloud."
+        user_prompt = f"""Write a podcast script based on the following content. Make it engaging and easy to follow.
 
-Context: {prompt if prompt else 'Create a natural, conversational podcast script.'}
+Context: {prompt if prompt else 'Create an informative and engaging podcast episode'}
 
-Content to transform:
+Content:
 {content}
 
-Generate a well-structured podcast script that explains this content in a clear, engaging way.
-"""
+Format the script in a clear, readable way with appropriate spacing and structure."""
 
         try:
             request_data = {
                 "model": model_id,
                 "messages": [
-                    {"role": "system", "content": "You are a professional podcast script writer who creates engaging, conversational content."},
-                    {"role": "user", "content": full_prompt}
-                ]
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": user_prompt}
+                ],
+                "temperature": 0.7,
+                "max_tokens": 2000
             }
             
             async with self.get_session() as session:
@@ -114,9 +114,14 @@ Generate a well-structured podcast script that explains this content in a clear,
                     f"{self.base_url}/chat/completions",
                     json=request_data
                 ) as response:
-                    response.raise_for_status()
+                    if response.status != 200:
+                        error_text = await response.text()
+                        logger.error(f"OpenRouter API error: {error_text}")
+                        raise ValueError(f"API request failed: {error_text}")
+                    
                     data = await response.json()
                     return data['choices'][0]['message']['content']
+                    
         except Exception as e:
             logger.error(f"Script generation failed", exc_info=True)
             raise