Update app.py

app.py

@@ -7,7 +7,6 @@ from google.genai import types
 
 app = FastAPI()
 
-# Gemini Client Initialize karein (Hugging Face Secrets se key lega)
 GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
 client = genai.Client(api_key=GEMINI_API_KEY)
 
@@ -15,10 +14,22 @@ class ChatRequest(BaseModel):
     message: str
 
 def execute_in_sandbox(command: str):
-    """Ye function terminal commands ko sandbox ke andar safe tareeke se chalata hai"""
     try:
-        result = subprocess.run(command, shell=True, capture_output=True, text=True, timeout=300)
-        return {"stdout": result.stdout, "stderr": result.stderr, "code": result.returncode}
+        # Commands hamesha writeable directory (/tmp) me run ho, isliye cd /tmp lagaya hai
+        full_command = f"cd /tmp && {command}"
+        
+        result = subprocess.run(
+            full_command, 
+            shell=True, 
+            capture_output=True, 
+            text=True, 
+            timeout=600 # 10 mins timeout taaki installation aaram se ho sake
+        )
+        return {
+            "stdout": result.stdout, 
+            "stderr": result.stderr, 
+            "exit_code": result.returncode
+        }
     except Exception as e:
         return {"error": str(e)}
 
@@ -26,30 +37,26 @@ def execute_in_sandbox(command: str):
 async def agent_chat(request: ChatRequest):
     user_prompt = request.message
     
-    # System instruction jo agent ko batati hai ki wo code run kar sakta hai
+    # AI ko instruction do ki use jo chahiye wo khud bash command se install kare (e.g., flutter, sdk)
     system_instruction = (
-        "You are an AI Agent Builder with access to a Linux terminal sandbox. "
-        "You can write and execute terminal commands to build Flutter apps, web apps, or Android apks. "
-        "If you need to run a command, respond with the exact command inside a triple backtick block starting with 'bash'."
+        "You are an AI Agent Builder operating inside a limited Linux sandbox. "
+        "You have full permission to download and install necessary tools (like Flutter, Node.js, Android cmdline-tools) "
+        "using bash commands when required. Always perform your operations inside the `/tmp` directory because it has write permissions. "
+        "If you want to execute any tool or bash setup, return the exact script inside a triple backtick block starting with 'bash'."
     )
     
     try:
-        # Gemini 2.5 Flash ya Pro use karein projects ke liye
         response = client.models.generate_content(
             model='gemini-2.5-flash',
             contents=user_prompt,
-            config=types.GenerateContentConfig(
-                system_instruction=system_instruction
-            )
+            config=types.GenerateContentConfig(system_instruction=system_instruction)
         )
         
         ai_response = response.text
         
-        # Agar AI ne koi command execute karne ko kaha hai (Basic automation logic)
         if "```bash" in ai_response:
-            # Command extract karne ka logic (Simple strip)
-            command = ai_response.split("```bash")[1].split("```")[0].strip()
-            # Command sandbox me chalayein
+            command = ai_response.split("
+```bash")[1].split("```")[0].strip()
             exec_result = execute_in_sandbox(command)
             return {"agent_response": ai_response, "execution_result": exec_result}