Spaces:
Sleeping
Sleeping
| # 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 = {} | |
| # 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: | |
| 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() | |