Update app.py

app.py

@@ -8,6 +8,7 @@ from google.genai import types
 
 app = FastAPI()
 
+# CORS Configurations - Saari settings pehle jaisi safe hain
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
@@ -22,25 +23,36 @@ client = genai.Client(api_key=GEMINI_API_KEY)
 class ChatRequest(BaseModel):
     message: str
 
-def execute_in_sandbox(command: str):
+# TOOL: Hardcore Terminal Execution System (Root Permission)
+def run_terminal_command(command: str) -> str:
+    """
+    Executes any bash command inside the secure Linux sandbox workspace (/tmp).
+    Use this to run python scripts, compile Flutter apps, build APKs, or manage git operations (like git worktree).
+    """
     try:
-        # Saari commands hamesha /tmp folder me execute hongi jahan explicit write permissions hain
         full_command = f"cd /tmp && {command}"
+        env = os.environ.copy()
         
         result = subprocess.run(
             full_command, 
             shell=True, 
             capture_output=True, 
             text=True, 
-            timeout=600  # 10 minutes max timeout per execution
+            errors="replace",
+            timeout=300,
+            env=env
         )
-        return {
-            "stdout": result.stdout, 
-            "stderr": result.stderr, 
-            "exit_code": result.returncode
-        }
+        
+        output = f"Exit Code: {result.returncode}\n"
+        if result.stdout:
+            output += f"STDOUT:\n{result.stdout}\n"
+        if result.stderr:
+            output += f"STDERR:\n{result.stderr}\n"
+        return output
+    except subprocess.TimeoutExpired:
+        return "ERROR: Command execution timed out after 5 minutes to prevent server freeze."
     except Exception as e:
-        return {"error": str(e)}
+        return f"SYSTEM ERROR: {str(e)}"
 
 @app.post("/chat")
 async def agent_chat(request: ChatRequest):
@@ -51,33 +63,81 @@ async def agent_chat(request: ChatRequest):
         "YOUR DEVELOPMENT ENVIRONMENT capabilities:\n"
         "- **Python 3** is installed. You can write and execute python scripts (`python3 script.py`).\n"
         "- **Flutter & Dart SDK** are pre-installed globally. You can run `flutter create` and `flutter build apk --release`.\n"
