Update main.py

main.py

@@ -615,7 +615,6 @@ def view_videos():
 
 
 #download archives endpoint
-
 @app.route('/api/download/story_archive/<story_id>', methods=['GET'])
 def download_story_archive(story_id):
     try:
@@ -686,31 +685,38 @@ def download_story_archive(story_id):
                 # If there's an image URL, download & embed in PDF
                 if image_url:
                     try:
-                        # Initialize Firebase Storage bucket based on URL type
+                        # Initialize Firebase Storage bucket
+                        bucket = storage.bucket()
+                        
+                        # Handle both gs:// and HTTPS URLs
                         if image_url.startswith('gs://'):
+                            # Extract bucket name and file path from gs:// URL
                             path_parts = image_url[5:].split('/', 1)
                             if len(path_parts) != 2:
                                 continue
                             bucket_name, file_path = path_parts
                             bucket = storage.bucket(bucket_name)
                         else:
+                            # Extract file path from HTTPS URL
                             if '/o/' not in image_url or '?' not in image_url:
                                 continue
                             file_path = image_url.split('/o/')[1].split('?')[0]
                             file_path = urllib.parse.unquote(file_path)
-                            bucket = storage.bucket()
-
-                        blob = bucket.blob(file_path)
-                        # Create a temporary file to hold the image
+                        
+                        # Download the image to a temporary file
                         with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as temp_file:
                             temp_img_path = temp_file.name
-                        # Download the image to temp file
+                        
+                        # Download the image
+                        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)
-                        pdf.image(temp_img_path, x=(pdf.w - image_width)/2, w=image_width)
-                        # Clean up temp file
+                        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 temporary file
                         os.unlink(temp_img_path)
                     except Exception as img_error:
                         print(f"PDF image error: {str(img_error)}")
@@ -741,6 +747,10 @@ def download_story_archive(story_id):
                 image_url = section_obj.get("image_url", "")
                 if image_url:
                     try:
+                        # Initialize Firebase Storage bucket
+                        bucket = storage.bucket()
+                        
+                        # Handle both gs:// and HTTPS URLs
                         if image_url.startswith('gs://'):
                             path_parts = image_url[5:].split('/', 1)
                             if len(path_parts) != 2:
@@ -752,10 +762,12 @@ def download_story_archive(story_id):
                                 continue
                             file_path = image_url.split('/o/')[1].split('?')[0]
                             file_path = urllib.parse.unquote(file_path)
-                            bucket = storage.bucket()
+                        
+                        # Download the image
                         blob = bucket.blob(file_path)
                         image_data = blob.download_as_bytes()
-                        # Get file extension from blob name
+                        
+                        # Determine file extension
                         ext = os.path.splitext(blob.name)[1].lower() or '.jpg'
                         zipf.writestr(f"image_section_{idx + 1}{ext}", image_data)
                     except Exception as img_error:
@@ -765,6 +777,10 @@ def download_story_archive(story_id):
                 audio_url = section_obj.get("audio_url", "")
                 if audio_url:
                     try:
+                        # Initialize Firebase Storage bucket
+                        bucket = storage.bucket()
+                        
+                        # Handle both gs:// and HTTPS URLs
                         if audio_url.startswith('gs://'):
                             path_parts = audio_url[5:].split('/', 1)
                             if len(path_parts) != 2:
@@ -776,9 +792,12 @@ def download_story_archive(story_id):
                                 continue
                             file_path = audio_url.split('/o/')[1].split('?')[0]
                             file_path = urllib.parse.unquote(file_path)
-                            bucket = storage.bucket()
+                        
+                        # Download the audio
                         blob = bucket.blob(file_path)
                         audio_data = blob.download_as_bytes()
+                        
+                        # Determine file extension
                         ext = os.path.splitext(blob.name)[1].lower() or '.mp3'
                         zipf.writestr(f"audio_section_{idx + 1}{ext}", audio_data)
                     except Exception as audio_error: