Update main.py

main.py

@@ -645,7 +645,6 @@ def download_story_archive(story_id):
             logging.warning(f"❌ Unauthorized access attempt for story {story_id}")
             return jsonify({'error': 'Unauthorized'}), 403
 
-        # Extract story details
         full_text = story_record.get("full_story", "")
         sections = story_record.get("sections", [])
         split_sentences = full_text.split('.', 1)
@@ -656,23 +655,20 @@ def download_story_archive(story_id):
         zip_buffer = io.BytesIO()
         with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zipf:
 
-            # ======================
-            # 1) Generate PDF
-            # ======================
+            # Generate PDF
             logging.info(f"📄 Generating PDF for {story_id}")
             pdf = FPDF()
             pdf.set_auto_page_break(auto=True, margin=15)
             pdf.add_page()
             pdf.set_font("Arial", size=12)
 
-            # Add story details
             pdf.multi_cell(0, 10, f"Story ID: {story_id}")
             pdf.multi_cell(0, 10, f"Title: {title}")
             pdf.ln(10)
             pdf.multi_cell(0, 10, "Full Story:")
             pdf.multi_cell(0, 10, full_text)
 
-            # Add sections with images
+            # Add sections
             bucket = storage.bucket()
             for idx, section_obj in enumerate(sections):
                 section_text = section_obj.get("section_text", "")
@@ -710,9 +706,7 @@ def download_story_archive(story_id):
             zipf.writestr(f"{story_id}.pdf", pdf_buffer.read())
             logging.info(f"✅ PDF added to ZIP for {story_id}")
 
-            # ======================
-            # 2) Generate a TXT file
-            # ======================
+            # Generate TXT
             logging.info(f"📄 Generating TXT for {story_id}")
             txt_content = f"Story ID: {story_id}\nTitle: {title}\n\nFull Story:\n\n{full_text}\n\n"
             for idx, section_obj in enumerate(sections):
@@ -721,9 +715,7 @@ def download_story_archive(story_id):
             zipf.writestr(f"{story_id}.txt", txt_content)
             logging.info(f"✅ TXT added to ZIP for {story_id}")
 
-            # ======================
-            # 3) Download section images
-            # ======================
+            # Download section images
             for idx, section_obj in enumerate(sections):
                 image_url = section_obj.get("image_url", "")
                 if image_url:
@@ -739,9 +731,7 @@ def download_story_archive(story_id):
                     except Exception as img_error:
                         logging.error(f"❌ Error downloading image for section {idx + 1}: {str(img_error)}")
 
-            # ======================
-            # 4) Download section audios
-            # ======================
+            # Download section audios
             for idx, section_obj in enumerate(sections):
                 audio_url = section_obj.get("audio_url", "")
                 if audio_url:
@@ -757,9 +747,7 @@ def download_story_archive(story_id):
                     except Exception as audio_error:
                         logging.error(f"❌ Error downloading audio for section {idx + 1}: {str(audio_error)}")
 
-        # ------------------------
         # Upload log to Firebase
-        # ------------------------
         log_url = upload_log()
         if log_url:
             logging.info(f"📤 Log uploaded: {log_url}")