Update app.py

app.py

@@ -4,40 +4,58 @@ import numpy as np
 import io
 from PIL import Image
 
-def generate_graph_from_code(code):
-    # Create a buffer to capture output
+def execute_code_and_get_plot(code):
+    # Clear any existing plots
+    plt.clf()
+    
+    # Create a BytesIO object to store the plot
     buf = io.BytesIO()
     
     try:
-        # Define a dummy multiplier if not present in the code
-        exec(code)
-        if 'multiplier' not in locals():
-            multiplier = 1
+        # Execute the code
+        exec(code, globals())
+        
+        # Save the current plot to the BytesIO object
+        plt.savefig(buf, format='png')
+        buf.seek(0)
+        
+        # Convert to PIL Image
+        image = Image.open(buf)
+        return image, "Code executed successfully!"
     except Exception as e:
-        return str(e)
-    
-    # Generate the graph
-    x = np.linspace(0, 10, 100)
-    y = np.sin(x) * multiplier
-    
-    plt.figure()
-    plt.plot(x, y)
-    plt.title("Generated Graph")
-    plt.xlabel("x")
-    plt.ylabel("y")
-    plt.grid(True)
-    plt.savefig(buf, format='png')
-    buf.seek(0)
-    img = Image.open(buf)
-    return img
+        return None, f"Error: {str(e)}"
 
-# Define the Gradio interface using the updated API
+# Define the Gradio interface
 interface = gr.Interface(
-    fn=generate_graph_from_code,
-    inputs=gr.Textbox(lines=10, placeholder="Paste your Python code here", label="Python Code"),
-    outputs=gr.Image(type="pil", label="Generated Graph"),
-    title="Python Code to Graph Generator",
-    description="Paste Python code that defines a variable `multiplier` to generate a graph."
+    fn=execute_code_and_get_plot,
+    inputs=[
+        gr.Code(
+            label="Python Code",
+            language="python",
+            lines=20,
+            placeholder="""
+# Example:
+import matplotlib.pyplot as plt
+import numpy as np
+
+x = np.linspace(0, 10, 100)
+y = np.sin(x)
+
+plt.figure(figsize=(8, 6))
+plt.plot(x, y)
+plt.title('Sine Wave')
+plt.xlabel('x')
+plt.ylabel('sin(x)')
+plt.grid(True)
+"""
+        )
+    ],
+    outputs=[
+        gr.Image(type="pil", label="Generated Plot"),
+        gr.Textbox(label="Output Message")
+    ],
+    title="Python Code to Matplotlib Plot Generator",
+    description="Enter Python code that generates a matplotlib plot. The plot will be displayed as an image."
 )
 
 if __name__ == "__main__":