nerdify

app.py

@@ -10,37 +10,35 @@ from PIL import Image
 def run_code(code):
     try:
         if any(bad_cmd in code for bad_cmd in ['rm', 'os', 'shutil']):
-            return "Error: This command is not allowed."
-        
-        # Capture the standard output
+            return create_image("")
+
         old_stdout = sys.stdout
         sys.stdout = io.StringIO()
 
         local_env = {'plt': plt, 'subprocess': subprocess}
-
         exec(code, {}, local_env)
 
         output = sys.stdout.getvalue()
         sys.stdout = old_stdout
         
-        buf = io.BytesIO()
-        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:
-            return img_array  
-        
-        return noplot()  
+        if any('plt' in line for line in code.splitlines()):
+            buf = io.BytesIO()
+            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:
+                return img_array
+
+        return create_image("")
 
     except Exception as e:
-        return noplot()  
+        return create_image("")
 
-def noplot():
+def create_image(message):
     fig, ax = plt.subplots()
-    ax.text(0.5, 0.5, "Required: matplotlib", fontsize=20, ha='center', va='center')
+    ax.text(0.5, 0.5, message, fontsize=20, ha='center', va='center')
     ax.axis('off')
     buf = io.BytesIO()
     plt.savefig(buf, format='png')
@@ -51,9 +49,10 @@ def noplot():
 
 interface = gr.Interface(
     fn=run_code, 
-    inputs=gr.Code(language="python", label="code"), 
+    inputs=gr.Code(language="python", label="Enter Python Code"), 
     outputs=gr.Image(type="numpy", label="Output"), 
-    title="matplotlib-graph-creation",
+    title="Python Interpreter",
+    description="Run Python code. Requires matplotlib for plots."
 )
 
 interface.launch()