Create app.py

app.py

@@ -0,0 +1,244 @@
+import streamlit as st
+import os
+import zipfile
+import subprocess
+import shutil
+import requests
+import re
+
+# --- CONFIGURATION ---
+API_ENDPOINT = "https://gen.pollinations.ai/v1/chat/completions"
+API_KEY = "sk_qDYtDvntvqzjmp23XKVzr9lBXYwYndaR"  # Apki di hui key
+AI_MODEL = "claude"
+
+# Page Config
+st.set_page_config(page_title="AI Android Builder", layout="centered")
+st.title("🤖 AI-Powered Android APK Generator")
+st.markdown("Pollinations AI (Claude) check karega, build karega, aur error aya to khud fix karega.")
+
+# --- HELPER: AI API CALL ---
+def ask_claude(system_prompt, user_message):
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": f"Bearer {API_KEY}"
+    }
+    payload = {
+        "model": AI_MODEL,
+        "messages": [
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_message}
+        ],
+        "temperature": 0.2  # Low temperature for precise code
+    }
+    
+    try:
+        response = requests.post(API_ENDPOINT, json=payload, headers=headers, timeout=60)
+        if response.status_code == 200:
+            return response.json()['choices'][0]['message']['content']
+        else:
+            return f"Error: {response.text}"
+    except Exception as e:
+        return f"Connection Error: {str(e)}"
+
+# --- HELPER: KEYSTORE ---
+def generate_keystore():
+    keystore_path = "debug.keystore"
+    if not os.path.exists(keystore_path):
+        cmd = [
+            "keytool", "-genkey", "-v", "-keystore", keystore_path, 
+            "-storepass", "android", "-alias", "androiddebugkey", 
+            "-keypass", "android", "-keyalg", "RSA", "-keysize", "2048", 
+            "-validity", "10000", "-dname", "CN=AndroidDebug,O=Android,C=US"
+        ]
+        subprocess.run(cmd, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
+    return keystore_path
+
+# --- AI LOGIC: PRE-CHECK ---
+def check_project_health(project_root):
+    # Sirf critical files parho taake AI confuse na ho
+    manifest_path = None
+    gradle_path = None
+    
+    for root, dirs, files in os.walk(project_root):
+        if "AndroidManifest.xml" in files:
+            manifest_path = os.path.join(root, "AndroidManifest.xml")
+        if "build.gradle" in files and "app" in root: # App level gradle
+            gradle_path = os.path.join(root, "build.gradle")
+            
+    if not manifest_path or not gradle_path:
+        return "❌ Critical files (Manifest/Gradle) missing. Cannot proceed."
+
+    with open(manifest_path, 'r', encoding='utf-8', errors='ignore') as f:
+        manifest_content = f.read()
+    with open(gradle_path, 'r', encoding='utf-8', errors='ignore') as f:
+        gradle_content = f.read()
+
+    prompt = f"""
+    You are an Android Build Expert. Check these files for critical syntax errors that would stop a build.
+    Do NOT change functionality. Just check if structure is valid.
+    
+    Manifest:
+    {manifest_content[:2000]}
+    
+    Gradle:
+    {gradle_content[:2000]}
+    
+    Reply ONLY with "SAFE" if it looks okay, or "ISSUE: [Explanation]" if bad.
+    """
+    return ask_claude("Analyze code structure.", prompt)
+
+# --- AI LOGIC: AUTO FIX ---
+def attempt_ai_repair(error_log, project_root):
+    # 1. Error log mein se file ka path dhoondo
+    # Format usually: /app/src/main/java/com/example/App.java:15: error: ...
+    match = re.search(r'(/[\w/-]+\.(java|kt|xml|gradle)):(\d+):', error_log)
+    
+    if not match:
+        return False, "AI could not identify the broken file in logs."
+        
+    broken_file_path = match.group(1)
+    
+    # Path correction (Absolute vs Relative)
+    if not os.path.exists(broken_file_path):
+        # Try to find it relative to project root
+        found = False
+        file_name = os.path.basename(broken_file_path)
+        for root, dirs, files in os.walk(project_root):
+            if file_name in files:
+                broken_file_path = os.path.join(root, file_name)
+                found = True
+                break
+        if not found:
+            return False, f"File {file_name} not found."
+
+    # 2. Read Broken Code
+    with open(broken_file_path, 'r', encoding='utf-8', errors='ignore') as f:
+        broken_code = f.read()
+        
+    # 3. Ask AI to fix
+    st.info(f"🤖 AI is fixing: **{os.path.basename(broken_file_path)}**")
+    
+    system_prompt = "You are an expert Android Developer. Fix the code error. Maintain original functionality. Output ONLY the full corrected code block without markdown wrappers."
+    user_msg = f"""
+    The build failed with this error:
+    {error_log[:1000]}
+    
+    Here is the broken file content:
+    {broken_code}
+    
+    Return the FULL CORRECTED file content.
+    """
+    
+    fixed_code = ask_claude(system_prompt, user_msg)
+    
+    # Clean AI output (remove ```java etc)
+    fixed_code = fixed_code.replace("```java", "").replace("```kotlin", "").replace("```xml", "").replace("```", "").strip()
+    
+    # 4. Overwrite file
+    try:
+        with open(broken_file_path, 'w', encoding='utf-8') as f:
+            f.write(fixed_code)
+        return True, f"Fixed {os.path.basename(broken_file_path)}"
+    except Exception as e:
+        return False, str(e)
+
+# --- MAIN UI ---
+uploaded_file = st.file_uploader("Project ZIP Upload", type=["zip"])
+
+if uploaded_file:
+    if st.button("🚀 AI Check & Build"):
+        status_box = st.status("Initializing...", expanded=True)
+        
+        # 1. EXTRACT
+        status_box.write("📂 Extracting files...")
+        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())
+        with zipfile.ZipFile("uploaded.zip", "r") as z: z.extractall("project_extract")
+        
+        # Root Find
+        project_root = None
+        for root, dirs, files in os.walk("project_extract"):
+            if "gradlew" in files:
+                project_root = root
+                break
+        
+        if not project_root:
+            status_box.update(label="❌ Invalid Project", state="error")
+            st.error("Gradlew not found.")
+            st.stop()
+            
+        subprocess.run(["chmod", "+x", os.path.join(project_root, "gradlew")])
+
+        # 2. AI PRE-CHECK
+        status_box.write("🧠 Claude is analyzing project structure...")
+        ai_verdict = check_project_health(project_root)
+        
+        if "SAFE" in ai_verdict or "manifest" in ai_verdict.lower(): # Loose check
+            status_box.write("✅ AI Check Passed.")
+        else:
+            status_box.write(f"⚠️ AI Warning: {ai_verdict}")
+            # Hum rukenge nahi, try karenge, shayad build ho jaye
+
+        # 3. BUILD PROCESS LOOP
+        build_success = False
+        attempts = 0
+        max_attempts = 2  # 1 Normal + 1 AI Repair
+        
+        while attempts < max_attempts:
+            status_box.write(f"🔨 Building APK (Attempt {attempts+1})...")
+            
+            cmd = ["./gradlew", "assembleRelease", "--no-daemon", "--stacktrace"]
+            result = subprocess.run(cmd, cwd=project_root, stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)
+            
+            if result.returncode == 0:
+                build_success = True
+                break
+            else:
+                attempts += 1
+                if attempts < max_attempts:
+                    status_box.write("❌ Build Failed. Activating AI Repair Agent...")
+                    success, msg = attempt_ai_repair(result.stderr, project_root)
+                    if success:
+                        status_box.write(f"✅ {msg}. Retrying build...")
+                    else:
+                        status_box.write(f"⚠️ AI Repair Failed: {msg}")
+                        break
+        
+        # 4. RESULT HANDLING
+        if build_success:
+            status_box.write("✅ Build Successful! Signing APK...")
+            
+            # Search APK
+            unsigned_apk = None
+            for root, dirs, files in os.walk(project_root):
+                for file in files:
+                    if file.endswith(".apk") and "release" in file:
+                        unsigned_apk = os.path.join(root, file)
+                        break
+            
+            if unsigned_apk:
+                # SIGNING
+                keystore = generate_keystore()
+                signed_name = "AI_Generated_App.apk"
+                subprocess.run(["zipalign", "-v", "-p", "4", unsigned_apk, "aligned.apk"], stdout=subprocess.DEVNULL)
+                
+                subprocess.run([
+                    "apksigner", "sign", "--ks", keystore, "--ks-pass", "pass:android",
+                    "--out", signed_name, "aligned.apk"
+                ])
+                
+                status_box.update(label="🎉 Process Complete!", state="complete")
+                
+                with open(signed_name, "rb") as f:
+                    st.download_button("⬇️ Download AI Verified APK", f, file_name=signed_name, mime="application/vnd.android.package-archive")
+            else:
+                status_box.update(label="❌ File Missing", state="error")
+                st.error("APK generated but file not found.")
+        else:
+            status_box.update(label="❌ Fatal Error", state="error")
+            st.error("Build Failed even after AI Repair.")
+            st.text_area("Last Error Log", result.stderr[-2000:], height=300)
+
+        # Cleanup
+        if os.path.exists("uploaded.zip"): os.remove("uploaded.zip")