Update main.py

main.py

@@ -18,6 +18,8 @@ import traceback
 from video_gen import create_video
 import zipfile
 from fpdf import FPDF
+import tempfile
+import urllib.parse
 
 # Initialize Flask app and CORS
 app = Flask(__name__)
@@ -650,15 +652,24 @@ def download_story_archive(story_id):
             # ======================
             # 1) Generate PDF
             # ======================
+            from fpdf import FPDF
+            
             pdf = FPDF()
             pdf.set_auto_page_break(auto=True, margin=15)
             pdf.add_page()
             pdf.set_font("Arial", size=12)
 
             # 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")
+            pdf.cell(0, 10, f"Story ID: {story_id}")
+            pdf.ln()
+            pdf.cell(0, 10, f"Title: {title}")
+            pdf.ln(15)
+            
+            # Add the full story text with proper line breaks
+            pdf.multi_cell(0, 10, "Full Story:")
+            pdf.ln(5)
+            pdf.multi_cell(0, 10, full_text)
+            pdf.ln(10)
 
             # For each section, create a new page
             for idx, section_obj in enumerate(sections):
@@ -668,24 +679,39 @@ def download_story_archive(story_id):
 
                 pdf.add_page()
                 pdf.set_font("Arial", style='B', size=14)
-                pdf.multi_cell(0, 10, f"Section {idx + 1}")
+                pdf.cell(0, 10, f"Section {idx + 1}")
+                pdf.ln(15)
                 pdf.set_font("Arial", size=12)
                 pdf.multi_cell(0, 10, section_text)
+                pdf.ln(10)
 
                 # 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()
-
-                    image_stream = io.BytesIO(image_data)
-    
-
-                    # Insert the image at (x=10, y=None => next line), width=100
-                    pdf.image(image_stream, x=10, y=None, w=100)
-                    # Clean up temp file
-                    os.remove(temp_img_path)
+                    try:
+                        bucket = storage.bucket()
+                        blob_path = urllib.parse.unquote(image_url.split('/o/')[-1].split('?')[0])
+                        blob = bucket.blob(blob_path)
+                        
+                        # Create a temporary file to store the image
+                        with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as temp_file:
+                            temp_img_path = temp_file.name
+                            # Download image to the temporary file
+                            blob.download_to_filename(temp_img_path)
+                        
+                        # Calculate appropriate image width (max 180 for A4 with margins)
+                        page_width = pdf.w - 2*pdf.l_margin
+                        image_width = min(page_width - 20, 160)  # Keep some margin
+                        
+                        # Add the image to PDF
+                        # The key fix is properly handling the image insertion
+                        pdf.image(temp_img_path, x=(pdf.w - image_width)/2, w=image_width)
+                        
+                        # Clean up temp file
+                        os.unlink(temp_img_path)
+                    except Exception as img_error:
+                        # Log error but continue if image insertion fails
+                        pdf.ln(5)
+                        pdf.multi_cell(0, 10, f"[Image could not be included: {str(img_error)}]")
 
             # Save PDF to memory
             pdf_buffer = io.BytesIO()
@@ -706,15 +732,18 @@ def download_story_archive(story_id):
             # ======================
             # 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 = image_url.split('/o/')[-1].split('?')[0]
-                    blob = bucket.blob(blob_path)
-                    image_data = blob.download_as_bytes()
-                    zipf.writestr(f"image_section_{idx + 1}.jpg", image_data)
+                    try:
+                        bucket = storage.bucket()
+                        blob_path = urllib.parse.unquote(image_url.split('/o/')[-1].split('?')[0])
+                        blob = bucket.blob(blob_path)
+                        image_data = blob.download_as_bytes()
+                        zipf.writestr(f"image_section_{idx + 1}.jpg", image_data)
+                    except Exception as img_error:
+                        # Log error but continue with the rest of the process
+                        print(f"Error downloading image for section {idx + 1}: {str(img_error)}")
 
             # ======================
             # 4) Download each section's audio file to the ZIP
@@ -722,11 +751,15 @@ def download_story_archive(story_id):
             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_section_{idx + 1}.mp3", audio_data)
+                    try:
+                        bucket = storage.bucket()
+                        blob_path = urllib.parse.unquote(audio_url.split('/o/')[-1].split('?')[0])
+                        blob = bucket.blob(blob_path)
+                        audio_data = blob.download_as_bytes()
+                        zipf.writestr(f"audio_section_{idx + 1}.mp3", audio_data)
+                    except Exception as audio_error:
+                        # Log error but continue with the rest of the process
+                        print(f"Error downloading audio for section {idx + 1}: {str(audio_error)}")
 
         # ------------------------------------------------------------------
         # Serve ZIP File
@@ -740,6 +773,8 @@ def download_story_archive(story_id):
         )
 
     except Exception as e:
+        import traceback
+        traceback.print_exc()
         return jsonify({'error': str(e)}), 500
 
 # ---------- Credit Request Endpoints ----------