Update app.py

app.py

@@ -5,6 +5,7 @@ import io
 import ast
 from PIL import Image, ImageDraw
 import google.generativeai as genai
+import traceback
 
 def process_file(api_key, file, instructions):
     try:
@@ -27,6 +28,11 @@ def process_file(api_key, file, instructions):
 
         # Extract code block safely
         code_block = response.text.split('```python')[1].split('```')[0].strip()
+        
+        # Print the code block for debugging
+        print("Generated code block:")
+        print(code_block)
+        
         plots = ast.literal_eval(code_block)
         
         # Generate visualizations
@@ -59,9 +65,11 @@ def process_file(api_key, file, instructions):
         return images if len(images) == 3 else images + [Image.new('RGB', (800, 600), (255,255,255))]*(3-len(images))
 
     except Exception as e:
-        error_image = Image.new('RGB', (800, 100), (255, 255, 255))
+        error_message = f"Error: {str(e)}\n\nTraceback:\n{traceback.format_exc()}"
+        print(error_message)  # Print to console for debugging
+        error_image = Image.new('RGB', (800, 400), (255, 255, 255))
         draw = ImageDraw.Draw(error_image)
-        draw.text((10, 40), f"Error: {str(e)}", fill=(255, 0, 0))
+        draw.text((10, 10), error_message, fill=(255, 0, 0))
         return [error_image] * 3
 
 with gr.Blocks(theme=gr.themes.Default(spacing_size="lg")) as demo: