Update app.py

app.py

@@ -1,9 +1,8 @@
 # kukubuddy_ai_light.py
-
+import random
 import gradio as gr
 from transformers import pipeline
 from TTS.api import TTS
-
 # Small & memory-efficient models
 summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 story_gen = pipeline("text2text-generation", model="google/flan-t5-small")
@@ -22,24 +21,71 @@ def generate_audio_digest(topic):
 # ---------- Feature 2: Interactive Story Generator ----------
 story_cache = {}
 
-def generate_story(genre, choice):
-    base_prompt = f"Start a short {genre} story. "
+# def generate_story(genre, choice):
+#     base_prompt = f"Start a short {genre} story. "
+    
+#     if genre not in story_cache:
+#         story_cache[genre] = base_prompt
+    
+#     if choice:
+#         story_cache[genre] += f"User chose: {choice}. Then, "
+    
+#     generated = story_gen(story_cache[genre])[0]['generated_text']
+#     story_cache[genre] = generated
     
+#     story_audio_path = "story.wav"
+#     tts.tts_to_file(text=generated, file_path=story_audio_path)
+    
+#     return generated, story_audio_path
+
+
+# More vivid genre-specific prompts
+genre_templates = {
+    "fantasy": "In a land of dragons and forgotten magic, a tale begins. ",
+    "sci-fi": "In a distant galaxy, governed by AI and cosmic laws, a new story unfolds. ",
+    "thriller": "It started on a stormy night, when a single knock on the door changed everything. ",
+    "romance": "They locked eyes for the first time in a quiet bookstore. ",
+    "horror": "The clock struck midnight, and something moved in the shadows. ",
+    "myth": "Long before time was time, gods and mortals shared the same sky. ",
+    "comedy": "There was once a chicken who wanted to cross a space highway... "
+}
+
+# Random creative twist options
+twist_prompts = [
+    "But little did they know, everything was about to change.",
+    "Suddenly, a mysterious voice echoed in the air.",
+    "Out of nowhere, a glowing symbol appeared.",
+    "But fate had other plans.",
+    "And just like that, the rules of the world shifted."
+]
+
+def generate_story(genre, choice):
+    genre = genre.lower()
+    base_prompt = genre_templates.get(genre, f"Begin a story in the genre: {genre}. ")
+
+    # If it's a new story
     if genre not in story_cache:
         story_cache[genre] = base_prompt
-    
+
+    # Add user decision and twist
     if choice:
-        story_cache[genre] += f"User chose: {choice}. Then, "
-    
+        twist = random.choice(twist_prompts)
+        story_cache[genre] += f"\nThe user chose: '{choice}'. {twist} "
+
+    # Generate new content
     generated = story_gen(story_cache[genre])[0]['generated_text']
+
+    # Update cache for continuity
     story_cache[genre] = generated
-    
+
+    # Audio output
     story_audio_path = "story.wav"
     tts.tts_to_file(text=generated, file_path=story_audio_path)
-    
+
     return generated, story_audio_path
 
 
+
 # ---------- Gradio UI ----------
 
 digest_ui = gr.Interface(