Create app.py

app.py

@@ -0,0 +1,122 @@
+# 📌 Step 1: Install required packages
+!pip install --upgrade openai gradio fpdf gTTS
+
+# 📌 Step 2: Import Libraries
+from dotenv import load_dotenv
+import os
+import gradio as gr
+from openai import OpenAI
+from fpdf import FPDF
+from datetime import datetime
+from gtts import gTTS
+
+# 📌 Step 3: Load environment variables from .env file
+load_dotenv()
+api_key = os.getenv("OPENAI_API_KEY")
+client = OpenAI(api_key=api_key)
+
+# 📌 Step 4: Directory to save stories
+story_dir = "/content/storybot_stories"
+os.makedirs(story_dir, exist_ok=True)
+
+# 📌 Step 5: Save story to PDF
+def save_story_as_pdf(story_text, character_name=""):
+    date_str = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+    file_name = f"{character_name or 'story'}_{date_str}.pdf"
+    file_path = os.path.join(story_dir, file_name)
+
+    pdf = FPDF()
+    pdf.add_page()
+    pdf.set_font("Arial", 'B', 16)
+    pdf.cell(200, 10, txt=f"{character_name}'s Adventure", ln=True, align='C')
+    pdf.set_font("Arial", size=12)
+    pdf.ln(10)
+
+    for line in story_text.split('\n'):
+        pdf.multi_cell(0, 10, line)
+
+    pdf.add_page()
+    pdf.set_font("Arial", 'I', 14)
+    pdf.cell(200, 10, txt="The End", ln=True, align='C')
+    pdf.ln(10)
+    pdf.set_font("Arial", size=10)
+    pdf.cell(200, 10, txt="Created with Magic StoryBot", ln=True, align='C')
+
+    pdf.output(file_path)
+    return file_path
+
+# 📌 Step 6: Convert story to speech
+def text_to_speech(text):
+    tts = gTTS(text)
+    audio_path = "/content/story_audio.mp3"
+    tts.save(audio_path)
+    return audio_path
+
+# 📌 Step 7: Generate story from LLM
+def generate_story(age_group, theme, character_name, tone, length):
+    length_instruction = {
+        "Short": "about 100 words",
+        "Medium": "about 200 words",
+        "Long": "about 350 words"
+    }.get(length, "about 150 words")
+
+    system_prompt = f"""
+You are a children's storyteller. Create a {length_instruction}, age-appropriate story for a child aged {age_group}.
+Use the theme "{theme}" and the main character's name is "{character_name}".
+Make the tone of the story {tone.lower()}. End it with the phrase "The End".
+"""
+
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user", "content": "Please generate the story now."}
+    ]
+
+    try:
+        response = client.chat.completions.create(
+            model="gpt-3.5-turbo",
+            messages=messages,
+            temperature=0.7,
+            max_tokens=1024,
+            stop=None
+        )
+        print("LLM Response:", response)  # Debug print
+        story = response.choices[0].message.content
+    except Exception as e:
+        print("LLM ERROR:", e)
+        story = f"An error occurred: {str(e)}"
+
+    audio_path = text_to_speech(story)
+    return story, story, audio_path
+
+# 📌 Step 8: Gradio Interface
+with gr.Blocks() as app:
+    gr.Markdown("## 📖 Magic StoryBot for Kids")
+
+    age_group = gr.Dropdown(["1–3", "3–5", "6–8", "9–12"], label="Select Age Group")
+    theme = gr.Dropdown(["Animals", "Adventure", "Magic", "Mystery", "Bedtime"], label="Choose a Story Theme")
+    character_name = gr.Textbox(label="Main Character's Name", placeholder="e.g., Luna, Max")
+    tone = gr.Dropdown(["Happy", "Funny", "Gentle", "Spooky", "Exciting", "Calm"], label="Story Tone")
+    length = gr.Radio(["Short", "Medium", "Long"], label="Story Length")
+
+    story_output = gr.Textbox(label="Generated Story", lines=10)
+    story_audio = gr.Audio(label="Listen", autoplay=True)
+    pdf_output = gr.File(label="Download Your Story", file_types=[".pdf"], visible=True)
+
+    def start_story(age, theme, name, tone, length):
+        story, _, audio = generate_story(age, theme, name, tone, length)
+        return story, audio
+
+    def export_pdf(story, name):
+        pdf_path = save_story_as_pdf(story, name)
+        return pdf_path
+
+    start_btn = gr.Button("🌟 Generate Story")
+    save_btn = gr.Button("📥 Save Story as PDF")
+
+    start_btn.click(start_story,
+                    inputs=[age_group, theme, character_name, tone, length],
+                    outputs=[story_output, story_audio])
+
+    save_btn.click(export_pdf, inputs=[story_output, character_name], outputs=pdf_output)
+
+app.launch()