Update app.py

app.py

@@ -11,11 +11,11 @@ API_ENDPOINT = "https://gen.pollinations.ai/v1/chat/completions"
 API_KEY = "sk_qDYtDvntvqzjmp23XKVzr9lBXYwYndaR"
 AI_MODEL = "claude"
 
-st.set_page_config(page_title="Auto-Upgrade Builder", layout="centered")
-st.title("🛡️ AI Builder + Auto-Upgrade Engine")
-st.markdown("Agar project purana hua to system khud usay modernize karega.")
+st.set_page_config(page_title="God Mode AI Builder", layout="centered")
+st.title("⚡ God Mode AI Android Builder")
+st.markdown("Ye system haar nahi maanega. Engine Upgrade + Full AI Surgery enabled.")
 
-# --- HELPER: AI API ---
+# --- HELPER: AI API (Robust) ---
 def ask_claude(system_prompt, user_message):
     headers = {
         "Content-Type": "application/json",
@@ -30,12 +30,12 @@ def ask_claude(system_prompt, user_message):
         "temperature": 0.1
     }
     try:
-        response = requests.post(API_ENDPOINT, json=payload, headers=headers, timeout=60)
+        response = requests.post(API_ENDPOINT, json=payload, headers=headers, timeout=120)
         if response.status_code == 200:
             return True, response.json()['choices'][0]['message']['content']
-        return False, "API Error"
-    except:
-        return False, "Connection Error"
+        return False, f"API Error: {response.text}"
+    except Exception as e:
+        return False, f"Connection Error: {str(e)}"
 
 # --- HELPER: KEYSTORE ---
 def generate_keystore():
@@ -49,57 +49,80 @@ def generate_keystore():
         ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
     return keystore_path
 
-# --- 🛠️ THE MODERNIZER (JADOO) ---
-# Ye function puranay project ko zabardasti naya bana dega
-def force_upgrade_engine(project_root, status_box):
-    status_box.write("🔧 Project Engine bahut purana hai. Zabardasti Upgrade kar raha hun...")
+# --- 🛠️ STEP 1: FORCE ENGINE UPGRADE ---
+def force_modernize_engine(project_root, status_box):
+    status_box.write("🔧 Force-Updating Gradle Engine to 8.0...")
     
-    # 1. Force Gradle Wrapper to 8.0 (Compatible with Java 17)
+    # 1. Wrapper Update
     wrapper_props = None
     for root, dirs, files in os.walk(project_root):
         if "gradle-wrapper.properties" in files:
             wrapper_props = os.path.join(root, "gradle-wrapper.properties")
             break
