Upload folder using huggingface_hub

frontend.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import requests
+import subprocess
 import time
 import os
 
@@ -10,119 +11,99 @@ def get_status():
         r = requests.get(f"{NODE_URL}/health", timeout=2)
         return r.json()
     except Exception as e:
-        return {"whatsappStatus": "Offline", "connected": False, "ready": False, "logs": [f"Error connecting to backend: {str(e)}"]}
+        return {"whatsappStatus": "Offline", "connected": False, "ready": False, "logs": [str(e)]}
 
 def update_ui():
     status = get_status()
+    msg = f"Status: {status.get('whatsappStatus', 'Unknown')}\nConnected: {status.get('connected', False)}\nReady: {status.get('ready', False)}"
     
-    # Connection info
-    status_msg = status.get('whatsappStatus', 'Unknown')
-    connected = "✅ Yes" if status.get('connected') else "❌ No"
-    ready = "🚀 Ready" if status.get('ready') else "⏳ Initializing"
-    
-    info_text = f"WhatsApp Status: {status_msg}\nConnected: {connected}\nReady: {ready}"
-    
-    # QR Code handling
-    qr_html = "<div style='text-align:center; padding:40px; color:#888;'>Waiting for WhatsApp to initialize...</div>"
+    qr_html = "<div style='text-align:center; padding:20px; color:#666;'>QR code will appear here.</div>"
     if status.get('hasQR'):
         try:
             qr_r = requests.get(f"{NODE_URL}/qr-data", timeout=1)
             if qr_r.status_code == 200:
                 qr_data = qr_r.json()
-                ascii_qr = qr_data.get('ascii', '')
+                ascii_qr = qr_data['ascii']
                 qr_html = f"""
-                <div style='background:white; color:black; padding:20px; display:flex; flex-direction:column; align-items:center; border-radius:12px; box-shadow: 0 8px 24px rgba(0,0,0,0.2);'>
-                    <h3 style='margin:0 0 15px 0; font-family:sans-serif; color:#128c7e;'>Scan with WhatsApp</h3>
-                    <pre style='font-size:7px; line-height:1; margin:0; font-family:monospace; font-weight:bold; letter-spacing:0;'>{ascii_qr}</pre>
-                    <p style='margin-top:15px; font-size:13px; font-family:sans-serif;'>Open WhatsApp > Linked Devices > Link a Device</p>
+                <div style='background:white; color:black; padding:15px; display:flex; flex-direction:column; align-items:center; border-radius:8px; box-shadow: 0 4px 12px rgba(0,0,0,0.15);'>
+                    <h3 style='margin:0 0 10px 0; font-family:sans-serif;'>Scan to Login</h3>
+                    <pre style='font-size:7px; line-height:1; margin:0; font-family:monospace; font-weight:bold;'>{ascii_qr}</pre>
                 </div>
                 """
         except:
             pass
     elif status.get('ready'):
-        qr_html = """
-        <div style='text-align:center; padding:40px; color:#4caf50; background:#f0f9f0; border-radius:12px; border:2px dashed #4caf50;'>
-            <h2 style='margin:0;'>✅ Connected!</h2>
-            <p>You can now use the API to send messages.</p>
-        </div>
-        """
+        qr_html = "<div style='text-align:center; padding:20px; color:#4caf50; font-weight:bold;'>WhatsApp is connected!</div>"
         
     logs = "\n".join(status.get('logs', []))
     
-    # Handle screenshot as local file to bypass Gradio SSRF protection
     screenshot_path = None
     try:
         sc_r = requests.get(f"{NODE_URL}/screenshot", timeout=5)
         if sc_r.status_code == 200:
-            screenshot_path = "current_debug_screen.png"
+            screenshot_path = "current_screen.png"
             with open(screenshot_path, "wb") as f:
                 f.write(sc_r.content)
     except:
         pass
     
-    return info_text, qr_html, logs, screenshot_path
+    return msg, qr_html, logs, screenshot_path
 
 def trigger_init():
     try:
         requests.post(f"{NODE_URL}/init")
