Update app.py

app.py

@@ -1,32 +1,17 @@
 import os
+import shutil
+import subprocess
 import asyncio
 import spaces
 import torch
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
 from threading import Thread
-from playwright.async_api import async_playwright
-from playwright_stealth import stealth
 
-# --- Playwright Installation Check ---
-try:
-    import playwright
-except ImportError:
-    os.system("pip install playwright playwright-stealth")
-    
-# Space startup par browser install karein
-os.system("playwright install chromium")
-
-# --- AI Model Setup ---
-# Qwen 2.5 1.5B - Best balance for ZeroGPU RAM
+# --- AI Setup ---
 MODEL_ID = "Qwen/Qwen2.5-1.5B-Instruct"
-
 tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
-model = AutoModelForCausalLM.from_pretrained(
-    MODEL_ID,
-    torch_dtype=torch.bfloat16, # torch_dtype fix
-    device_map=None # Manual control
-).to("cpu")
+model = AutoModelForCausalLM.from_pretrained(MODEL_ID, torch_dtype=torch.bfloat16, device_map=None).to("cpu")
 
 @spaces.GPU
 def chat_fn(message, history):
@@ -36,19 +21,11 @@ def chat_fn(message, history):
         messages.append({"role": "assistant", "content": h[1]})
     messages.append({"role": "user", "content": message})
     
-    model.to("cuda") # Move to GPU only when needed
-    
-    input_ids = tokenizer.apply_chat_template(
-        messages, 
-        tokenize=True, 
-        add_generation_prompt=True, 
-        return_tensors="pt"
-    ).to("cuda")
-    
+    model.to("cuda")
+    input_ids = tokenizer.apply_chat_template(messages, tokenize=True, add_generation_prompt=True, return_tensors="pt").to("cuda")
     streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
-    gen_kwargs = dict(input_ids=input_ids, streamer=streamer, max_new_tokens=1024, temperature=0.7)
     
-    thread = Thread(target=model.generate, kwargs=gen_kwargs)
+    thread = Thread(target=model.generate, kwargs=dict(input_ids=input_ids, streamer=streamer, max_new_tokens=512))
     thread.start()
     
     response = ""
@@ -56,72 +33,89 @@ def chat_fn(message, history):
         response += new_text
         yield response
 
-# --- Improved Browser Logic ---
-async def browser_task(url, selector=None, text=None, action="nav"):
-    async with async_playwright() as p:
-        # Browser launch with args to prevent sandbox crashes
-        browser = await p.chromium.launch(headless=True, args=["--no-sandbox", "--disable-setuid-sandbox"])
-        context = await browser.new_context(
-            viewport={'width': 1280, 'height': 720},
-            user_agent="Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/122.0.0.0 Safari/537.36"
-        )
-        page = await context.new_page()
-        await stealth(page)
-        
-        try:
-            # Timeout barha diya hai
-            await page.goto(url, wait_until="load", timeout=60000)
-            
-            if action == "click" and selector:
-                await page.wait_for_selector(selector, timeout=5000)
-                await page.click(selector)
-                await asyncio.sleep(2)
-            elif action == "type" and selector and text:
-                await page.wait_for_selector(selector, timeout=5000)
-                await page.fill(selector, text)
-                await page.keyboard.press("Enter")
-                await asyncio.sleep(3)
-            
-            screenshot = await page.screenshot()
-            await browser.close()
-            return screenshot
-        except Exception as e:
-            print(f"Browser Error: {e}")
-            await browser.close()
-            return None
+# --- React Preview Logic ---
+BUILD_DIR = "web_app"
+OUTPUT_DIR = "web_app/dist"
 
-def web_run(url, sel, txt, action):
+def handle_zip_upload(zip_file):
+    if zip_file is None:
+        return "Please upload a zip file."
+    
+    # 1. Cleanup old files
+    if os.path.exists(BUILD_DIR):
+        shutil.rmtree(BUILD_DIR)
+    os.makedirs(BUILD_DIR)
+    
     try:
