| |
| import gradio as gr |
| from groq import Groq |
| from gtts import gTTS |
| import os |
|
|
| client = Groq(api_key=os.environ["GROQ_API_KEY"]) |
|
|
| def initialize_messages(): |
| return [{"role": "system", |
| "content": "You are a talented mythological storyteller. Tell short, engaging Indian mythology stories in a simple, emotional style. End every story with a moral."}] |
|
|
| messages_prmt = initialize_messages() |
|
|
| def generate_story(user_input): |
| global messages_prmt |
|
|
| try: |
| messages_prmt.append({"role": "user", "content": user_input}) |
| response = client.chat.completions.create( |
| messages=messages_prmt, |
| model="llama3-8b-8192", |
| ) |
| story = response.choices[0].message.content |
| messages_prmt.append({"role": "assistant", "content": story}) |
|
|
| |
| audio_path = "story.mp3" |
| try: |
| tts = gTTS(story) |
| tts.save(audio_path) |
| except Exception as e: |
| print("TTS Error:", e) |
| audio_path = None |
|
|
| return story, audio_path |
|
|
| except Exception as e: |
| return f"⚠️ Error: {str(e)}", None |
|
|
| with gr.Blocks(theme=gr.themes.Soft()) as app: |
| gr.Markdown(""" |
| # 🕉️ Mythology Storyteller Bot |
| _Ask for short Indian mythology stories and hear them come alive!_ |
| """) |
| |
| with gr.Row(): |
| with gr.Column(scale=2): |
| user_input = gr.Textbox( |
| label="Your Prompt", |
| placeholder="e.g. Tell me a story about Krishna", |
| lines=2 |
| ) |
| generate_btn = gr.Button("🪄 Generate Story") |
|
|
| with gr.Column(scale=3): |
| story_output = gr.Textbox(label="📜 Story", lines=8, interactive=False) |
| audio_output = gr.Audio(label="🔊 Listen", autoplay=True) |
|
|
| generate_btn.click(fn=generate_story, inputs=user_input, outputs=[story_output, audio_output]) |
|
|
| app.launch() |
|
|
