app.py

import os
from groq import Groq
import gradio as gr
from gtts import gTTS
import re

groq_key = os.environ.get("GROQ_API_KEY")
client = Groq(api_key=groq_key)

def generate_story(age, theme, language):
    system_msg = {"role": "system", "content": "You write age-appropriate stories with complete endings and a moral."}
    age_i = int(age)
    length = "300 words" if age_i <= 6 else "600 words" if age_i <= 9 else "1000 words"
    user_msg = {
        "role": "user",
        "content": f"Write a complete story for a {age}-year-old child. Theme: {theme}. Language: {language}. Length about {length}. End with a clear moral of the story."
    }
    response = client.chat.completions.create(
        messages=[system_msg, user_msg],
        model="openai/gpt-oss-20b",
        max_tokens=2000,
        temperature=0.8
    )
    return response.choices[0].message.content.strip()

def story_to_speech(story, language_code='en'):
    tts = gTTS(text=story, lang=language_code)
    filename = "/tmp/story.mp3"
    tts.save(filename)
    return filename

def create_story_interface(age, theme, language, tts_option):
    story = generate_story(age, theme, language)
    story = re.sub(r'[*_#`~]', '', story)
    lines = story.split("\n", 1)
    title, rest_of_story = (lines[0], lines[1]) if len(lines) > 1 else (lines[0], "")
    if language.lower().startswith("u"):
        story_html = f'<div style="text-align:center; font-size:1.3em; direction:rtl; margin-bottom:5px;">{title}</div><div style="font-size:1.2em; direction:rtl; text-align:right;">{rest_of_story}</div>'
        lang_code = "ur"
    else:
        story_html = f'<div style="text-align:center; font-size:1.3em; direction:ltr; margin-bottom:5px;">{title}</div><div style="font-size:1.2em; direction:ltr; text-align:left;">{rest_of_story}</div>'
        lang_code = "en"
    audio_file = story_to_speech(story, lang_code) if tts_option else None
    return story_html, audio_file

age_options = [str(i) for i in range(3, 13)]
theme_options = ["Adventure","Animals","Fantasy","Educational","Friendship","Magic","Science","Mystery","Space","Nature","Kindness","Courage"]
language_options = ["English", "Urdu"]

# -----------------------------------------------
# GRADIO APP
# -----------------------------------------------
with gr.Blocks() as iface:
    gr.HTML("""<style>
    @import url('https://fonts.googleapis.com/css2?family=Comic+Neue:wght@400;700&display=swap');
    .gradio-container { font-family: 'Comic Neue', cursive !important; background: linear-gradient(135deg, #ffafbd, #ffc3a0) !important; padding-top: 10px !important; margin-top: -70px !important;}
    .main-header { text-align: center; font-weight: bold !important; font-size: 3.2em !important; margin-top: 40px !important; margin-bottom: 20px !important; text-shadow: 2px 2px 4px rgba(0,0,0,0.1); color: #12979B;}
    .story-container { background:#fff5ee; border-radius: 20px; padding: 25px; border: 5px solid #FFD166; box-shadow: 0 8px 25px rgba(0,0,0,0.1); min-height: 400px; margin-top: -20px !important;}
    .control-panel { background: #fff5ee; border-radius: 15px; padding: 20px; border: 3px dashed #4ECDC4; font-size: 1.2em !important; font-weight: 600; margin-top: -20px !important;}
    .generate-btn { background: linear-gradient(45deg, #FF6B6B, #FF9E7D) !important; border: none !important; border-radius: 50px !important; padding: 10px 22px !important; font-size: 1em !important; font-weight: bold !important; color: white !important; margin-top: 20px !important;}
    .generate-btn:hover { transform: scale(1.05); box-shadow: 0 5px 15px rgba(255, 107, 107, 0.4);}
    .dropdown, .checkbox { background: white !important; border-radius: 10px !important; border: 2px solid #96CEB4 !important;}
    .audio-player { border-radius: 15px !important; margin-top: 20px;}
    .subtitle { text-align: left; font-size: 1.5em !important; font-weight: 600; color: white; margin-top: -10px !important; margin-bottom: 20px; text-shadow: 1px 1px 2px rgba(0,0,0,0.1);}
    .footer { text-align: center; margin-top: 20px; color: #666; font-style: italic; font-size: 1.5em !important;}
    </style>
    """)

    
    gr.HTML("""<div class="main-header">🌈 StoryTime AI🎪</div>""")
    gr.HTML("""<div class="subtitle">🎯 Create amazing stories that spark imagination and teach valuable lessons!</div>""")
    
    with gr.Row():
        with gr.Column(scale=1, min_width=300):
            with gr.Group(elem_classes="control-panel"):
                age_input = gr.Dropdown(age_options, label="👶 Child's Age", value="5")
                theme_input = gr.Dropdown(theme_options, label="🎭 Story Theme", value="Adventure")
                language_input = gr.Dropdown(language_options, label="🌍 Language", value="English")
                tts_input = gr.Checkbox(label="🎵 Include Audio Story", value=True)
                generate_btn = gr.Button("✨ Create Magical Story! ✨", elem_classes="generate-btn")
        with gr.Column(scale=2):
            with gr.Group(elem_classes="story-container"):
                story_output = gr.Markdown("Your story will appear here!", show_label=False)
            audio_output = gr.Audio(label="Audio Story", elem_classes="audio-player")
    
    gr.HTML("""<div class="footer">Made with ❤️ for young readers everywhere! | Watch your stories come to life! ✨</div>""")
    
    generate_btn.click(
        create_story_interface,
        [age_input, theme_input, language_input, tts_input],
        [story_output, audio_output]
    )

iface.launch(ssr_mode=False)