Update main.py

main.py

@@ -570,7 +570,7 @@ def view_projects():
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
-
+# view videos endpoint
 @app.route('/api/view/videos', methods=['GET'])
 def view_videos():
     """
@@ -613,8 +613,8 @@ def view_videos():
         return jsonify({'error': str(e)}), 500
 
 
+#download archives endpoint
 
-@app.route('/api/download/story_archive/<story_id>', methods=['GET'])
 def download_story_archive(story_id):
     try:
         # --- Authentication ---
@@ -629,75 +629,118 @@ def download_story_archive(story_id):
         # --- Fetch the Story ---
         stories_ref = db.reference('stories')
         story_record = stories_ref.child(story_id).get()
-
         if not story_record:
             return jsonify({'error': 'Story not found'}), 404
         if story_record.get('uid') != uid:
             return jsonify({'error': 'Unauthorized'}), 403
 
-        # --- Create ZIP File ---
+        # Prepare data
+        full_text = story_record.get("full_story", "")
+        sections = story_record.get("sections", [])
+        
+        # Derive a title from the first sentence
+        split_sentences = full_text.split('.', 1)
+        title = split_sentences[0].strip() if split_sentences and split_sentences[0].strip() else "Untitled"
+
+        # ------------------------------------------------------------------
+        # Create an in-memory ZIP
+        # ------------------------------------------------------------------
         zip_buffer = io.BytesIO()
         with zipfile.ZipFile(zip_buffer, "w", zipfile.ZIP_DEFLATED) as zipf:
 
-            # --- Generate a PDF with Story Content ---
+            # ======================
+            # 1) Generate PDF
+            # ======================
             pdf = FPDF()
             pdf.set_auto_page_break(auto=True, margin=15)
             pdf.add_page()
             pdf.set_font("Arial", size=12)
-            pdf.multi_cell(0, 10, f"Story ID: {story_id}\n\n")
-            pdf.multi_cell(0, 10, f"Title: {story_record.get('title', 'Untitled')}\n\n")
-            pdf.multi_cell(0, 10, f"Full Story:\n\n{story_record.get('full_story', '')}\n\n")
 
-            # Add sections if available
-            sections = story_record.get("sections", [])
-            for idx, section in enumerate(sections):
+            # Basic info
+            pdf.multi_cell(0, 10, f"Story ID: {story_id}\n")
+            pdf.multi_cell(0, 10, f"Title: {title}\n")
+            pdf.multi_cell(0, 10, f"Full Story:\n\n{full_text}\n\n")
+
+            # For each section, create a new page
+            for idx, section_obj in enumerate(sections):
+                # Each section is a dict with "section_text" and "image_url"
+                section_text = section_obj.get("section_text", "")
+                image_url = section_obj.get("image_url", "")
+
                 pdf.add_page()
                 pdf.set_font("Arial", style='B', size=14)
-                pdf.multi_cell(0, 10, f"Section {idx + 1}\n")
+                pdf.multi_cell(0, 10, f"Section {idx + 1}")
                 pdf.set_font("Arial", size=12)
-                pdf.multi_cell(0, 10, section)
+                pdf.multi_cell(0, 10, section_text)
+
+                # If there's an image URL, download & embed in PDF
+                if image_url and "firebasestorage.googleapis.com" in image_url:
+                    bucket = storage.bucket()
+                    blob_path = image_url.split('/o/')[-1].split('?')[0]
+                    blob = bucket.blob(blob_path)
+                    image_data = blob.download_as_bytes()
+
+                    # Save the image to a temp file so FPDF can read it
+                    temp_img_path = f"/tmp/{uuid.uuid4().hex}.jpg"
+                    with open(temp_img_path, "wb") as f:
+                        f.write(image_data)
+
+                    # Insert the image at (x=10, y=None => next line), width=100
+                    pdf.image(temp_img_path, x=10, y=None, w=100)
+                    # Clean up temp file
+                    os.remove(temp_img_path)
 
-            # Save PDF to buffer and add to ZIP
+            # Save PDF to memory
             pdf_buffer = io.BytesIO()
             pdf.output(pdf_buffer)
             pdf_buffer.seek(0)
+            # Add PDF to the ZIP
             zipf.writestr(f"{story_id}.pdf", pdf_buffer.read())
 
-            # --- Generate a TXT file with Story Content ---
-            txt_content = f"Story ID: {story_id}\n\n"
-            txt_content += f"Title: {story_record.get('title', 'Untitled')}\n\n"
-            txt_content += f"Full Story:\n\n{story_record.get('full_story', '')}\n\n"
-
-            # Add sections if available
-            for idx, section in enumerate(sections):
-                txt_content += f"\n\nSection {idx + 1}:\n{section}"
-
-            # Add TXT file to ZIP
+            # ======================
+            # 2) Generate a TXT file
+            # ======================
+            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):
+                section_text = section_obj.get("section_text", "")
+                txt_content += f"\n\nSection {idx + 1}:\n{section_text}"
             zipf.writestr(f"{story_id}.txt", txt_content)
 
-            # --- Download Images ---
-            image_urls = story_record.get("image_urls", [])  # Expecting a list of image URLs
-            for idx, img_url in enumerate(image_urls):
-                if "firebasestorage.googleapis.com" in img_url:
+            # ======================
+            # 3) Download each section's image file to the ZIP
+            # ======================
+            # If you want each image in a separate folder, you can do "images/section_{idx+1}.jpg"
+            for idx, section_obj in enumerate(sections):
+                image_url = section_obj.get("image_url", "")
+                if image_url and "firebasestorage.googleapis.com" in image_url:
                     bucket = storage.bucket()
-                    blob_path = img_url.split('/o/')[-1].split('?')[0]
+                    blob_path = image_url.split('/o/')[-1].split('?')[0]
                     blob = bucket.blob(blob_path)
                     image_data = blob.download_as_bytes()
-                    zipf.writestr(f"image_{idx + 1}.jpg", image_data)
-
-            # --- Download Audio Files ---
-            audio_urls = story_record.get("audio_urls", [])  # Expecting a list of audio URLs
-            for idx, audio_url in enumerate(audio_urls):
-                if "firebasestorage.googleapis.com" in audio_url:
+                    zipf.writestr(f"image_section_{idx + 1}.jpg", image_data)
+
+            # ======================
+            # 4) Download each section's audio file to the ZIP
+            # ======================
+            for idx, section_obj in enumerate(sections):
+                audio_url = section_obj.get("audio_url", "")
+                if audio_url and "firebasestorage.googleapis.com" in audio_url:
                     bucket = storage.bucket()
                     blob_path = audio_url.split('/o/')[-1].split('?')[0]
                     blob = bucket.blob(blob_path)
                     audio_data = blob.download_as_bytes()
-                    zipf.writestr(f"audio_{idx + 1}.mp3", audio_data)
+                    zipf.writestr(f"audio_section_{idx + 1}.mp3", audio_data)
 
-        # --- Serve ZIP File ---
+        # ------------------------------------------------------------------
+        # Serve ZIP File
+        # ------------------------------------------------------------------
         zip_buffer.seek(0)
-        return send_file(zip_buffer, mimetype='application/zip', as_attachment=True, download_name=f"story_{story_id}.zip")
+        return send_file(
+            zip_buffer, 
+            mimetype='application/zip', 
+            as_attachment=True, 
+            download_name=f"story_{story_id}.zip"
+        )
 
     except Exception as e:
         return jsonify({'error': str(e)}), 500