| import gradio as gr | |
| import subprocess | |
| import matplotlib.pyplot as plt | |
| import traceback | |
| import sys | |
| import io | |
| import numpy as np | |
| 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} | |
| exec(code, {}, local_env) | |
| output = sys.stdout.getvalue() | |
| sys.stdout = old_stdout | |
| 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 create_image("") | |
| def create_image(message): | |
| fig, ax = plt.subplots() | |
| ax.text(0.5, 0.5, message, fontsize=20, ha='center', va='center') | |
| ax.axis('off') | |
| buf = io.BytesIO() | |
| plt.savefig(buf, format='png') | |
| buf.seek(0) | |
| img = Image.open(buf) | |
| plt.close() | |
| return np.array(img) | |
| 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." | |
| ) | |
| interface.launch() | |