-        return asyncio.run(browser_task(url, sel, txt, action))
+        # 2. Extract Zip
+        shutil.unpack_archive(zip_file.name, BUILD_DIR)
+        
+        # 3. Find where package.json is (sometimes it's inside a subfolder)
+        root_path = BUILD_DIR
+        for root, dirs, files in os.walk(BUILD_DIR):
+            if "package.json" in files:
+                root_path = root
+                break
+        
+        # 4. Install & Build
+        yield "Installing libraries (npm install)..."
+        subprocess.run(["npm", "install"], cwd=root_path, check=True)
+        
+        yield "Building project (npm run build)..."
+        subprocess.run(["npm", "run", "build"], cwd=root_path, check=True)
+        
+        # 5. Serve the 'dist' folder
+        dist_path = os.path.join(root_path, "dist")
+        if os.path.exists(dist_path):
+            # Gradio static access setup
+            static_url = f"/file={dist_path}/index.html"
+            yield f"### ✅ Build Successful!\nClick to preview below."
+        else:
+            yield "❌ Error: 'dist' folder not found after build."
+            
     except Exception as e:
-        return None
+        yield f"❌ Error: {str(e)}"
 
-# --- Custom UI with Gradio ---
-with gr.Blocks(theme=gr.themes.Default()) as demo:
-    gr.HTML("<h1 style='text-align:center;'>🚀 AI Agent + Web Stealth Browser</h1>")
+# --- Gradio UI Layout ---
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.HTML("<h1 style='text-align:center;'>Master AI: Chat + Browser + React Preview</h1>")
     
-    with gr.Row():
-        with gr.Column(scale=1):
-            # Standard Chatbot logic
-            chat_ui = gr.ChatInterface(fn=chat_fn, type="messages")
-            
-        with gr.Column(scale=1):
-            url_input = gr.Textbox(label="🌐 Website URL", value="https://google.com")
+    with gr.Tabs():
+        # TAB 1: AI Chat
+        with gr.TabItem("💬 AI Chatbot"):
+            gr.ChatInterface(fn=chat_fn, type="messages")
+        
+        # TAB 2: React Live Preview
+        with gr.TabItem("⚛️ React/Vite Preview"):
+            gr.Markdown("### Upload your Vite Project ZIP")
             with gr.Row():
-                sel_input = gr.Textbox(label="🔍 Selector (ID/Class)", placeholder="e.g. input[name='q']")
-                txt_input = gr.Textbox(label="⌨️ Text to Type", placeholder="Search query...")
+                zip_input = gr.File(label="Upload Project ZIP", file_types=[".zip"])
+                build_logs = gr.Markdown("Status: Waiting for upload...")
             
-            with gr.Row():
-                btn_visit = gr.Button("Visit", variant="primary")
-                btn_click = gr.Button("Click")
-                btn_type = gr.Button("Type & Enter")
+            build_btn = gr.Button("Build & Preview", variant="primary")
             
-            browser_img = gr.Image(label="Live View", type="numpy")
+            # Preview Iframe
+            # Note: In Hugging Face, serving local HTML is restricted, 
+            # we use a workaround to show the build result status.
+            preview_area = gr.HTML(label="Preview Screen")
+
+            def start_preview(file):
+                for status in handle_zip_upload(file):
+                    if "✅" in status:
+                        # Hum file path return karenge jo browser me render ho sake
+                        # Hugging Face par direct iframe preview ke liye 'dist' folder ko serve karna hota hai
+                        return status, "Build complete! Check your logs."
+                    return status, ""
+
+            build_btn.click(start_preview, inputs=zip_input, outputs=[build_logs, preview_area])
 
-    # Handlers
-    btn_visit.click(lambda u: web_run(u, None, None, "nav"), [url_input], browser_img)
-    btn_click.click(lambda u, s: web_run(u, s, None, "click"), [url_input, sel_input], browser_img)
-    btn_type.click(lambda u, s, t: web_run(u, s, t, "type"), [url_input, sel_input, txt_input], browser_img)
+        # TAB 3: Stealth Browser
+        with gr.TabItem("🌐 Stealth Browser"):
+            # (Pichla wala browser code yahan aayega)
+            gr.Markdown("Browser is active in background...")
+
+# Playwright installation
+os.system("playwright install chromium")
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.queue().launch(allowed_paths=["."]) # allowed_paths zaroori hai files serve karne ke liye