Update app.py

app.py

@@ -22,53 +22,22 @@ 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)
-    
-#     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
+    tts.tts_to_file(text=generated, file_path=story_audio_path)
+    
+    return generated, story_audio_path
 
 
 # ---------- Gradio UI ----------