Update app.py

app.py

@@ -29,15 +29,21 @@ def generate_ideas(tag):
     Generate a list of 5 diverse and creative ideas related to {tag} that can be used for a TikTok video.
     Each idea should be a short sentence describing a specific scene or concept.
     Return the response as a JSON object with a single key 'ideas' containing a list of 5 ideas.
+    Ensure the response is strictly in JSON format.
     Example: {{"ideas": ["A neon-lit gaming setup with RGB lights flashing", "A futuristic robot assembling a gadget"]}}
     """
     try:
         response = client.models.generate_content(
-            model='gemini-2.5-flash-preview-04-17',
+            model='gemini-2.0-flash',  # Updated to a potentially supported model
             contents=[prompt],
             config=types.GenerateContentConfig(temperature=1.2)
         )
+        print(f"Raw response for ideas: {response.text}")  # Debugging
+        if not response.text or response.text.isspace():
+            raise ValueError("Empty response from API")
         response_json = json.loads(response.text.strip())
+        if 'ideas' not in response_json or not isinstance(response_json['ideas'], list):
+            raise ValueError("Invalid JSON format: 'ideas' key missing or not a list")
         return response_json['ideas']
     except Exception as e:
         print(f"Error generating ideas: {e}")
@@ -67,15 +73,21 @@ def generate_item(tag, ideas):
     - 'caption': A short, viral TikTok-style caption with hashtags.
     - 'image_prompt': A detailed image prompt for generating a high-quality visual scene.
     The image prompt should describe the scene vividly, specify a perspective and style, and ensure no text or letters are included.
+    Ensure the response is strictly in JSON format.
     Example: {{"caption": "Neon vibes only! 🌌 #tech", "image_prompt": "A close-up view of a neon-lit gaming setup with RGB lights flashing, in a futuristic style, no text or letters"}}
     """
     try:
         response = client.models.generate_content(
-            model='gemini-2.5-flash-preview-04-17',
+            model='gemini-2.0-flash',  # Updated to a potentially supported model
             contents=[prompt],
             config=types.GenerateContentConfig(temperature=1.2)
         )
+        print(f"Raw response for item: {response.text}")  # Debugging
+        if not response.text or response.text.isspace():
+            raise ValueError("Empty response from API")
         response_json = json.loads(response.text.strip())
+        if 'caption' not in response_json or 'image_prompt' not in response_json:
+            raise ValueError("Invalid JSON format: 'caption' or 'image_prompt' key missing")
         text = response_json['caption']
         image_prompt = response_json['image_prompt']
     except Exception as e: