Update app.py

app.py

@@ -0,0 +1,62 @@
+# app.py
+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})
+
+        # TTS generation
+        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()