Update app.py

app.py

@@ -53,10 +53,22 @@ def execute_code(code: str):
             if err.getvalue():
                 output_text += "\nErrors:\n" + err.getvalue()
         
-        # Check if there's a plot
+        # Check if there's a plot and convert to image
         if plt.gcf().axes:
-            output_plot = plt.gcf()
+            # Save plot to a temporary file
+            import tempfile
+            import os
+            
+            # Create a temporary file with .png extension
+            temp_dir = tempfile.gettempdir()
+            temp_file = os.path.join(temp_dir, f"plot_{os.getpid()}.png")
+            
+            # Save the current figure
+            plt.savefig(temp_file, bbox_inches='tight', dpi=300)
             plt.close()
+            
+            # Set the output_plot to the file path
+            output_plot = temp_file
         
         # Check for DataFrame output
         for var in namespace:
@@ -108,7 +120,7 @@ print(df.head())
         
         with gr.Column():
             output_text = gr.Textbox(label="Output", lines=5)
-            output_plot = gr.Plot(label="Plot Output")
+            output_plot = gr.Image(label="Plot Output")
             output_df = gr.Dataframe(label="DataFrame Output")
     
     # Handle code execution