Update app.py

app.py

@@ -37,6 +37,7 @@ def clean_response_text(response_text):
 def generate_ideas(user_input):
     """
     Generate a diverse set of ideas based on the user's input concept using the LLM.
+    Validate the relevance of each idea using a cheaper LLM (gemini-2.0-flash-lite).
     
     Args:
         user_input (str): The user's input concept or idea (e.g., "blindfolded Rubik's Cube challenge").
@@ -44,6 +45,7 @@ def generate_ideas(user_input):
     Returns:
         list: A list of ideas as strings.
     """
+    # Step 1: Generate ideas using gemini-2.0-flash
     prompt = f"""
     The user has provided the concept: "{user_input}". You must generate 5 diverse and creative ideas for a TikTok video that are directly and explicitly related to "{user_input}".
     Each idea must clearly incorporate and focus on the core theme of "{user_input}" without deviating into unrelated topics.
@@ -73,11 +75,43 @@ def generate_ideas(user_input):
         if 'ideas' not in response_json or not isinstance(response_json['ideas'], list) or len(response_json['ideas']) != 5:
             raise ValueError("Invalid JSON format: 'ideas' key missing, not a list, or incorrect length")
         
-        # Validate that ideas are related to user_input
         ideas = response_json['ideas']
+        
+        # Step 2: Validate relevance of each idea using gemini-2.0-flash-lite
         for idea in ideas:
-            if user_input.lower() not in idea.lower():
-                print(f"Warning: Idea '{idea}' does not seem related to '{user_input}'. Falling back to default ideas.")
+            validation_prompt = f"""
+            Determine if the following idea for a TikTok video is related to the user's concept.
+            User's concept: "{user_input}"
+            Idea: "{idea}"
+            Respond with a JSON object containing a single key 'is_related' with a boolean value (true or false).
+            Example:
+            {{"is_related": true}}
+            """
+            try:
+                validation_response = client.models.generate_content(
+                    model='gemini-2.0-flash-lite',
+                    contents=[validation_prompt],
+                    config=types.GenerateContentConfig(temperature=0.0)  # Low temperature for deterministic output
+                )
+                print(f"Validation response for idea '{idea}': {validation_response.text}")  # Debugging
+                if not validation_response.text or validation_response.text.isspace():
+                    raise ValueError("Empty validation response from API")
+                cleaned_validation_text = clean_response_text(validation_response.text)
+                validation_json = json.loads(cleaned_validation_text)
+                if 'is_related' not in validation_json or not isinstance(validation_json['is_related'], bool):
+                    raise ValueError("Invalid validation JSON format: 'is_related' key missing or not a boolean")
+                
+                if not validation_json['is_related']:
+                    print(f"Idea '{idea}' is not related to '{user_input}'. Falling back to default ideas.")
+                    return [
+                        f"A dramatic {user_input} scene with cinematic lighting",
+                        f"A close-up of {user_input} in a futuristic setting",
+                        f"A high-energy {user_input} moment with vibrant colors",
+                        f"A serene {user_input} scene with soft focus",
+                        f"An action-packed {user_input} challenge with dynamic angles"
+                    ]
+            except Exception as e:
+                print(f"Error validating idea '{idea}': {e}. Falling back to default ideas.")
                 return [
                     f"A dramatic {user_input} scene with cinematic lighting",
                     f"A close-up of {user_input} in a futuristic setting",
@@ -85,7 +119,10 @@ def generate_ideas(user_input):
                     f"A serene {user_input} scene with soft focus",
                     f"An action-packed {user_input} challenge with dynamic angles"
                 ]
+        
+        # All ideas are related, return them
         return ideas
+    
     except Exception as e:
         print(f"Error generating ideas: {e}")
         return [
@@ -824,7 +861,7 @@ with gr.Blocks(
     # Handle magic button click to generate next item
     magic_button.click(
         fn=load_next,
-        inputs=[user_input, generate_video_checkbox, current_index, feed_items],  # Use user_input directly from textbox
+        inputs=[user_input, generate_video_checkbox, current_index, feed_items],
         outputs=[current_user_input, current_index, feed_items, feed_html, share_html, is_loading]
     )