Update app.py

app.py

@@ -1,21 +1,19 @@
-# kukubuddy_ai.py
+# kukubuddy_ai_light.py
 
 import gradio as gr
 from transformers import pipeline
 from TTS.api import TTS
 
-# Load Hugging Face pipelines
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-story_gen = pipeline("text-generation", model="tiiuae/falcon-7b-instruct", max_length=300)
+# 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):
-    # Simulate a long dummy content (replace with real data from KukuFM backend)
-    dummy_text = f"This is a sample podcast on {topic}. " * 20
-    summary = summarizer(dummy_text, max_length=120, min_length=30, do_sample=False)[0]["summary_text"]
+    dummy_text = f"This is a sample Kuku FM podcast about {topic}. " * 10
+    summary = summarizer(dummy_text, max_length=100, min_length=30, do_sample=False)[0]["summary_text"]
     
-    # Convert summary to speech
     audio_path = "digest.wav"
     tts.tts_to_file(text=summary, file_path=audio_path)
     
@@ -25,56 +23,42 @@ def generate_audio_digest(topic):
 story_cache = {}
 
 def generate_story(genre, choice):
-    base_prompt = f"Start a {genre} story. "
+    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"\nUser chose: {choice}. Then, "
+        story_cache[genre] += f"User chose: {choice}. Then, "
     
     generated = story_gen(story_cache[genre])[0]['generated_text']
-    story_cache[genre] = generated  # Save for next turn
+    story_cache[genre] = generated
     
-    # Convert story to audio
     story_audio_path = "story.wav"
     tts.tts_to_file(text=generated, file_path=story_audio_path)
     
     return generated, story_audio_path
 
-# ---------- Gradio Interface ----------
+# ---------- Gradio UI ----------
 
-# Daily Digest UI
 digest_ui = gr.Interface(
     fn=generate_audio_digest,
-    inputs=gr.Textbox(label="Enter your topic of interest", placeholder="e.g. motivation, startups, mental health"),
-    outputs=[
-        gr.Text(label="AI-Generated Summary"),
-        gr.Audio(label="Listen to Your Digest", type="filepath")
-    ],
-    title="🎧 KukuBuddy: Personalized Daily 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 Generator UI
 story_ui = gr.Interface(
     fn=generate_story,
     inputs=[
-        gr.Textbox(label="Choose a genre", placeholder="e.g. sci-fi, romance, horror"),
-        gr.Textbox(label="Your last choice (optional)", placeholder="e.g. Enter the cave")
-    ],
-    outputs=[
-        gr.Text(label="Next part of the story"),
-        gr.Audio(label="Narration", type="filepath")
+        gr.Textbox(label="Genre", placeholder="e.g. sci-fi, myth, thriller"),
+        gr.Textbox(label="Your last choice", placeholder="e.g. Enter the castle")
     ],
-    title="📖 KukuBuddy: Interactive Audio Story Generator"
+    outputs=[gr.Text(label="Next Scene"), gr.Audio(label="Narration", type="filepath")],
+    title="📖 KukuBuddy: Interactive Story"
 )
 
-# Tabbed app
-app = gr.TabbedInterface(
-    interface_list=[digest_ui, story_ui],
-    tab_names=["📌 Daily Audio Digest", "🧠 Interactive Story"]
-)
+app = gr.TabbedInterface([digest_ui, story_ui], tab_names=["📌 Daily Digest", "🧠 Interactive Story"])
 
-# Run the app
 if __name__ == "__main__":
     app.launch()