-            
-    if wrapper_props:
-        new_props = """distributionBase=GRADLE_USER_HOME
+    
+    # Agar wrapper nahi bhi mila, tab bhi naya banao
+    if not wrapper_props:
+        os.makedirs(os.path.join(project_root, "gradle", "wrapper"), exist_ok=True)
+        wrapper_props = os.path.join(project_root, "gradle", "wrapper", "gradle-wrapper.properties")
+
+    new_props = """distributionBase=GRADLE_USER_HOME
 distributionPath=wrapper/dists
 distributionUrl=https\://services.gradle.org/distributions/gradle-8.0-bin.zip
 zipStoreBase=GRADLE_USER_HOME
 zipStorePath=wrapper/dists
 """
-        with open(wrapper_props, "w") as f:
-            f.write(new_props)
-        status_box.write("✅ Gradle Wrapper set to 8.0")
-        
-    # 2. Update Classpath in build.gradle
-    root_gradle = None
+    with open(wrapper_props, "w") as f: f.write(new_props)
+
+    # 2. Permissions Fix
+    subprocess.run(["chmod", "+x", os.path.join(project_root, "gradlew")])
+
+# --- 🧠 STEP 2: AI SURGERY (Build.gradle Fixer) ---
+def ai_fix_build_gradle(project_root, error_log, status_box):
+    status_box.write("🧠 AI is performing surgery on build.gradle...")
+    
+    # Find root build.gradle and app build.gradle
+    root_gradle = os.path.join(project_root, "build.gradle")
+    app_gradle = None
     for root, dirs, files in os.walk(project_root):
-        if "build.gradle" in files and "app" not in root: # Root level
-            root_gradle = os.path.join(root, "build.gradle")
+        if "build.gradle" in files and "app" in root:
+            app_gradle = os.path.join(root, "build.gradle")
             break
-            
-    if root_gradle:
-        with open(root_gradle, "r") as f: content = f.read()
-        # Regex to replace old classpath with new one
-        new_content = re.sub(r"classpath\s+['\"]com.android.tools.build:gradle:.*?['\"]", 
-                             "classpath 'com.android.tools.build:gradle:8.0.0'", content)
+    
+    files_to_fix = []
+    if os.path.exists(root_gradle): files_to_fix.append(root_gradle)
+    if app_gradle: files_to_fix.append(app_gradle)
+    
+    for gradle_file in files_to_fix:
+        with open(gradle_file, 'r', errors='ignore') as f: content = f.read()
         
-        # Add google() if missing
-        if "google()" not in new_content:
-            new_content = new_content.replace("repositories {", "repositories {\n        google()\n        mavenCentral()")
-            
-        with open(root_gradle, "w") as f: f.write(new_content)
-        status_box.write("✅ Build Plugin updated to 8.0.0")
+        prompt = """
+        You are an Android Build Expert. The build failed on Java 17 + Gradle 8.
+        Fix the 'build.gradle' file content.
+        
+        RULES:
+        1. If it's Root gradle: Use 'classpath "com.android.tools.build:gradle:8.0.0"'
+        2. If it's App gradle: Ensure 'namespace' is present (if missing, use package name from manifest logic or generic 'com.example.app').
+        3. Remove deprecated 'jcenter()', use 'mavenCentral()' and 'google()'.
+        4. Output ONLY the code.
+        """
+        
+        user_msg = f"Error Logs:\n{error_log[-1000:]}\n\nFile Content:\n{content}"
+        
+        success, new_code = ask_claude(prompt, user_msg)
+        
+        if success:
+            clean_code = new_code.replace("```gradle", "").replace("```groovy", "").replace("```", "").strip()
+            with open(gradle_file, 'w') as f: f.write(clean_code)
+            status_box.write(f"✅ AI Rewrote {os.path.basename(gradle_file)}")
 
 # --- MAIN LOGIC ---
 uploaded_file = st.file_uploader("Project ZIP Upload", type=["zip"])
 
 if uploaded_file:
-    if st.button("🚀 Start Auto-Build"):
-        status_box = st.status("Initializing...", expanded=True)
+    if st.button("🚀 Start God Mode Build"):
+        status_box = st.status("Initializing God Mode...", expanded=True)
         
-        # 1. EXTRACT
+        # EXTRACT
         if os.path.exists("project_extract"): shutil.rmtree("project_extract")
         os.makedirs("project_extract", exist_ok=True)
         with open("uploaded.zip", "wb") as f: f.write(uploaded_file.getbuffer())
@@ -116,66 +139,60 @@ if uploaded_file:
             st.stop()
 
         subprocess.run(["chmod", "+x", os.path.join(project_root, "gradlew")])
-
-        # 2. ATTEMPT 1: NORMAL BUILD
-        status_box.write("🔨 Attempt 1: Building with default settings...")
+        
+        # === ATTEMPT 1: RAW BUILD ===
+        status_box.write("🔨 Attempt 1: Initial Build...")
         cmd = ["./gradlew", "clean", "assembleRelease", "--no-daemon", "--stacktrace"]
         result = subprocess.run(cmd, cwd=project_root, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
         
-        success = False
+        final_success = False
+        
         if result.returncode == 0:
-            success = True
+            final_success = True
         else:
-            # 3. FAILURE DETECTED -> UPGRADE ENGINE -> ATTEMPT 2
-            status_box.write("❌ Attempt 1 Failed. Checking if engine is old...")
+            # === ATTEMPT 2: FORCE ENGINE UPGRADE ===
+            status_box.write("❌ Attempt 1 Failed. Force-Upgrading Engine...")
+            force_modernize_engine(project_root, status_box)
+            
+            status_box.write("🔄 Attempt 2: Rebuilding with Gradle 8...")
+            result = subprocess.run(cmd, cwd=project_root, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
             
-            # Agar 'Daemon' ya 'Java' ka error hai to Modernizer chalao
-            logs = result.stderr
-            if "Daemon" in logs or "Java" in logs or "Minimum supported Gradle" in logs:
-                force_upgrade_engine(project_root, status_box)
+            if result.returncode == 0:
+                final_success = True
+            else:
+                # === ATTEMPT 3: AI SURGERY (NO IF/ELSE CHECK - ALWAYS RUN) ===
+                status_box.write("❌ Attempt 2 Failed. Activating Claude AI Surgery...")
+                ai_fix_build_gradle(project_root, result.stderr, status_box)
                 
-                status_box.write("🔄 Attempt 2: Rebuilding with NEW Engine...")
+                status_box.write("🔄 Attempt 3: Final Rebuild after AI Fix...")
                 result = subprocess.run(cmd, cwd=project_root, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
                 
                 if result.returncode == 0:
-                    success = True
+                    final_success = True
                 else:
-                    # 4. STILL FAILING -> USE AI FOR CODE ERRORS -> ATTEMPT 3
-                    status_box.write("❌ Attempt 2 Failed. Now checking Code Errors with AI...")
-                    
+                    # === LAST RESORT: Try to find code error ===
                     logs = result.stderr
-                    # Find broken file from logs
-                    match = re.search(r'(/[\w/-]+\.(java|kt|xml|gradle)):(\d+):', logs)
+                    match = re.search(r'(/[\w/-]+\.(java|kt|xml)):(\d+):', logs)
                     if match:
                         broken_file = match.group(1)
-                        # Fix path
                         if not os.path.exists(broken_file):
-                            for r, d, f in os.walk(project_root):
+                             for r, d, f in os.walk(project_root):
                                 if os.path.basename(broken_file) in f:
                                     broken_file = os.path.join(r, os.path.basename(broken_file))
                                     break
-                                    
+                        
                         if os.path.exists(broken_file):
-                            with open(broken_file, 'r') as f: code = f.read()
-                            status_box.write(f"🤖 AI Fixing: {os.path.basename(broken_file)}")
-                            
-                            prompt = "Fix this compile error strictly. Return only code."
-                            msg = f"Error: {logs[-1000:]}\n\nCode:\n{code}"
-                            ok, fixed_code = ask_claude(prompt, msg)
-                            
+                            status_box.write(f"🚑 Emergency Fix on: {os.path.basename(broken_file)}")
+                            with open(broken_file, 'r') as f: c = f.read()
+                            ok, fix = ask_claude("Fix code error. Return ONLY code.", f"Error: {logs[-500:]}\nCode: {c}")
                             if ok:
-                                clean_code = fixed_code.replace("```java","").replace("```","").strip()
-                                with open(broken_file, 'w') as f: f.write(clean_code)
-                                status_box.write("✅ Code Fixed. Final Retry...")
-                                
+                                with open(broken_file, 'w') as f: f.write(fix.replace("```java","").replace("```","").strip())
                                 result = subprocess.run(cmd, cwd=project_root, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
-                                if result.returncode == 0: success = True
+                                if result.returncode == 0: final_success = True
 
-        # 5. FINAL RESULT
-        if success:
+        # === FINISH ===
+        if final_success:
             status_box.update(label="✅ Build Successful! Signing...", state="complete")
-            
-            # Find APK
             target_apk = None
             for root, dirs, files in os.walk(project_root):
                 for file in files:
@@ -185,15 +202,15 @@ if uploaded_file:
             
             if target_apk:
                 keystore = generate_keystore()
-                signed = "Auto_Fixed_App.apk"
+                signed = "GodMode_App.apk"
                 subprocess.run(["zipalign", "-v", "-p", "4", target_apk, "aligned.apk"], stdout=subprocess.DEVNULL)
                 subprocess.run(["apksigner", "sign", "--ks", keystore, "--ks-pass", "pass:android", "--out", signed, "aligned.apk"])
                 
                 with open(signed, "rb") as f:
-                    st.download_button("⬇️ Download 100% Working APK", f, file_name=signed)
+                    st.download_button("⬇️ Download APK", f, file_name=signed)
         else:
-            status_box.update(label="❌ Fatal Error", state="error")
-            st.error("System could not auto-fix this project.")
-            st.text_area("Last Logs", result.stderr[-2000:], height=300)
+            status_box.update(label="❌ Mission Failed", state="error")
+            st.error("Even God Mode couldn't fix this project.")
+            st.text_area("Final Logs", result.stderr[-2000:], height=300)
 
         if os.path.exists("uploaded.zip"): os.remove("uploaded.zip")