Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import json
 import re
+import openai
 import pandas as pd
 from datetime import datetime
 from dotenv import load_dotenv
@@ -46,6 +47,17 @@ llm = ChatOpenAI(temperature=0.6, model_name="gpt-3.5-turbo", openai_api_key=ope
 teen_memory = ConversationBufferMemory()
 embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2")
 
+def moderate_output(text):
+    openai.api_key = openai_key  # already loaded from env
+    try:
+        response = openai.Moderation.create(input=text)
+        flagged = response["results"][0]["flagged"]
+        categories = response["results"][0]["categories"]
+        return flagged, categories
+    except Exception as e:
+        print(f"❌ Moderation error: {e}")
+        return False, {}
+
 def load_youtube_links():
     path = os.path.join(os.path.dirname(__file__), "kids_youtube_links.csv")
     if os.path.exists(path):
@@ -157,7 +169,13 @@ def generate_kids(age, theme, tone, name, length):
     messages = kids_prompt.format_messages(age=age, theme=theme, tone=tone, name=name, length=length)
     story = llm(messages).content
     save_story_to_file(name, story, age_group="kids", tone=tone, theme=theme)
-    video_html = get_youtube_link(age, theme)  # from your CSV file
+    video_html = get_youtube_link(age, theme) # from your CSV file
+    flagged, categories = moderate_output(story)
+    if flagged:
+        return "⚠️ Story flagged by safety filters. Please try different inputs.", None
+
+    save_story_to_file(name, story, age_group="kids", tone=tone, theme=theme)
+    video_html = get_youtube_link(age, theme)
     return story, video_html
     
 def generate_kids_audio(story):
@@ -296,6 +314,15 @@ def generate_teen(genre, tone, name, length, uploaded_file=None, edited_text="",
             length=prompt_len
         )
         story = llm(messages).content.strip()
+        # ✅ Check story with moderation filter
+        flagged, categories = moderate_output(story)
+        if flagged:
+            return (
+                "⚠️ This story was flagged by OpenAI's moderation system for safety concerns. "
+                "Please try using a different character name, tone, or inspiration.",
+                uploaded_preview
+            )
+
         last_story_text["story"] = story
         save_story_to_file(name, story, age_group="teen", tone=tone, theme=genre)
         return story, uploaded_preview