Update app.py

app.py

@@ -1,7 +1,136 @@
 import gradio as gr
+import subprocess
+import re
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Working")
+
+# =========================
+# AI CHAT
+# =========================
+def ask_ai(prompt):
+    return f"AI Reply:\n\n{prompt}"
+
+
+# =========================
+# CODE EXECUTION
+# =========================
+def execute_code(code):
+    try:
+        result = subprocess.run(
+            ["python3", "-c", code],
+            capture_output=True,
+            text=True,
+            timeout=30
+        )
+
+        if result.returncode == 0:
+            return result.stdout or "Execution completed successfully."
+
+        return result.stderr
+
+    except subprocess.TimeoutExpired:
+        return "Execution timed out."
+
+    except Exception as e:
+        return str(e)
+
+
+# =========================
+# GENERATE + RUN
+# =========================
+def generate_and_run(prompt):
+
+    ai_response = ask_ai(prompt)
+
+    code_match = re.search(
+        r"```(?:python)?\s*(.*?)\s*```",
+        ai_response,
+        re.DOTALL
+    )
+
+    if code_match:
+
+        code = code_match.group(1)
+
+        result = execute_code(code)
+
+        return (
+            f"AI Response:\n\n{ai_response}\n\n"
+            f"Execution Result:\n\n{result}"
+        )
+
+    return ai_response
+
+
+# =========================
+# UI
+# =========================
+with gr.Blocks(
+    title="AI + Sandbox",
+    theme=gr.themes.Soft()
+) as demo:
+
+    gr.Markdown("# 🧠 AI Chat & Code Sandbox")
+
+    with gr.Tab("Chat with AI"):
+
+        chat_input = gr.Textbox(
+            label="Prompt",
+            placeholder="Type your message..."
+        )
+
+        chat_output = gr.Textbox(
+            label="Response",
+            lines=10
+        )
+
+        chat_btn = gr.Button("Send")
+
+        chat_btn.click(
+            fn=ask_ai,
+            inputs=chat_input,
+            outputs=chat_output
+        )
+
+    with gr.Tab("Code Sandbox"):
+
+        code_input = gr.Textbox(
+            label="Python Code",
+            lines=10,
+            placeholder="print('hello')"
+        )
+
+        code_output = gr.Textbox(
+            label="Output",
+            lines=10
+        )
+
+        run_btn = gr.Button("Run Code")
+
+        run_btn.click(
+            fn=execute_code,
+            inputs=code_input,
+            outputs=code_output
+        )
+
+    with gr.Tab("Generate & Run"):
+
+        auto_prompt = gr.Textbox(
+            label="Instruction",
+            placeholder="Write python code..."
+        )
+
+        auto_output = gr.Textbox(
+            label="Result",
+            lines=15
+        )
+
+        auto_btn = gr.Button("Generate & Run")
+
+        auto_btn.click(
+            fn=generate_and_run,
+            inputs=auto_prompt,
+            outputs=auto_output
+        )
 
 demo.launch(
     server_name="0.0.0.0",