-        return "Initialization command sent."
+        return "Initialization triggered."
     except Exception as e:
         return f"Error: {str(e)}"
 
-def send_poll(number, title, options):
+def trigger_reload():
     try:
-        opts = [o.strip() for o in options.split(",")]
-        r = requests.post(f"{NODE_URL}/send-poll", json={"telnumber": number, "name": title, "choices": opts})
-        return r.json()
+        requests.post(f"{NODE_URL}/reload")
+        return "Reload command sent."
     except Exception as e:
-        return {"error": str(e)}
+        return f"Error: {str(e)}"
 
-def send_message(number, message):
+def send_poll(number, title, options):
     try:
-        r = requests.post(f"{NODE_URL}/send-message", json={"telnumber": number, "message": message})
+        opts = [o.strip() for o in options.split(",")]
+        r = requests.post(f"{NODE_URL}/send-poll", json={"telnumber": number, "name": title, "choices": opts})
         return r.json()
     except Exception as e:
         return {"error": str(e)}
 
 with gr.Blocks(title="WPPConnect Dashboard") as demo:
-    gr.Markdown("# 🤖 WPPConnect WhatsApp Control Center")
+    gr.Markdown("# 🤖 WPPConnect WhatsApp Control Panel")
     
     with gr.Row():
         with gr.Column(scale=1):
-            status_display = gr.Label(label="Session Info")
-            init_btn = gr.Button("🚀 Start/Restart Session", variant="primary")
+            status_display = gr.Label(label="System Status")
+            with gr.Row():
+                init_btn = gr.Button("🚀 Start/Retry Session", variant="primary")
+                reload_btn = gr.Button("🔄 Force Reload Page", variant="secondary")
             refresh_btn = gr.Button("🔄 Refresh UI")
         
         with gr.Column(scale=2):
-            qr_display = gr.HTML(label="Connection Status / QR Code")
+            qr_display = gr.HTML(label="Login QR Code")
     
     with gr.Tabs():
         with gr.TabItem("📊 Send Poll"):
             with gr.Row():
                 poll_num = gr.Textbox(label="Phone Number", placeholder="5511999999999")
                 poll_title = gr.Textbox(label="Poll Question")
-            poll_opts = gr.Textbox(label="Choices (comma separated)", placeholder="Option 1, Option 2, Option 3")
-            poll_send = gr.Button("Send Poll", variant="primary")
+            poll_opts = gr.Textbox(label="Choices (comma separated)", placeholder="Yes, No, Maybe")
+            poll_send = gr.Button("Send Poll Now", variant="primary")
             poll_out = gr.JSON(label="Result")
             
-        with gr.TabItem("💬 Send Text"):
-            with gr.Row():
-                msg_num = gr.Textbox(label="Phone Number")
-                msg_txt = gr.Textbox(label="Message", lines=2)
-            msg_btn = gr.Button("Send Message")
-            msg_out = gr.JSON(label="Result")
-            
-        with gr.TabItem("📋 Debug Logs"):
-            log_display = gr.Code(label="Backend Logs", lines=15, language="markdown")
+        with gr.TabItem("📋 System Logs"):
+            log_display = gr.Code(label="Recent Logs", lines=15, language="markdown")
             
-        with gr.TabItem("🔍 Live Browser"):
-            gr.Markdown("Visual debugging of the underlying browser page.")
-            screenshot_display = gr.Image(label="Browser Screenshot")
+        with gr.TabItem("🔍 Browser Debug"):
+            screenshot_display = gr.Image(label="Live Browser View")
 
     init_btn.click(trigger_init, outputs=status_display)
+    reload_btn.click(trigger_reload, outputs=status_display)
     refresh_btn.click(update_ui, outputs=[status_display, qr_display, log_display, screenshot_display])
     poll_send.click(send_poll, inputs=[poll_num, poll_title, poll_opts], outputs=poll_out)
