Update app.py

app.py

@@ -9,27 +9,43 @@ import requests
 
 # Configuration
 OPENROUTER_API_KEY = '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'
-OPENROUTER_MODEL = "google/gemini-2.0-flash-exp:free"
+OPENROUTER_MODEL = "tngtech/deepseek-r1t-chimera:free"
 TARGET_RESOLUTION = (1080, 1920)
 OUTPUT_VIDEO_FILENAME = "final_video.mp4"
 CAPTION_COLOR = "white"
+YOUR_SITE_URL = "http://localhost"  # Replace with your site URL
+YOUR_SITE_NAME = "YouTube Short Creator"  # Replace with your site name
 
 # Placeholder for Kokoro TTS
 def kokoro_tts(text):
     # TODO: Replace with actual Kokoro TTS implementation
     # Should return path to generated audio file
     return "dummy_audio.wav"
+
 def generate_script(topic):
     try:
-        prompt = f"Generate a script about {topic} divided into parts, and output it as a JSON array of strings."
         response = requests.post(
-            "https://api.openrouter.com/v1/completions",
-            headers={"Authorization": f"Bearer {OPENROUTER_API_KEY}"},
-            json={"model": OPENROUTER_MODEL, "prompt": prompt},
-            timeout=10  # Add timeout to avoid hanging
+            url="https://openrouter.ai/api/v1/chat/completions",
+            headers={
+                "Authorization": f"Bearer {OPENROUTER_API_KEY}",
+                "Content-Type": "application/json",
+                "HTTP-Referer": YOUR_SITE_URL,
+                "X-Title": YOUR_SITE_NAME,
+            },
+            data=json.dumps({
+                "model": OPENROUTER_MODEL,
+                "messages": [
+                    {
+                        "role": "user",
+                        "content": f"Generate a script about {topic} divided into parts, and output it as a JSON array of strings."
+                    }
+                ],
+            }),
+            timeout=10
         )
-        response.raise_for_status()  # Raise an exception for HTTP errors
-        script_json = response.json()["choices"][0]["text"]
+        response.raise_for_status()
+        response_data = response.json()
+        script_json = response_data["choices"][0]["message"]["content"]
         return json.loads(script_json)
     except requests.exceptions.ConnectionError as e:
         raise Exception(f"Failed to connect to OpenRouter API: {str(e)}. Please check your internet connection or DNS settings.")
@@ -37,8 +53,8 @@ def generate_script(topic):
         raise Exception(f"OpenRouter API returned an error: {str(e)}. Please verify your API key and model.")
     except requests.exceptions.RequestException as e:
         raise Exception(f"An error occurred while contacting OpenRouter API: {str(e)}")
-    except json.JSONDecodeError:
-        raise Exception("Failed to parse API response as JSON. Please check the API response format.")
+    except (json.JSONDecodeError, KeyError):
+        raise Exception("Failed to parse API response as JSON or unexpected response format.")
 
 def generate_audio(script_parts, temp_folder):
     full_audio = AudioSegment.empty()