Update app.py

app.py

@@ -17,6 +17,30 @@ except KeyError:
     raise ValueError("Please set the GEMINI_API_KEY environment variable.")
 client = genai.Client(api_key=api_key)
 
+# Define safety settings to disable all filters
+SAFETY_SETTINGS = [
+    types.SafetySetting(
+        category=types.HarmCategory.HARM_CATEGORY_HARASSMENT,
+        threshold=types.HarmBlockThreshold.BLOCK_NONE,
+    ),
+    types.SafetySetting(
+        category=types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+        threshold=types.HarmBlockThreshold.BLOCK_NONE,
+    ),
+    types.SafetySetting(
+        category=types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+        threshold=types.HarmBlockThreshold.BLOCK_NONE,
+    ),
+    types.SafetySetting(
+        category=types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+        threshold=types.HarmBlockThreshold.BLOCK_NONE,
+    ),
+    types.SafetySetting(
+        category=types.HarmCategory.HARM_CATEGORY_CIVIC_INTEGRITY,
+        threshold=types.HarmBlockThreshold.BLOCK_NONE,
+    ),
+]
+
 def clean_response_text(response_text):
     """
     Clean the API response by removing Markdown code block markers.
@@ -65,7 +89,10 @@ def generate_ideas(user_input):
         response = client.models.generate_content(
             model='gemini-2.0-flash',
             contents=[prompt],
-            config=types.GenerateContentConfig(temperature=1.2)
+            config=types.GenerateContentConfig(
+                temperature=1.2,
+                safety_settings=SAFETY_SETTINGS
+            )
         )
         print(f"Raw response for ideas: {response.text}")  # Debugging
         if not response.text or response.text.isspace():
@@ -91,7 +118,10 @@ def generate_ideas(user_input):
                 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
+                    config=types.GenerateContentConfig(
+                        temperature=0.0,  # Low temperature for deterministic output
+                        safety_settings=SAFETY_SETTINGS
+                    )
                 )
                 print(f"Validation response for idea '{idea}': {validation_response.text}")  # Debugging
                 if not validation_response.text or validation_response.text.isspace():
@@ -169,7 +199,10 @@ def generate_item(user_input, ideas, generate_video=False, max_retries=3):
                 response = client.models.generate_content(
                     model='gemini-2.0-flash',
                     contents=[prompt],
-                    config=types.GenerateContentConfig(temperature=1.2)
+                    config=types.GenerateContentConfig(
+                        temperature=1.2,
+                        safety_settings=SAFETY_SETTINGS
+                    )
                 )
                 print(f"Raw response for item (image attempt {image_attempt + 1}, total attempt {total_attempts}): {response.text}")  # Debugging
                 if not response.text or response.text.isspace():
@@ -192,7 +225,8 @@ def generate_item(user_input, ideas, generate_video=False, max_retries=3):
                     prompt=image_prompt,
                     config=types.GenerateImagesConfig(
                         aspect_ratio="9:16",
-                        number_of_images=1
+                        number_of_images=1,
+                        safety_settings=SAFETY_SETTINGS
                     )
                 )
                 if imagen.generated_images and len(imagen.generated_images) > 0:
@@ -272,7 +306,8 @@ def generate_item(user_input, ideas, generate_video=False, max_retries=3):
                             aspect_ratio="9:16",
                             number_of_videos=1,
                             duration_seconds=8,
-                            negative_prompt="blurry, low quality, text, letters"
+                            negative_prompt="blurry, low quality, text, letters",
+                            safety_settings=SAFETY_SETTINGS
                         )
                     )
                     # Wait for video to generate