Upload folder using huggingface_hub

frontend.py

@@ -20,20 +20,21 @@ def get_status():
 def update_ui():
     status = get_status()
     if not status:
-        return "Backend Offline", "Waiting for Node.js to start...", "No logs available"
+        return "Backend Offline", "Waiting for Node.js...", ""
     
-    msg = f"Status: {status.get('whatsappStatus', 'Unknown')}\nConnected: {status.get('connected')}\nReady: {status.get('ready')}"
+    msg = f"Status: {status.get('whatsappStatus', 'Unknown')}\nConnected: {status.get('connected', False)}\nReady: {status.get('ready', False)}"
     
-    qr_html = "No QR code generated yet."
+    qr_html = "No QR code available yet. Click 'Start WhatsApp Session'."
     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['ascii']
             qr_html = f"""
-            <div style='background:white; color:black; padding:20px; display:flex; flex-direction:column; align-items:center;'>
-                <pre style='font-size:7px; line-height:1; margin:0; font-family:monospace;'>{ascii_qr}</pre>
-                <p style='margin-top:10px; font-weight:bold;'>Scan this code with your phone</p>
+            <div style='background:white; color:black; padding:15px; display:flex; flex-direction:column; align-items:center; border-radius:8px;'>
+                <h3 style='margin:0 0 10px 0;'>WhatsApp QR Code</h3>
+                <pre style='font-size:7px; line-height:1; margin:0; font-family:monospace; font-weight:bold;'>{ascii_qr}</pre>
+                <p style='margin-top:10px; font-size:12px;'>Scan this with your WhatsApp mobile app</p>
             </div>
             """
     except:
@@ -45,8 +46,8 @@ def update_ui():
 
 def trigger_init():
     try:
-        r = requests.post(f"{NODE_URL}/init")
-        return "Initialization requested. Click 'Refresh' in a few seconds."
+        requests.post(f"{NODE_URL}/init")
+        return "Initialization triggered. Waiting for QR..."
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -58,35 +59,51 @@ def send_poll(number, title, options):
     except Exception as e:
         return {"error": str(e)}
 
+def send_message(number, message):
+    try:
+        r = requests.post(f"{NODE_URL}/send-message", json={"telnumber": number, "message": message})
+        return r.json()
+    except Exception as e:
+        return {"error": str(e)}
+
 with gr.Blocks(theme=gr.themes.Default()) as demo:
-    gr.Markdown("# 🤖 WPPConnect WhatsApp Control")
+    gr.Markdown("# 🤖 WPPConnect WhatsApp Control Panel")
     
     with gr.Row():
         with gr.Column(scale=1):
-            status_display = gr.Textbox(label="System Status", lines=3)
-            init_btn = gr.Button("🚀 Start/Retry WhatsApp", variant="primary")
-            refresh_btn = gr.Button("🔄 Refresh Status/QR")
+            status_display = gr.Textbox(label="System Status", lines=3, interactive=False)
+            init_btn = gr.Button("🚀 Start WhatsApp Session", variant="primary")
+            refresh_btn = gr.Button("🔄 Refresh Status")
+        
         with gr.Column(scale=2):
             qr_display = gr.HTML(label="QR Code")
     
-    with gr.Tab("Send Poll"):
-        poll_num = gr.Textbox(label="Phone Number (e.g. 5511999999999)")
-        poll_title = gr.Textbox(label="Question")
-        poll_opts = gr.Textbox(label="Options (comma separated)")
-        poll_send = gr.Button("Send Poll")
-        poll_output = gr.JSON(label="Result")
+    with gr.Tabs():
+        with gr.TabItem("💬 Send Message"):
+            with gr.Row():
+                msg_num = gr.Textbox(label="Phone Number", placeholder="e.g. 5511999999999")
+                msg_txt = gr.Textbox(label="Message", lines=2)
+            msg_btn = gr.Button("Send Text", variant="secondary")
+            msg_out = gr.JSON(label="Result")
+            
+        with gr.TabItem("📊 Send Poll"):
+            with gr.Row():
+                poll_num = gr.Textbox(label="Phone Number")
+                poll_title = gr.Textbox(label="Poll Question")
+            poll_opts = gr.Textbox(label="Choices (comma separated)", placeholder="Yes, No, Maybe")
+            poll_btn = gr.Button("Send Poll", variant="secondary")
+            poll_out = gr.JSON(label="Result")
 
-    log_display = gr.Code(label="Internal Logs", language="markdown")
+    with gr.Accordion("📋 System Logs", open=False):
+        log_display = gr.Code(label="Recent logs", lines=10, language="markdown")
 
     init_btn.click(trigger_init, outputs=status_display)
     refresh_btn.click(update_ui, outputs=[status_display, qr_display, log_display])
-    poll_send.click(send_poll, inputs=[poll_num, poll_title, poll_opts], outputs=poll_output)
+    msg_btn.click(send_message, inputs=[msg_num, msg_txt], outputs=msg_out)
+    poll_btn.click(send_poll, inputs=[poll_num, poll_title, poll_opts], outputs=poll_out)
     
-    # Try to use every if Gradio version is new enough, otherwise fallback
-    try:
-        demo.load(update_ui, outputs=[status_display, qr_display, log_display], every=10)
-    except:
-        demo.load(update_ui, outputs=[status_display, qr_display, log_display])
+    # Auto-refresh UI on load
+    demo.load(update_ui, outputs=[status_display, qr_display, log_display])
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)

src/app.ts

@@ -1,13 +1,9 @@
 import express from 'express';
 import * as wppconnect from './index';
 import dns from 'dns';
-import path from 'path';
+import { Resolver } from 'dns';
 import fs from 'fs';
-
-// Attempt to use public DNS
-try {
-  dns.setServers(['1.1.1.1', '8.8.8.8']);
-} catch (e) {}
+import path from 'path';
 
 const app = express();
 const port = process.env.NODE_PORT || 3000;
@@ -23,7 +19,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());
@@ -74,7 +70,23 @@ async function startWPP() {
   statusMsg = 'Initializing...';
   log('Starting WPPConnect...');
   
-  // Cleanup tokens
+  // Try to get IP for web.whatsapp.com manually
+  let waIP = '157.240.22.60'; // Default fallback
+  try {
+    const resolver = new Resolver();
+    resolver.setServers(['8.8.8.8', '1.1.1.1']);
+    const addrs = await new Promise<string[]>((resolve, reject) => {
+      resolver.resolve4('web.whatsapp.com', (err, addresses) => err ? reject(err) : resolve(addresses));
+    });
+    if (addrs && addrs.length > 0) {
+        waIP = addrs[0];
+        log(`Resolved web.whatsapp.com to ${waIP}`);
+    }
+  } catch (e) {
+    log(`DNS check failed, 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 });
@@ -82,7 +94,7 @@ async function startWPP() {
 
   try {
     whatsappClient = await wppconnect.create({
-      session: 'hf-session',
+      session: 'hf-session-' + Date.now(),
       catchQR: (base64, ascii) => { 
         lastQR = { base64, ascii }; 
         statusMsg = 'Waiting for scan'; 
@@ -102,14 +114,14 @@ async function startWPP() {
           '--disable-setuid-sandbox',
           '--disable-dev-shm-usage',
           '--disable-gpu',
-          '--dns-servers=1.1.1.1,8.8.8.8'
+          `--host-resolver-rules=MAP web.whatsapp.com ${waIP}`,
         ],
       },
       autoClose: 0,
       updatesLog: false,
       waitForLogin: false,
     });
-    log('Client object created');
+    log('Client ready');
     isInitializing = false;
   } catch (e) {
     log('Init Error: ' + e.message);