fix again; redid;

app.py

@@ -1,7 +1,5 @@
 import gradio as gr
-import subprocess
 import matplotlib.pyplot as plt
-import traceback
 import sys
 import io
 import numpy as np
@@ -9,32 +7,33 @@ from PIL import Image
 
 def run_code(code):
     try:
-        if any(bad_cmd in code for bad_cmd in ['rm', 'os', 'shutil']):
-            return create_image("")
-
         old_stdout = sys.stdout
         sys.stdout = io.StringIO()
 
-        local_env = {'plt': plt, 'subprocess': subprocess}
+        local_env = {'plt': plt, 'np': np}
         exec(code, {}, local_env)
 
         output = sys.stdout.getvalue()
         sys.stdout = old_stdout
         
-        if any('plt' in line for line in code.splitlines()):
+        if 'plt' in code:
             buf = io.BytesIO()
+            plt.draw()
             plt.savefig(buf, format='png')
             buf.seek(0)
             img = Image.open(buf)
             img_array = np.array(img)
             plt.close()
-            if np.array(img_array).sum() > 0:
+
+            if img_array.sum() > 0:
                 return img_array
+            else:
+                return create_image("Empty plot")
 
-        return create_image("")
+        return create_image("No valid matplotlib plot found")
 
     except Exception as e:
-        return create_image("")
+        return create_image(f"Error: {str(e)}")
 
 def create_image(message):
     fig, ax = plt.subplots()
@@ -51,8 +50,8 @@ interface = gr.Interface(
     fn=run_code, 
     inputs=gr.Code(language="python", label="Enter Python Code"), 
     outputs=gr.Image(type="numpy", label="Output"), 
-    title="Python Interpreter",
-    description="Run Python code. Requires matplotlib for plots."
+    title="Matplotlib Plot Generator",
+    description="Enter Python code that uses matplotlib to generate plots."
 )
 
-interface.launch()
+interface.launch()