Update main.py

main.py

@@ -247,11 +247,9 @@ def generate_story_endpoint():
             return jsonify({'error': 'Invalid or expired token'}), 401
 
         # --- Read Request Data ---
-        # If the user is uploading a file (PDF or CSV/Excel), we can read from request.files
-        # If the user is sending JSON only, we read request.get_json()
         data = request.form.to_dict()  # For multipart/form-data fields
         input_type = data.get('input_type', 'text')  # "text", "pdf", "wiki", "bible", "youtube", "dataframe"
-        prompt = data.get('prompt')  # For "text" or fallback
+        prompt = data.get('prompt')  # For "text" only
         story_type = data.get('story_type', 'free_form')
         style = data.get('style', 'whimsical')
         voice_model = data.get('voice_model', 'aura-asteria-en')
@@ -262,7 +260,6 @@ def generate_story_endpoint():
         if input_type not in ["text", "pdf", "wiki", "bible", "youtube", "dataframe"]:
             return jsonify({'error': 'Unsupported input_type'}), 400
 
-        # 1) Generate the full story text
         from stories import generate_story_from_text
         from stories import get_pdf_text
         from stories import get_df
@@ -271,22 +268,20 @@ def generate_story_endpoint():
         full_story = None
 
         if input_type == "text":
+            # <-- CHANGE HERE
+            # We only check 'prompt' if input_type == "text"
             if not prompt:
                 return jsonify({'error': 'Prompt is required for text input'}), 400
             full_story = generate_story_from_text(prompt, story_type)
 
         elif input_type == "pdf":
-            # Expecting a file in request.files["file"]
             uploaded_file = request.files.get("file")
             if not uploaded_file:
                 return jsonify({'error': 'No PDF file uploaded'}), 400
-
-            # Convert PDF to text
             pdf_text = get_pdf_text(uploaded_file)
             full_story = generate_story_from_text(pdf_text, story_type)
 
         elif input_type == "dataframe":
-            # Expecting a file in request.files["file"] and an "ext" field (csv, xlsx, xls)
             uploaded_file = request.files.get("file")
             ext = data.get("ext")  # e.g. "csv", "xlsx", "xls"
             if not uploaded_file or not ext:
@@ -360,7 +355,6 @@ def generate_story_endpoint():
             os.remove(image_filename)
 
             # Generate audio from section text WITHOUT <image> description
-            # e.g. remove <...> from text
             audio_text = re.sub(r"<.*?>", "", section_text)  # remove anything in angle brackets
             audio_start = time.time()
             audio_file_path = generate_audio(audio_text, voice_model, audio_model=audio_model)