Update to google-genai SDK for Gemini API

modules/story_reels/services/script_generator.py

@@ -1,25 +1,28 @@
 """
-Script Generator using Gemini API
+Script Generator using Google GenAI SDK
 Generates story scripts from topics for TTS narration
 """
 import logging
-import requests
+import json
 from typing import Optional
 
+from google import genai
+
 logger = logging.getLogger(__name__)
 
 
 class ScriptGenerator:
     """
-    Generates story scripts using Google Gemini API.
+    Generates story scripts using Google Gemini API via google-genai SDK.
     
     Features:
     - Topic → Full narration script (<=1000 chars)
     - Character-aware script generation
+    - AI image prompt generation
     - Optimized for TTS output
     """
     
-    GEMINI_API_URL = "https://generativelanguage.googleapis.com/v1beta/models/gemini-2.0-flash:generateContent"
+    MODEL = "gemini-2.0-flash"
     
     # System prompt for script generation
     SYSTEM_PROMPT = """You are a professional script writer for short-form video content (TikTok, Reels, Shorts).
@@ -37,6 +40,7 @@ If a character is provided, write the story from their perspective or about them
 
     def __init__(self, api_key: str):
         self.api_key = api_key
+        self.client = genai.Client(api_key=api_key)
         
     def generate_script(
         self,
@@ -66,30 +70,12 @@ If a character is provided, write the story from their perspective or about them
         logger.info(f"Generating script for topic: {topic[:50]}...")
         
         try:
-            response = requests.post(
-                f"{self.GEMINI_API_URL}?key={self.api_key}",
-                headers={"Content-Type": "application/json"},
-                json={
-                    "contents": [
-                        {
-                            "role": "user",
-                            "parts": [{"text": self.SYSTEM_PROMPT + "\n\n" + user_prompt}]
-                        }
-                    ],
-                    "generationConfig": {
-                        "temperature": 0.7,
-                        "maxOutputTokens": 500,
-                        "topP": 0.9
-                    }
-                },
-                timeout=30
+            response = self.client.models.generate_content(
+                model=self.MODEL,
+                contents=self.SYSTEM_PROMPT + "\n\n" + user_prompt
             )
-            response.raise_for_status()
-            
-            data = response.json()
             
-            # Extract text from response
-            script = data["candidates"][0]["content"]["parts"][0]["text"]
+            script = response.text
             
             # Enforce character limit
             if len(script) > max_chars:
@@ -98,12 +84,9 @@ If a character is provided, write the story from their perspective or about them
             logger.info(f"Generated script: {len(script)} chars")
             return script.strip()
             
-        except requests.exceptions.RequestException as e:
+        except Exception as e:
             logger.error(f"Gemini API error: {e}")
             raise Exception(f"Script generation failed: {e}")
-        except (KeyError, IndexError) as e:
-            logger.error(f"Failed to parse Gemini response: {e}")
-            raise Exception("Invalid response from Gemini API")
     
     @staticmethod
     def test_connection(api_key: str) -> bool:
@@ -160,8 +143,6 @@ Return ONLY valid JSON array, no markdown, no explanation:
         Returns:
             List of {chunk_id, prompt} dicts
         """
-        import json
-        
         all_prompts = []
         total_chunks = len(chunks)
         
@@ -198,28 +179,12 @@ IMPORTANT: Include this character description in EVERY prompt!
             user_prompt += "\nGenerate detailed image prompts for each chunk. Return ONLY JSON array."
             
             try:
-                response = requests.post(
-                    f"{self.GEMINI_API_URL}?key={self.api_key}",
-                    headers={"Content-Type": "application/json"},
-                    json={
-                        "contents": [
-                            {
-                                "role": "user",
-                                "parts": [{"text": self.IMAGE_PROMPT_SYSTEM + "\n\n" + user_prompt}]
-                            }
-                        ],
-                        "generationConfig": {
-                            "temperature": 0.7,
-                            "maxOutputTokens": 4000,
-                            "topP": 0.9
-                        }
-                    },
-                    timeout=60
+                response = self.client.models.generate_content(
+                    model=self.MODEL,
+                    contents=self.IMAGE_PROMPT_SYSTEM + "\n\n" + user_prompt
                 )
-                response.raise_for_status()
                 
-                data = response.json()
-                text = data["candidates"][0]["content"]["parts"][0]["text"]
+                text = response.text
                 
                 # Clean response - remove markdown if present
                 text = text.strip()

requirements.txt

@@ -18,6 +18,7 @@ numpy<2.0.0
 
 # AI/ML
 faster-whisper
+google-genai
 
 # Utilities
 python-multipart