Spaces:

AUXteam
/

wppconnect-api

Paused

App Files Files Community

AUXteam commited on Apr 1

Commit

1be4e49

verified ·

1 Parent(s): d074942

Upload folder using huggingface_hub

Browse files

Files changed (2) hide show

frontend.py +48 -25
src/app.ts +44 -14

frontend.py CHANGED Viewed

@@ -22,18 +22,22 @@ def update_ui():
     if not status:
         return "Backend Offline", "Waiting for Node.js to start...", ""
-    msg = f"WhatsApp Status: {status['whatsappStatus']}\nConnected: {status['connected']}\nReady: {status['ready']}"
-    qr_text = ""
     try:
         qr_r = requests.get(f"{NODE_URL}/qr-data", timeout=1)
         if qr_r.status_code == 200:
-            qr_text = qr_r.json()['ascii']
     except:
         pass
-    qr_html = f"<html><body style='background:white;color:black;padding:10px;display:flex;justify-content:center;'><pre style='font-size:6px;line-height:1;margin:0;'>{qr_text}</pre></body></html>" if qr_text else "No QR code generated yet."
     logs = "\n".join(status.get('logs', []))
     return msg, qr_html, logs
@@ -41,7 +45,7 @@ def update_ui():
 def trigger_init():
     try:
         requests.post(f"{NODE_URL}/init")
-        return "Initialization triggered."
     except Exception as e:
         return f"Error: {str(e)}"
@@ -53,30 +57,49 @@ def send_poll(number, title, options):
     except Exception as e:
         return {"error": str(e)}
-with gr.Blocks(title="WPPConnect") as demo:
-    gr.Markdown("# 🤖 WPPConnect Control Panel")
     with gr.Row():
-        with gr.Column():
-            status_text = gr.Label(label="System Status")
-            init_btn = gr.Button("🚀 Start/Retry WhatsApp", variant="primary")
-        with gr.Column():
-            qr_display = gr.HTML(label="WhatsApp QR Code")
-    with gr.Tab("Send Poll"):
-        poll_number = gr.Textbox(label="Phone Number (with country code)")
-        poll_title = gr.Textbox(label="Poll Question")
-        poll_options = gr.Textbox(label="Options (comma separated)")
-        poll_send = gr.Button("Send Poll")
-        poll_output = gr.JSON(label="Result")
-    with gr.Accordion("System Logs", open=False):
-        log_display = gr.Code(label="Logs", language="markdown", lines=10)
-    init_btn.click(trigger_init, outputs=status_text)
-    poll_send.click(send_poll, inputs=[poll_number, poll_title, poll_options], outputs=poll_output)
-    demo.load(update_ui, outputs=[status_text, qr_display, log_display], every=5)
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)

     if not status:
         return "Backend Offline", "Waiting for Node.js to start...", ""
+    msg = f"Status: {status['whatsappStatus']}\nConnected: {status['connected']}\nReady: {status['ready']}"
+    qr_html = "No QR code generated yet."
     try:
         qr_r = requests.get(f"{NODE_URL}/qr-data", timeout=1)
         if qr_r.status_code == 200:
+            qr_data = qr_r.json()
+            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;'>{qr_data['ascii']}</pre>
+                <p style='margin-top:10px; font-weight:bold;'>Scan this code with your phone</p>
+            </div>
+            \"\"\"
     except:
         pass
     logs = "\n".join(status.get('logs', []))
     return msg, qr_html, logs
 def trigger_init():
     try:
         requests.post(f"{NODE_URL}/init")
+        return "Initialization requested. Please wait 30-60 seconds and check logs."
     except Exception as e:
         return f"Error: {str(e)}"
     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.Soft()) as demo:
+    gr.Markdown("# 🤖 WPPConnect WhatsApp Dashboard")
     with gr.Row():
+        with gr.Column(scale=1):
+            status_display = gr.Label(label="System Status")
+            init_btn = gr.Button("🚀 Start WhatsApp Session", variant="primary")
+            refresh_btn = gr.Button("🔄 Force Refresh UI")
+        with gr.Column(scale=2):
+            qr_display = gr.HTML(label="WhatsApp QR Code", value="Waiting for initialization...")
+    with gr.Tabs():
+        with gr.TabItem("💬 Send Message"):
+            msg_num = gr.Textbox(label="Phone Number", placeholder="e.g. 5511999999999")
+            msg_txt = gr.Textbox(label="Message", lines=3)
+            msg_btn = gr.Button("Send Text")
+            msg_out = gr.JSON(label="Result")
+        with gr.TabItem("📊 Send Poll"):
+            poll_num = gr.Textbox(label="Phone Number")
+            poll_title = gr.Textbox(label="Question")
+            poll_opts = gr.Textbox(label="Options", placeholder="Option 1, Option 2, Option 3")
+            poll_btn = gr.Button("Send Poll")
+            poll_out = gr.JSON(label="Result")
+    with gr.Accordion("📋 System Logs", open=True):
+        log_display = gr.Textbox(label="Recent logs", lines=10, interactive=False)
+    init_btn.click(trigger_init, outputs=status_display)
+    refresh_btn.click(update_ui, outputs=[status_display, qr_display, log_display])
+    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)
+    # Auto-refresh every 5 seconds
+    demo.load(update_ui, outputs=[status_display, qr_display, log_display], every=5)
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)

