Update app.py

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import io
 import contextlib
+import platform
+import psutil
+import os
 
 def execute_python_code(code):
     # Capture the output of the code execution
@@ -16,18 +19,52 @@ def execute_python_code(code):
     except Exception as e:
         return f"Error: {str(e)}"
 
-# Create the Gradio interface
-iface = gr.Interface(
-    fn=execute_python_code,
-    inputs=gr.Textbox(label="Python Code", lines=10, placeholder="Enter your Python code here..."),
-    outputs=gr.Textbox(label="Execution Result", lines=10),
-    title="Python Code Executor",
-    description="Enter Python code and see the execution results. Note: This executes code in a limited environment.",
-    examples=[
-        ["print('Hello, World!')"],
-        ["for i in range(5):\n    print(i)"],
-        ["def square(x):\n    return x*x\n\nprint(square(5))"]
-    ]
-)
+def get_system_info():
+    # Get system information
+    info = {
+        "System": platform.system(),
+        "Node Name": platform.node(),
+        "Release": platform.release(),
+        "Version": platform.version(),
+        "Machine": platform.machine(),
+        "Processor": platform.processor(),
+        "CPU Cores": psutil.cpu_count(logical=False),
+        "Logical CPUs": psutil.cpu_count(logical=True),
+        "RAM": f"{round(psutil.virtual_memory().total / (1024**3), 2)} GB",
+        "RAM Used": f"{round(psutil.virtual_memory().used / (1024**3), 2)} GB",
+        "RAM Free": f"{round(psutil.virtual_memory().free / (1024**3), 2)} GB",
+        "Disk Usage": f"{psutil.disk_usage('/').percent}%",
+        "Disk Total": f"{round(psutil.disk_usage('/').total / (1024**3), 2)} GB",
+        "Disk Free": f"{round(psutil.disk_usage('/').free / (1024**3), 2)} GB",
+        "Python Version": platform.python_version(),
+    }
+    
+    formatted_info = "\n".join([f"{k}: {v}" for k, v in info.items()])
+    return formatted_info
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Python Code Executor")
+    gr.Markdown("Enter Python code and see the execution results. Note: This executes code in a limited environment.")
+    
+    with gr.Row():
+        with gr.Column():
+            code_input = gr.Textbox(label="Python Code", lines=10, 
+                                  placeholder="Enter your Python code here...")
+            execute_btn = gr.Button("Execute Code")
+        with gr.Column():
+            output = gr.Textbox(label="Execution Result", lines=10)
+            sys_info_btn = gr.Button("Show System Info")
+    
+    gr.Examples(
+        examples=[
+            ["print('Hello, World!')"],
+            ["for i in range(5):\n    print(i)"],
+            ["def square(x):\n    return x*x\n\nprint(square(5))"]
+        ],
+        inputs=code_input
+    )
+    
+    execute_btn.click(fn=execute_python_code, inputs=code_input, outputs=output)
+    sys_info_btn.click(fn=get_system_info, inputs=None, outputs=output)
 
-iface.launch()
+demo.launch()