Update main.py

main.py

@@ -625,7 +625,6 @@ if not os.path.exists(LOG_FILE_PATH):
         pass  # create empty file
 
 logging.basicConfig(filename=LOG_FILE_PATH, level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
-
 @app.route('/api/download/story_archive/<story_id>', methods=['GET'])
 def download_story_archive(story_id):
     try:
@@ -636,7 +635,7 @@ def download_story_archive(story_id):
         if not auth_header.startswith('Bearer '):
             logging.warning("❌ Missing or invalid token")
             return jsonify({'error': 'Missing or invalid token'}), 401
-        
+
         token = auth_header.split(' ')[1]
         uid = verify_token(token)
         if not uid:
@@ -672,10 +671,7 @@ def download_story_archive(story_id):
 
             pdf.multi_cell(190, 10, f"Story ID: {story_id}")
             pdf.multi_cell(190, 10, f"Title: {title}")
-            pdf.ln(5)
-            pdf.multi_cell(190, 10, "Full Story:")
-            pdf.multi_cell(190, 10, full_text)
-            pdf.ln(10)
+            pdf.ln(10)  # Add spacing before sections start
 
             bucket = storage.bucket()
 
@@ -706,8 +702,9 @@ def download_story_archive(story_id):
 
                         blob = bucket.blob(file_path)
                         blob.download_to_filename(temp_img_path)
-                        
-                        pdf.image(temp_img_path, x=10, w=100)
+
+                        # Adjust image placement and size to fit allocated space
+                        pdf.image(temp_img_path, x=10, w=pdf.w - 20, h=100)  # Full allocated width, fixed height
                         os.unlink(temp_img_path)
                         logging.info(f"✅ Image embedded in PDF for section {idx + 1}")
                     except Exception as img_error:
@@ -787,7 +784,6 @@ def download_story_archive(story_id):
         traceback.print_exc()
         return jsonify({'error': str(e)}), 500
 
-
 def extract_firebase_path(public_url: str) -> str:
     """
     Attempt to parse a Firebase/Google Storage URL, returning just the