Update app.py

app.py

@@ -6,10 +6,15 @@ import io
 import re
 import os
 import requests
+import logging
 from pptx import Presentation
 from pptx.util import Inches, Pt
 from concurrent.futures import ThreadPoolExecutor
 
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
 # Stability AI API key
 STABILITY_API_KEY = os.getenv('STABILITY_API_KEY')
 
@@ -102,14 +107,15 @@ def generate_image(prompt, style_preset):
         ]
     }
     
-    response = requests.post(url, json=body, headers=headers)
-    
-    if response.status_code == 200:
+    try:
+        response = requests.post(url, json=body, headers=headers)
+        response.raise_for_status()
         data = response.json()
         image_data = base64.b64decode(data['artifacts'][0]['base64'])
+        logger.info(f"Image generated successfully for prompt: {prompt[:30]}...")
         return image_data
-    else:
-        print(f"Error: {response.status_code}, {response.text}")
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Error generating image: {str(e)}")
         return None
 
 def enhance_prompt(text):
@@ -168,7 +174,7 @@ def markdown_to_pptx(md_text):
             title_box.text_frame.paragraphs[0].font.size = Pt(32)
             title_box.text_frame.paragraphs[0].font.bold = True
 
-            # Add content
+            # Add content (left side)
             content_box = current_slide.shapes.add_textbox(Inches(0.5), Inches(1.5), Inches(4.5), Inches(6))
             text_frame = content_box.text_frame
             for line in lines[1:]:
@@ -183,11 +189,16 @@ def markdown_to_pptx(md_text):
                     p.font.size = Pt(18)
                     p.level = 1
 
-            # Add image
+            # Add image (right side)
             image_data = future.result()
             if image_data:
                 image_stream = io.BytesIO(image_data)
                 current_slide.shapes.add_picture(image_stream, Inches(5.5), Inches(1.5), width=Inches(4), height=Inches(6))
+            else:
+                logger.warning(f"Failed to generate image for slide: {title}")
+                # Add a placeholder textbox for the missing image
+                placeholder = current_slide.shapes.add_textbox(Inches(5.5), Inches(1.5), Inches(4), Inches(6))
+                placeholder.text_frame.text = "Image generation failed"
 
     output = io.BytesIO()
     prs.save(output)
@@ -231,6 +242,7 @@ def convert_markdown(n_clicks, file_contents, markdown_text):
             className="mt-3"
         )
     except Exception as e:
+        logger.exception("An error occurred during conversion")
         return dbc.Alert(f"An error occurred: {str(e)}", color="danger")
 
 if __name__ == '__main__':