Update app.py

app.py

@@ -22,22 +22,54 @@ 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 genre not in story_cache:
+#         story_cache[genre] = base_prompt
     
-    if choice:
-        story_cache[genre] += f"User chose: {choice}. Then, "
+#     if choice:
+#         story_cache[genre] += f"User chose: {choice}. Then, "
     
-    generated = story_gen(story_cache[genre])[0]['generated_text']
-    story_cache[genre] = generated
+#     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)
+#     story_audio_path = "story.wav"
+#     tts.tts_to_file(text=generated, file_path=story_audio_path)
     
-    return generated, story_audio_path
+#     return generated, story_audio_path
+
+def generate_story(genre, scene_input):
+    # Define genre-specific styles
+    genre_styles = {
+        "action": "Include fast-paced sequences, intense confrontations, and physical conflict. Add thrilling suspense and end with a dramatic cliffhanger.",
+        "romance": "Highlight emotional tension, subtle gestures, and deep connection between characters. Use heartfelt dialogue and end with a tender or surprising moment.",
+        "comedy": "Use witty dialogue, humorous misunderstandings, and light-hearted narration. Keep the tone playful and funny.",
+        "thriller": "Add psychological tension, mysterious clues, and an unsettling atmosphere. End with a chilling or puzzling moment.",
+        "fantasy": "Include magical elements, otherworldly creatures, and vivid scenery. Use epic tone and mystical storytelling.",
+        "horror": "Use eerie descriptions, building dread, and jump-scare moments. End on a terrifying note.",
+        "sci-fi": "Incorporate futuristic tech, space travel, or AI. Build an immersive setting and challenge the reader’s imagination.",
+        "drama": "Emphasize emotional depth, personal conflict, and realistic dialogue. Add introspective moments.",
+    }
+
+    # Fallback style if genre not found
+    style = genre_styles.get(genre.lower(), "Write in an immersive and engaging tone relevant to the genre.")
+
+    # Build the prompt
+    prompt = (
+        f"[INST] You are a brilliant storyteller. Write a vivid and immersive scene in a {genre} story. "
+        f"The scene starts like this: {scene_input}. {style} [/INST]"
+    )
+
+    # Generate the story using your pipeline (story_gen)
+    response = story_gen(prompt, max_length=300, do_sample=True, temperature=0.8, top_p=0.95)[0]['generated_text']
+
+    # Convert to speech
+    story_audio_path = "story.wav"
+    tts.tts_to_file(text=response, file_path=story_audio_path)
+
+    return response, story_audio_path
+
 
 # ---------- Gradio UI ----------