Update app.py

app.py

@@ -1,8 +1,102 @@
-# kukubuddy_ai_light.py
+# # 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")
+# tts = TTS(model_name="tts_models/en/ljspeech/tacotron2-DDC", progress_bar=False, gpu=False)
+
+# # ---------- Feature 1: Daily Audio Digest ----------
+# def generate_audio_digest(topic):
+#     dummy_text = f"This is a sample Kuku FM podcast about {topic}. " * 10
+#     summary = summarizer(dummy_text, max_length=300, min_length=30, do_sample=False)[0]["summary_text"]
+    
+#     audio_path = "digest.wav"
+#     tts.tts_to_file(text=summary, file_path=audio_path)
+    
+#     return summary, audio_path
+
+# # ---------- Feature 2: Interactive Story Generator ----------
+# story_cache = {}
+
+# # 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:
+#         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(
+#     fn=generate_audio_digest,
+#     inputs=gr.Textbox(label="Enter your topic of interest", placeholder="e.g. motivation, productivity"),
+#     outputs=[gr.Text(label="Summary"), gr.Audio(label="Audio Digest", type="filepath")],
+#     title="🎧 KukuBuddy: Daily Audio Digest"
+# )
+
+# story_ui = gr.Interface(
+#     fn=generate_story,
+#     inputs=[
+#         gr.Textbox(label="Genre", placeholder="e.g. sci-fi, myth, thriller"),
+#         gr.Textbox(label="Your last choice", placeholder="e.g. Enter the castle")
+#     ],
+#     outputs=[gr.Text(label="Next Scene"), gr.Audio(label="Narration", type="filepath")],
+#     title="📖 KukuBuddy: Interactive Story"
+# )
+
+# app = gr.TabbedInterface([digest_ui, story_ui], tab_names=["📌 Daily Digest", "🧠 Interactive Story"])
+
+# if __name__ == "__main__":
+#     app.launch()
+
 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")
@@ -16,29 +110,11 @@ def generate_audio_digest(topic):
     audio_path = "digest.wav"
     tts.tts_to_file(text=summary, file_path=audio_path)
     
-    return summary, audio_path
+    return summary, audio_path, audio_path  # return twice for play and download
 
 # ---------- Feature 2: Interactive Story Generator ----------
 story_cache = {}
 
-# 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. ",
@@ -82,16 +158,18 @@ def generate_story(genre, choice):
     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, story_audio_path  # return twice for play and download
 
 # ---------- Gradio UI ----------
 
 digest_ui = gr.Interface(
     fn=generate_audio_digest,
     inputs=gr.Textbox(label="Enter your topic of interest", placeholder="e.g. motivation, productivity"),
-    outputs=[gr.Text(label="Summary"), gr.Audio(label="Audio Digest", type="filepath")],
+    outputs=[
+        gr.Text(label="Summary"),
+        gr.Audio(label="🔊 Audio Digest", type="filepath"),
+        gr.File(label="⬇️ Download Audio File")
+    ],
     title="🎧 KukuBuddy: Daily Audio Digest"
 )
 
@@ -101,11 +179,18 @@ story_ui = gr.Interface(
         gr.Textbox(label="Genre", placeholder="e.g. sci-fi, myth, thriller"),
         gr.Textbox(label="Your last choice", placeholder="e.g. Enter the castle")
     ],
-    outputs=[gr.Text(label="Next Scene"), gr.Audio(label="Narration", type="filepath")],
+    outputs=[
+        gr.Text(label="📖 Next Scene"),
+        gr.Audio(label="🔊 Narration", type="filepath"),
+        gr.File(label="⬇️ Download Narration")
+    ],
     title="📖 KukuBuddy: Interactive Story"
 )
 
-app = gr.TabbedInterface([digest_ui, story_ui], tab_names=["📌 Daily Digest", "🧠 Interactive Story"])
+app = gr.TabbedInterface(
+    [digest_ui, story_ui],
+    tab_names=["📌 Daily Digest", "🧠 Interactive Story"]
+)
 
 if __name__ == "__main__":
     app.launch()