Update app.py

app.py

@@ -1,26 +1,86 @@
 import gradio as gr
 import subprocess
+import os
 
+# ========== AI ko call karne ka function (pehle jaise) ==========
 def ask_ai(prompt):
-    # OpenClaw ko call karein (command adjust karein agar jaroori ho)
     try:
-        # Example: "openclaw run <prompt>"
+        # Yahan aapne jo sahi command identify kari thi wo daalen (e.g., "openclaw ask")
+        # agar abhi pata nahi to "openclaw --help" se dekh len. Example:
         result = subprocess.run(
-            ["openclaw", "run", prompt],
+            ["openclaw", "ask", prompt],   # ya "chat", "generate" etc.
             capture_output=True,
             text=True,
-            timeout=120
+            timeout=120,
+            env={**os.environ, "OPENCLAW_DEBUG": "0"}
         )
-        return result.stdout or result.stderr
+        output = result.stdout or result.stderr
+        return output
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"AI Error: {str(e)}"
 
-iface = gr.Interface(
-    fn=ask_ai,
-    inputs=gr.Textbox(label="Prompt", placeholder="Kuchh bhi puchhiye..."),
-    outputs=gr.Textbox(label="OpenClaw Response"),
-    title="OpenClaw AI",
-    description="Ye Space OpenClaw + sabhi tools ke saath ready hai."
-)
+# ========== Code Sandbox function ==========
+def execute_code(code):
+    # Warning: ye container ke andar raw code run karta hai - sirf trusted code ke liye
+    try:
+        proc = subprocess.run(
+            ["python3", "-c", code],
+            capture_output=True,
+            text=True,
+            timeout=30,
+            env={**os.environ, "PYTHONUNBUFFERED": "1"}
+        )
+        stdout = proc.stdout
+        stderr = proc.stderr
+        if proc.returncode == 0:
+            return f"✅ Execution Success\n\nOutput:\n{stdout}"
+        else:
+            return f"❌ Error (exit code {proc.returncode})\n\n{stderr or stdout}"
+    except subprocess.TimeoutExpired:
+        return "⏰ Timeout: Code took too long (>30s)"
+    except Exception as e:
+        return f"🔥 Sandbox Error: {str(e)}"
+
+# ========== Gradio interface (Tabs) ==========
+with gr.Blocks(title="AI + Sandbox", theme="soft") as demo:
+    gr.Markdown("# 🧠 AI Chat & Code Sandbox")
+
+    with gr.Tab("Chat with AI"):
+        gr.Markdown("OpenClaw se baat karein. Sahi command `openclaw ask` ya `openclaw chat` set karein.")
+        chat_input = gr.Textbox(label="Aapka prompt", placeholder="Type message...")
+        chat_output = gr.Textbox(label="AI Response", lines=10)
+        chat_button = gr.Button("Send")
+        chat_button.click(fn=ask_ai, inputs=chat_input, outputs=chat_output)
+
+    with gr.Tab("Code Sandbox"):
+        gr.Markdown("Python code run karein. **⚠️ Safety:** Only run trusted code.")
+        code_input = gr.Textbox(
+            label="Python Code",
+            placeholder="print('Hello World')",
+            lines=8,
+            language="python"
+        )
+        code_output = gr.Textbox(label="Output", lines=10)
+        run_button = gr.Button("Run Code")
+        run_button.click(fn=execute_code, inputs=code_input, outputs=code_output)
+
+        gr.Markdown("---\n**Aur advanced: AI ke response se code extract karke run karna**")
+        auto_prompt = gr.Textbox(label="AI ko instruction dein (jo code generate kare)", placeholder="Write a Python script to list files")
+        auto_generate = gr.Button("Generate & Run Code")
+        auto_output = gr.Textbox(label="AI Response + Code Execution", lines=15)
+
+        def generate_and_run(prompt):
+            # Pehle AI se code generato karao
+            ai_response = ask_ai(prompt + "\nReturn only the Python code inside triple backticks.")
+            # Code extract karein (simple regex)
+            import re
+            code_match = re.search(r"```(?:python)?\s*(.*?)\s*```", ai_response, re.DOTALL)
+            if code_match:
+                code = code_match.group(1).strip()
+                exec_result = execute_code(code)
+                return f"**AI Response:**\n{ai_response}\n\n**Executed Code:**\n```python\n{code}\n```\n\n**Execution Result:**\n{exec_result}"
+            else:
+                return f"**AI Response:**\n{ai_response}\n\n*(No code block found to execute)*"
+        auto_generate.click(fn=generate_and_run, inputs=auto_prompt, outputs=auto_output)
 
-iface.launch(server_name="0.0.0.0", server_port=7860)
+demo.launch(server_name="0.0.0.0", server_port=7860)