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=["*"],
@@ -27,18 +28,29 @@ def execute_in_sandbox(command: str):
         # Saari commands hamesha /tmp folder me execute hongi jahan explicit write permissions hain
         full_command = f"cd /tmp && {command}"
         
+        # Hugging Face ke saare Environment Variables (tokens, paths) ko sub-process me pass karne ke liye
+        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",  # Large logs ya binary text par crash hone se bachata hai
+            timeout=300,       # 5 minutes optimized timeout taaki HF gateway connection drop na kare
+            env=env
         )
         return {
             "stdout": result.stdout, 
             "stderr": result.stderr, 
             "exit_code": result.returncode
         }
+    except subprocess.TimeoutExpired:
+        return {
+            "stdout": "", 
+            "stderr": "Execution timed out! Command took longer than 5 minutes to prevent server freeze.", 
+            "exit_code": -1
+        }
     except Exception as e:
         return {"error": str(e)}
 
@@ -46,6 +58,7 @@ def execute_in_sandbox(command: str):
 async def agent_chat(request: ChatRequest):
     user_prompt = request.message
     
+    # SYSTEM INSTRUCTION: Gofile rules ko nikaal kar ise ekdum dynamic aur general bana diya hai
     system_instruction = (
         "You are an expert AI Agent Builder operating inside a fully equipped Linux sandbox.\n\n"
         "YOUR DEVELOPMENT ENVIRONMENT capabilities:\n"
@@ -53,31 +66,68 @@ async def agent_chat(request: ChatRequest):
         "- **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'."
+        "GENERAL EXECUTION & UPLOAD RULES:\n"
+        "- You have full internet access inside the sandbox. You can download dependencies, interact with external APIs, or upload compilation artifacts (APKs, files, logs) to any hosting provider using tools like 'curl' or custom Python scripts.\n"
+        "- If the user asks you to upload a file to a specific platform (e.g., gofile.io, catbox, etc.) or perform an external operation, use your general knowledge to draft the correct API calls or bash routines using available environment variables.\n\n"
+        "OUTPUT FORMATTING:\n"
+        "If you want to create a project, write python scripts, compile an application, or run diagnostics, output the exact script inside a single triple backtick block starting with 'bash'.\n"
+        "Analyze the tool execution results carefully in each turn. If a command fails or throws an error, analyze the output, debug it, fix your script, and re-run it until the task is successfully completed. Once done, present a clean summary or download links to the user."
     )
     
     try:
-        response = client.models.generate_content(
+        # Multi-turn Chat session taaki AI background me tools chala sake
+        chat = client.chats.create(
             model='gemini-2.5-flash',
-            contents=user_prompt,
             config=types.GenerateContentConfig(system_instruction=system_instruction)
         )
         
+        # Turn 1: User ka input bhejte hain
+        response = chat.send_message(user_prompt)
         ai_response = response.text
         
-        if "```bash" in ai_response:
+        max_loops = 4  # Infinite loops aur HF memory crash se bachne ke liye safe limit
+        current_loop = 0
+        execution_history = []
+        
+        # SOLID AGENT LOOP: Jab tak AI bash block dega, tab tak loop chalega
+        while "```bash" in ai_response and current_loop < max_loops:
             parts = ai_response.split("```bash")
             if len(parts) > 1:
                 command = parts[1].split("```")[0].strip()
+                
+                # Command ko sandbox me run karein
                 exec_result = execute_in_sandbox(command)
-                return {"agent_response": ai_response, "execution_result": exec_result}
+                
+                # History maintain karne ke liye (Aapke UI/logs ke liye)
+                execution_history.append({
+                    "step": current_loop + 1,
+                    "command_executed": command,
+                    "result": exec_result
+                })
+                
+                # Safe feedback string bana rahe hain bina kisi memory leakage ke
+                feedback = (
+                    f"Execution Result for Step {current_loop + 1}:\n"
+                    f"Exit Code: {exec_result.get('exit_code')}\n"
+                    f"Stdout: {exec_result.get('stdout')}\n"
+                    f"Stderr: {exec_result.get('stderr')}\n"
+                )
+                if "error" in exec_result:
+                    feedback += f"System Error: {exec_result['error']}\n"
+                
+                feedback += "\nAnalyze this output. If the task succeeded, provide the final response or download links. If it failed, fix the code and output a new bash block to retry."
+                
+                # AI ko feedback bhej kar agla turn check karte hain
+                response = chat.send_message(feedback)
+                ai_response = response.text
+                
+            current_loop += 1
             
-        return {"agent_response": ai_response}
+        # UI par clean output dikhane ke liye execution history aur final text ko alag kar diya
+        return {
+            "agent_response": ai_response,
+            "execution_history": execution_history if execution_history else None
+        }
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))