-        "- **Android Native Build System** is pre-installed. Full OpenJDK 17, Android SDK (API 34), and Build-Tools (34.0.0) are ready. You can create Native Android apps (using gradle wrappers) and build them directly.\n"
-        "- You CANNOT use 'sudo'. Perform all workspace creation inside the `/tmp` directory.\n\n"
-        "GOFILE.IO UPLOAD RULES:\n"
-        "To upload a compiled file or APK, you MUST follow these exact steps in your bash script:\n"
-        "1. Fetch an active server name: SERVER_NAME=$(curl -s https://api.gofile.io/getServer | grep -oP '\"server\":\"\\K[^\"]+')\n"
-        "2. Upload using that server: UPLOAD_RESPONSE=$(curl -F \"file=@path_to_your_file\" https://${SERVER_NAME}.gofile.io/uploadFile)\n"
-        "3. Extract download link: DOWNLOAD_LINK=$(echo \"${UPLOAD_RESPONSE}\" | grep -oP '\"downloadPage\":\"\\K[^\"]+')\n\n"
-        "If you want to create a project, write python scripts, or build an APK, output the exact script inside a triple backtick block starting with 'bash'."
+        "- **Android Native Build System** is pre-installed. Full OpenJDK 17, Android SDK (API 34), Build-Tools (34.0.0), and Gradle Build Binary are ready. You can create Native Android apps (using gradle wrappers) and build them directly.\n"
+        "- **HOP-AI SDK** is pre-installed. You can use 'import hop' in your python scripts to interact with remote cloud volumes, manage infrastructure, or offload long-running tasks if necessary.\n"
+        "- You are operating with full administrative permissions inside the `/tmp` directory.\n\n"
+        "YOUR AVAILABLE TOOLS:\n"
+        "- `run_terminal_command`: Call this tool to run any shell/bash script or build commands. Do NOT wrap shell code in text-based markdown blocks if you intend to run it; execute it directly via this tool.\n"
+        "- `Google Search`: You have native access to live Google Search. Use it automatically whenever you need to look up documentation for APIs (like catbox.moe, gofile.io, transfer.sh), research web endpoints, check dependency updates, or debug compilation errors.\n\n"
+        "CRITICAL APPLICATION GENERATION RULES:\n"
+        "- The workspace sandbox is initially empty. There are no pre-existing Android or Flutter templates inside `/tmp`. If you are tasked to create an app, you MUST programmatically write and create every single source directory, file structure, configuration script, layout resource, and source file entirely from scratch.\n"
+        "- Always use standard POSIX compatible shell scripting loops (e.g., `for i in 1 2 3; do ... done`) instead of advanced bash-specific loops like `for ((i=0; i<3; i++))` because the sandbox default runtime shell executor is `/bin/sh`.\n\n"
+        "ADVANCED GIT WORKSPACE MANAGEMENT:\n"
+        "- You are encouraged to use `git worktree` when managing parallel tasks, feature implementations, or isolated testing branches. Instead of risky checkout switching or stashing uncommitted modifications, utilize `git worktree add -b <branch-name> <path>` to spin up isolated, standalone directories inside `/tmp` for separate branches. This keeps the main codebase safe and unpolluted during builds.\n\n"
+        "CRITICAL EXECUTION RULES:\n"
+        "- Analyze tool execution outputs carefully. If a command or build script fails, analyze the error log, use Google Search to find a solution or documentation if needed, self-correct, update your code, and execute again until successful.\n"
+        "- Once the objective is completed successfully (e.g., app compiled or file uploaded), present a clean summary and direct download links to the user."
     )
     
     try:
-        response = client.models.generate_content(
-            model='gemini-2.5-flash',
-            contents=user_prompt,
-            config=types.GenerateContentConfig(system_instruction=system_instruction)
+        # Custom functional tools mapping
+        custom_tools = {
+            "run_terminal_command": run_terminal_command
+        }
+        
+        # Gemma-2-9b-it model configurations
+        config = types.GenerateContentConfig(
+            system_instruction=system_instruction,
+            tools=[run_terminal_command, {'google_search': {}}],
+            temperature=0.2
         )
         
-        ai_response = response.text
+        # Chat session explicitly locked to Gemma
+        chat = client.chats.create(model='gemma2-9b-it', config=config)
+        response = chat.send_message(user_prompt)
         
-        if "```bash" in ai_response:
-            parts = ai_response.split("```bash")
-            if len(parts) > 1:
-                command = parts[1].split("```")[0].strip()
-                exec_result = execute_in_sandbox(command)
-                return {"agent_response": ai_response, "execution_result": exec_result}
+        max_turns = 6 
+        current_turn = 0
+        execution_history = []
+        
+        # AGENT EXECUTION LOOP
+        while response.function_calls and current_turn < max_turns:
+            function_responses = []
+            
+            for function_call in response.function_calls:
+                name = function_call.name
+                args = function_call.args
+                call_id = function_call.id
+                
+                if name in custom_tools:
+                    tool_output = custom_tools[name](**args)
+                    
+                    execution_history.append({
+                        "turn": current_turn + 1,
+                        "tool_called": name,
+                        "arguments": args,
+                        "raw_output": tool_output
+                    })
+                    
+                    function_responses.append(
+                        types.Part.from_function_response(
+                            name=name,
+                            response={"result": tool_output},
+                            id=call_id
+                        )
+                    )
+            
+            if function_responses:
+                response = chat.send_message(function_responses)
+            else:
+                break
+                
+            current_turn += 1
             
-        return {"agent_response": ai_response}
+        return {
+            "agent_response": response.text,
+            "execution_history": execution_history if execution_history else None
+        }
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))