Update app.py

app.py

@@ -7,46 +7,49 @@ from IPython.core.interactiveshell import InteractiveShell
 from IPython.utils.capture import capture_output
 
 def run_jupyter_style_code(code):
-    # Close any old plots
-    plt.close('all')
+    plt.close('all')  # Close any old plots
 
-    # Set up an interactive shell instance
     shell = InteractiveShell.instance()
     img_path = None
+    output = ""
 
     try:
-        # Capture output (both text and rich outputs like HTML, plots)
+        # Capture output (stdout, stderr) and display anything that gets returned
         with capture_output(display=True) as captured:
             exec(code, {"plt": plt, "pd": __import__('pandas'), "display": display})
 
-        # Check for any figures and save the last one as an image
+        # Get any figures generated by Matplotlib
         figures = [plt.figure(i) for i in plt.get_fignums()]
         if figures:
             img_path = "plot.png"
             figures[-1].savefig(img_path)
 
-        # Return the captured HTML/text outputs
-        html_output = captured.stdout + ''.join(str(out) for out in captured.outputs)
-        return html_output, img_path
+        # Join captured output (stdout + rich outputs like HTML or plots)
+        output = captured.stdout + ''.join(str(out) for out in captured.outputs)
+        return output, img_path
 
     except Exception as e:
-        # Return any errors as string
-        return traceback.format_exc(), None
+        # If an error occurs, return the traceback as output
+        output = traceback.format_exc()
+        return output, None
 
-
-# Gradio Interface
+# Create the Gradio UI
 with gr.Blocks() as demo:
-    gr.Markdown("## 🧪 Python Code Runner with Jupyter-style Output")
+    gr.Markdown("## 🧪 Python Code Runner with Jupyter-style Terminal")
+
+    # Code input area (multi-line text box simulating terminal)
+    code_input = gr.Textbox(label="Python Terminal", placeholder="Enter Python code...", lines=10)
     
-    # Code input area
-    code_input = gr.Code(language="python", label="Enter Python Code")
+    # Output for the console/logs (including rich content)
+    output_text = gr.Textbox(label="Console Output", lines=12, interactive=False)
     
-    # Outputs (HTML + Plot)
-    output_text = gr.HTML(label="Console Output + Rich Result")
+    # Display for any generated plots
     plot_output = gr.Image(label="Generated Plot", type="filepath")
-
-    # Run button
+    
+    # Button to execute the code
     run_button = gr.Button("Run Code")
+    
+    # Trigger execution on button click
     run_button.click(fn=run_jupyter_style_code, inputs=code_input, outputs=[output_text, plot_output])
 
 if __name__ == "__main__":