|
|
|
|
|
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() |
|
|
|
