Update main.py

main.py

@@ -637,8 +637,8 @@ def download_story_archive(story_id):
 
         # Prepare data
         full_text = story_record.get("full_story", "")
-        sections = story_record.get("sections",)
-
+        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"
@@ -655,24 +655,14 @@ def download_story_archive(story_id):
             pdf = FPDF()
             pdf.set_auto_page_break(auto=True, margin=15)
             pdf.add_page()
-
-            # --- Register and set Unicode fonts ---
-            # Assuming you have TTF files named 'DejaVuSans.ttf' (regular) and 'DejaVuSans-Bold.ttf' (bold)
-            # in the same directory
-            try:
-                pdf.add_font('DejaVu', '', 'DejaVuSans.ttf', uni=True)
-                pdf.add_font('DejaVu', 'B', 'dejaVu-sans-bold.ttf', uni=True)
-                pdf.set_font('DejaVu', '', 12) # Set default font to regular
-            except Exception as e:
-                print(f"Error loading Unicode font: {e}")
-                pdf.set_font("Arial", size=12) # Fallback to Arial if DejaVu fails
+            pdf.set_font("Arial", size=12)
 
             # Basic info
             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)
@@ -686,52 +676,46 @@ def download_story_archive(story_id):
                 image_url = section_obj.get("image_url", "")
 
                 pdf.add_page()
-                pdf.set_font('DejaVu', 'B', 14) # Set font to bold for section title
+                pdf.set_font("Arial", style='B', size=14)
                 pdf.cell(0, 10, f"Section {idx + 1}")
                 pdf.ln(15)
-                pdf.set_font('DejaVu', '', 12) # Set font back to regular for section text
+                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:
                     try:
-                        # Initialize Firebase Storage bucket (ensure this is correctly configured elsewhere)
+                        # Initialize Firebase Storage bucket
                         bucket = storage.bucket()
-
+                        
                         # Get the image file path from the URL
+                        # Firebase Storage URLs are in the format:
+                        # https://firebasestorage.googleapis.com/v0/b/[bucket]/o/[file_path]?[params]
                         file_path = ""
                         if '/o/' in image_url and '?' in image_url:
                             file_path = image_url.split('/o/')[1].split('?')[0]
                             file_path = urllib.parse.unquote(file_path)
-
-                        if file_path:
-                            # Create a temporary file to hold the image
-                            with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as temp_file:
-                                temp_img_path = temp_file.name
-
-                                # Download the image to temp file
-                                blob = bucket.blob(file_path)
-                                blob.download_to_filename(temp_img_path)
-
-                                # Calculate appropriate image width and place it on the page
-                                page_width = pdf.w - 2*pdf.l_margin
-                                image_width = min(page_width - 20, 160)  # Keep some margin
-                                pdf.image(temp_img_path, x=(pdf.w - image_width)/2, w=image_width)
-
-                                # Clean up temp file
-                                os.unlink(temp_img_path)
-                        else:
-                            print(f"PDF image error: Could not parse image URL for section {idx + 1}: {image_url}")
-                            pdf.ln(5)
-                            pdf.multi_cell(0, 10, f"[Image URL invalid]")
-
+                        
+                        # Create a temporary file to hold the image
+                        with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as temp_file:
+                            temp_img_path = temp_file.name
+                        
+                        # Download the image to temp file
+                        blob = bucket.blob(file_path)
+                        blob.download_to_filename(temp_img_path)
+                        
+                        # Calculate appropriate image width and place it on the page
+                        page_width = pdf.w - 2*pdf.l_margin
+                        image_width = min(page_width - 20, 160)  # Keep some margin
+                        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:
-                        print(f"PDF image error for section {idx + 1}: {str(img_error)}")
-                        import traceback
-                        traceback.print_exc()
+                        print(f"PDF image error: {str(img_error)}")
                         pdf.ln(5)
-                        pdf.multi_cell(0, 10, f"[Image could not be included for section {idx + 1}]")
+                        pdf.multi_cell(0, 10, f"[Image could not be included]")
 
             # Save PDF to memory
             pdf_buffer = io.BytesIO()
@@ -762,20 +746,14 @@ def download_story_archive(story_id):
                         if '/o/' in image_url and '?' in image_url:
                             file_path = image_url.split('/o/')[1].split('?')[0]
                             file_path = urllib.parse.unquote(file_path)
-
-                        if file_path:
-                            # Get the blob and download as bytes
-                            blob = bucket.blob(file_path)
-                            image_data = blob.download_as_bytes()
-                            zipf.writestr(f"image_section_{idx + 1}.jpg", image_data)
-                            print(f"Added image for section {idx + 1}")
-                        else:
-                            print(f"Error downloading image for section {idx + 1}: Could not parse URL: {image_url}")
-
+                        
+                        # Get the blob and download as bytes
+                        blob = bucket.blob(file_path)
+                        image_data = blob.download_as_bytes()
+                        zipf.writestr(f"image_section_{idx + 1}.jpg", image_data)
+                        print(f"Added image for section {idx + 1}")
                     except Exception as img_error:
                         print(f"Error downloading image for section {idx + 1}: {str(img_error)}")
-                        import traceback
-                        traceback.print_exc()
 
             # ======================
             # 4) Download each section's audio file to the ZIP
@@ -789,29 +767,23 @@ def download_story_archive(story_id):
                         if '/o/' in audio_url and '?' in audio_url:
                             file_path = audio_url.split('/o/')[1].split('?')[0]
                             file_path = urllib.parse.unquote(file_path)
-
-                        if file_path:
-                            # Get the blob and download as bytes
-                            blob = bucket.blob(file_path)
-                            audio_data = blob.download_as_bytes()
-                            zipf.writestr(f"audio_section_{idx + 1}.mp3", audio_data)
-                            print(f"Added audio for section {idx + 1}")
-                        else:
-                            print(f"Error downloading audio for section {idx + 1}: Could not parse URL: {audio_url}")
-
+                        
+                        # Get the blob and download as bytes
+                        blob = bucket.blob(file_path)
+                        audio_data = blob.download_as_bytes()
+                        zipf.writestr(f"audio_section_{idx + 1}.mp3", audio_data)
+                        print(f"Added audio for section {idx + 1}")
                     except Exception as audio_error:
                         print(f"Error downloading audio for section {idx + 1}: {str(audio_error)}")
-                        import traceback
-                        traceback.print_exc()
 
         # ------------------------------------------------------------------
         # Serve ZIP File
         # ------------------------------------------------------------------
         zip_buffer.seek(0)
         return send_file(
-            zip_buffer,
-            mimetype='application/zip',
-            as_attachment=True,
+            zip_buffer, 
+            mimetype='application/zip', 
+            as_attachment=True, 
             download_name=f"story_{story_id}.zip"
         )