Initial

app.py

@@ -1,13 +1,15 @@
 import os
 import gradio as gr
 from openai import OpenAI
+from gtts import gTTS
+import tempfile
 
 client = OpenAI(api_key=os.environ['OPENAI_API_KEY'])
 
 
 def generate_story(password, topic, language):
     if password != os.environ.get("PASSWORD"):
-        return "Incorrect password!"
+        return "Incorrect password!", None
 
     prompt = f"""You are a professional storyteller. Your task is to take any given topic and creatively transform it 
 into an engaging, captivating, and imaginative fairy tale, preserving the core elements and characteristics of the 
@@ -33,19 +35,30 @@ Topic: {topic}
         temperature=0.5
     )
     story = response.choices[0].message.content.strip()
-    return story
+
+    lang_code = {"English": "en", "French": "fr", "Turkish": "tr"}.get(language, "en")
+
+    tts = gTTS(text=story, lang=lang_code)
+
+    audio_file = tempfile.NamedTemporaryFile(delete=False, suffix=".mp3")
+    tts.save(audio_file.name)
+
+    return story, audio_file.name
 
 
 iface = gr.Interface(
     fn=generate_story,
     inputs=[
-        gr.Textbox(label="Story Topic", placeholder="e.g., The adventures of a brave rabbit..."),
         gr.Textbox(label="Password", type="password", placeholder="Enter your password..."),
+        gr.Textbox(label="Story Topic", placeholder="e.g., The adventures of a brave rabbit..."),
         gr.Radio(label="Language", choices=["French", "English", "Turkish"], value="English")
     ],
-    outputs=gr.Textbox(label="Generated Story"),
+    outputs=[
+        gr.Textbox(label="Generated Story"),
+        gr.Audio(label="Listen to the Story", type="filepath")
+    ],
     title="✨ UmaiMother ✨",
-    description="Please enter the correct password, specify the story topic and choose a language."
+    description="Please enter the correct password, specify the story topic, and choose a language."
 )
 
 iface.launch()

requirements.txt

@@ -1,2 +1,3 @@
 gradio
 openai
+gTTS