Update app.py

app.py

@@ -1,25 +1,23 @@
-# 📌 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
+import openai
 from fpdf import FPDF
 from datetime import datetime
 from gtts import gTTS
 
-# 📌 Step 3: Load environment variables from .env file
+# Load environment variables from .env
 load_dotenv()
 api_key = os.getenv("OPENAI_API_KEY")
-client = OpenAI(api_key=api_key)
+if not api_key:
+    raise ValueError("❌ OPENAI_API_KEY not found in environment.")
+
+openai.api_key = api_key
 
-# 📌 Step 4: Directory to save stories
+# Directory to save stories
 story_dir = "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"
@@ -45,14 +43,12 @@ def save_story_as_pdf(story_text, character_name=""):
     pdf.output(file_path)
     return file_path
 
-# 📌 Step 6: Convert story to speech
 def text_to_speech(text):
+    audio_path = "story_audio.mp3"
     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",
@@ -72,23 +68,21 @@ Make the tone of the story {tone.lower()}. End it with the phrase "The End".
     ]
 
     try:
-        response = client.chat.completions.create(
+        response = openai.ChatCompletion.create(
             model="gpt-3.5-turbo",
             messages=messages,
             temperature=0.7,
-            max_tokens=1024,
-            stop=None
+            max_tokens=1024
         )
-        print("LLM Response:", response)  # Debug print
         story = response.choices[0].message.content
+        print("✅ Story generated")
     except Exception as e:
-        print("LLM ERROR:", e)
-        story = f"An error occurred: {str(e)}"
+        print("❌ LLM ERROR:", e)
+        story = "An error occurred while generating the story."
 
     audio_path = text_to_speech(story)
-    return story, story, audio_path
+    return story, audio_path
 
-# 📌 Step 8: Gradio Interface
 with gr.Blocks() as app:
     gr.Markdown("## 📖 Magic StoryBot for Kids")
 
@@ -103,12 +97,11 @@ with gr.Blocks() as app:
     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)
+        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
+        return save_story_as_pdf(story, name)
 
     start_btn = gr.Button("🌟 Generate Story")
     save_btn = gr.Button("📥 Save Story as PDF")
@@ -117,6 +110,8 @@ with gr.Blocks() as app:
                     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)
+    save_btn.click(export_pdf,
+                   inputs=[story_output, character_name],
+                   outputs=pdf_output)
 
 app.launch()