-    msg_btn.click(send_message, inputs=[msg_num, msg_txt], outputs=msg_out)
     
     try:
         demo.load(update_ui, outputs=[status_display, qr_display, log_display, screenshot_display], every=10)
@@ -130,4 +111,4 @@ with gr.Blocks(title="WPPConnect Dashboard") as demo:
         demo.load(update_ui, outputs=[status_display, qr_display, log_display, screenshot_display])
 
 if __name__ == "__main__":
-    demo.launch(server_name="0.0.0.0", server_port=7860, theme=gr.themes.Soft(primary_hue="green"))
+    demo.launch(server_name="0.0.0.0", server_port=7860)

src/app.ts

@@ -18,7 +18,7 @@ const log = (msg: string) => {
   const entry = `${new Date().toISOString()} - ${msg}`;
   console.log(entry);
   appLogs.push(entry);
-  if (appLogs.length > 200) appLogs.shift();
+  if (appLogs.length > 100) appLogs.shift();
 };
 
 app.use(express.json());
@@ -40,7 +40,7 @@ app.get('/qr-data', (req, res) => {
 });
 
 app.get('/screenshot', async (req, res) => {
-  if (whatsappClient && whatsappClient.page) {
+  if (whatsappClient && whatsappClient.page && !whatsappClient.page.isClosed()) {
     try {
       const screenshot = await whatsappClient.page.screenshot({ encoding: 'base64' });
       const img = Buffer.from(screenshot as string, 'base64');
@@ -50,7 +50,7 @@ app.get('/screenshot', async (req, res) => {
       res.status(500).json({ error: e.message });
     }
   } else {
-    res.status(404).json({ error: 'Browser not active' });
+    res.status(404).json({ error: 'Browser not active or page closed' });
   }
 });
 
@@ -60,13 +60,18 @@ app.post('/init', (req, res) => {
   res.json({ success: true });
 });
 
-app.post('/send-message', async (req, res) => {
-  const { telnumber, message } = req.body;
-  if (!whatsappClient || !isReady) return res.status(503).json({ error: 'WhatsApp not ready' });
-  try {
-    const result = await whatsappClient.sendText(`${telnumber}@c.us`, message);
-    res.json({ success: true, result });
-  } catch (e: any) { res.status(500).json({ error: e.message }); }
+app.post('/reload', async (req, res) => {
+  if (whatsappClient && whatsappClient.page) {
+    try {
+      await whatsappClient.page.reload();
+      log('Page reloaded manually');
+      res.json({ success: true });
+    } catch (e: any) {
+      res.status(500).json({ error: e.message });
+    }
+  } else {
+    res.status(404).json({ error: 'Client not active' });
+  }
 });
 
 app.post('/send-poll', async (req, res) => {
@@ -84,8 +89,7 @@ async function startWPP() {
   statusMsg = 'Initializing...';
   log('Starting WPPConnect init...');
   
-  // Try to get IP for web.whatsapp.com manually
-  let waIP = '157.240.22.60'; 
+  let waIP = '31.13.66.56'; 
   try {
     const resolver = new Resolver();
     resolver.setServers(['8.8.8.8', '1.1.1.1']);
@@ -100,11 +104,9 @@ async function startWPP() {
     log(`DNS failure, using fallback IP ${waIP}`);
   }
 
-  // Cleanup tokens to avoid locks
   try {
     const tokensDir = path.join(process.cwd(), 'tokens');
     if (fs.existsSync(tokensDir)) fs.rmSync(tokensDir, { recursive: true, force: true });
-    log('Cleaned tokens directory');
   } catch (e) {}
 
   try {
@@ -133,6 +135,7 @@ async function startWPP() {
         ],
       },
       autoClose: 0,
+      deviceSyncTimeout: 0, // Prevent timeout during slow sync
       updatesLog: false,
       waitForLogin: false,
     });
@@ -147,6 +150,5 @@ async function startWPP() {
 
 app.listen(port, () => {
   log(`Node Backend started on port ${port}`);
-  // Initial start after 5 seconds
   setTimeout(startWPP, 5000);
 });