src/app.ts CHANGED Viewed

@@ -1,22 +1,29 @@
 import express from 'express';
 import * as wppconnect from './index';
-import fs from 'fs';
 import path from 'path';
 const app = express();
 const port = process.env.NODE_PORT || 3000;
 let whatsappClient: wppconnect.Whatsapp | null = null;
-let lastQR: { ascii: string } | null = null;
-let statusMsg: string = 'Offline';
 let appLogs: string[] = [];
 let isReady = false;
 const log = (msg: string) => {
   const entry = `${new Date().toISOString()} - ${msg}`;
   console.log(entry);
   appLogs.push(entry);
-  if (appLogs.length > 100) appLogs.shift();
 };
 app.use(express.json());
@@ -38,33 +45,48 @@ app.get('/qr-data', (req, res) => {
 });
 app.post('/init', (req, res) => {
-  log('Manual init triggered');
   startWPP();
   res.json({ success: true });
 });
 app.post('/send-poll', async (req, res) => {
   const { telnumber, name, choices } = req.body;
   if (!whatsappClient || !isReady) return res.status(503).json({ error: 'WhatsApp not ready' });
   try {
     const result = await whatsappClient.sendPollMessage(`${telnumber}@c.us`, name, choices);
     res.json({ success: true, result });
-  } catch (e) {
-    log('Poll Error: ' + e.message);
-    res.status(500).json({ error: e.message });
-  }
 });
 async function startWPP() {
   statusMsg = 'Initializing...';
   log('Starting WPPConnect...');
   try {
     whatsappClient = await wppconnect.create({
       session: 'hf-session',
-      catchQR: (b64, ascii) => {
-        lastQR = { ascii };
         statusMsg = 'Waiting for scan';
       },
       statusFind: (status) => {
         statusMsg = status;
@@ -75,19 +97,27 @@ async function startWPP() {
       useChrome: false,
       puppeteerOptions: {
         executablePath: '/usr/bin/chromium',
-        args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage', '--disable-gpu'],
       },
       autoClose: 0,
       updatesLog: false,
       waitForLogin: false,
     });
-    log('Client Created');
   } catch (e) {
     log('Init Error: ' + e.message);
     statusMsg = 'Error: ' + e.message;
   }
 }
 app.listen(port, () => {
-  log(`Node Backend started on port ${port}`);
 });

 import express from 'express';
 import * as wppconnect from './index';
+import dns from 'dns';
 import path from 'path';
+import fs from 'fs';
+// Attempt to use public DNS
+try {
+  dns.setServers(['1.1.1.1', '8.8.8.8']);
+} catch (e) {}
 const app = express();
 const port = process.env.NODE_PORT || 3000;
 let whatsappClient: wppconnect.Whatsapp | null = null;
+let lastQR: { ascii: string, base64: string } | null = null;
+let statusMsg: string = 'Disconnected';
 let appLogs: string[] = [];
 let isReady = false;
+let isInitializing = false;
 const log = (msg: string) => {
   const entry = `${new Date().toISOString()} - ${msg}`;
   console.log(entry);
   appLogs.push(entry);
+  if (appLogs.length > 200) appLogs.shift();
 };
 app.use(express.json());
 });
 app.post('/init', (req, res) => {
+  if (isInitializing) return res.json({ success: false, message: 'Already initializing' });
   startWPP();
   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) { res.status(500).json({ error: e.message }); }
+});
 app.post('/send-poll', async (req, res) => {
   const { telnumber, name, choices } = req.body;
   if (!whatsappClient || !isReady) return res.status(503).json({ error: 'WhatsApp not ready' });
   try {
     const result = await whatsappClient.sendPollMessage(`${telnumber}@c.us`, name, choices);
     res.json({ success: true, result });
+  } catch (e) { res.status(500).json({ error: e.message }); }
 });
 async function startWPP() {
+  if (isInitializing) return;
+  isInitializing = true;
   statusMsg = 'Initializing...';
   log('Starting WPPConnect...');
+  // Cleanup tokens
+  try {
+    const tokensDir = path.join(process.cwd(), 'tokens');
+    if (fs.existsSync(tokensDir)) fs.rmSync(tokensDir, { recursive: true, force: true });
+  } catch (e) {}
   try {
     whatsappClient = await wppconnect.create({
       session: 'hf-session',
+      catchQR: (base64, ascii) => {
+        lastQR = { base64, ascii };
         statusMsg = 'Waiting for scan';
+        log('QR generated');
       },
       statusFind: (status) => {
         statusMsg = status;
       useChrome: false,
       puppeteerOptions: {
         executablePath: '/usr/bin/chromium',
+        args: [
+          '--no-sandbox',
+          '--disable-setuid-sandbox',
+          '--disable-dev-shm-usage',
+          '--disable-gpu',
+          '--dns-servers=1.1.1.1,8.8.8.8'
+        ],
       },
       autoClose: 0,
       updatesLog: false,
       waitForLogin: false,
     });
+    log('Client object created');
+    isInitializing = false;
   } catch (e) {
     log('Init Error: ' + e.message);
     statusMsg = 'Error: ' + e.message;
+    isInitializing = false;
   }
 }
 app.listen(port, () => {
+  log(`Node Backend on port ${port}`);
 });