Spaces:

AUXteam
/

wppconnect-api

Paused

App Files Files Community

AUXteam commited on Apr 1

Commit

aca0d15

verified ·

1 Parent(s): b7a002d

Upload folder using huggingface_hub

Browse files

Files changed (4) hide show

Dockerfile +3 -3
README.md +9 -8
frontend.py +31 -38
src/app.ts +22 -57

Dockerfile CHANGED Viewed

@@ -16,11 +16,11 @@ ENV PUPPETEER_SKIP_CHROMIUM_DOWNLOAD=true \
 WORKDIR /app
 COPY package*.json ./
-RUN node -e "const fs = require('fs'); const pkg = JSON.parse(fs.readFileSync('package.json', 'utf8')); pkg.scripts.prepare = ''; pkg.scripts.start = 'node dist/app.js'; pkg.dependencies.express = '^4.18.2'; pkg.devDependencies['@types/express'] = '^4.17.17'; fs.writeFileSync('package.json', JSON.stringify(pkg, null, 2));"
 RUN npm install --legacy-peer-deps
 RUN pip3 install --break-system-packages gradio requests
 COPY . .
 RUN npm run build
-RUN chmod +x start.sh
 EXPOSE 7860
-CMD ["/bin/bash", "./start.sh"]

 WORKDIR /app
 COPY package*.json ./
+RUN node -e "const fs = require('fs'); const pkg = JSON.parse(fs.readFileSync('package.json', 'utf8')); pkg.scripts.prepare = ''; pkg.scripts.start = ''; pkg.dependencies.express = '^4.18.2'; pkg.devDependencies['@types/express'] = '^4.17.17'; fs.writeFileSync('package.json', JSON.stringify(pkg, null, 2));"
 RUN npm install --legacy-peer-deps
 RUN pip3 install --break-system-packages gradio requests
 COPY . .
 RUN npm run build
 EXPOSE 7860
+CMD ["python3", "frontend.py"]

README.md CHANGED Viewed

@@ -1,18 +1,19 @@
 ---
-title: WPPConnect Control Panel
 emoji: 🤖
 colorFrom: blue
 colorTo: green
 sdk: docker
 app_port: 7860
 ---
-# WPPConnect Control Panel
-WhatsApp automation with Gradio UI.
-## Features
-- Dynamic QR Code display
-- Status monitoring
-- Send text messages
-- Send polls

 ---
+title: WPPConnect API
 emoji: 🤖
 colorFrom: blue
 colorTo: green
 sdk: docker
 app_port: 7860
+pinned: false
 ---
+# WPPConnect API
+Node.js WhatsApp automation API running in Docker.
+## Endpoints
+- GET /health
+- GET /qr
+- POST /send-message
+- POST /send-poll

frontend.py CHANGED Viewed

@@ -1,48 +1,49 @@
 import gradio as gr
 import requests
 import time
 import os
 NODE_URL = "http://localhost:3000"
 def get_status():
     try:
-        r = requests.get(f"{NODE_URL}/health", timeout=2)
-        if r.status_code == 200:
-            data = r.json()
-            return data
     except:
-        pass
-    return None
 def update_ui():
     status = get_status()
     if not status:
-        return "Backend Offline", None, "Waiting for Node.js..."
-    msg = f"Status: {status.get('whatsappStatus', 'Unknown')}\nConnected: {status.get('connected')}\nReady: {status.get('ready')}"
-    qr_img = None
     try:
-        qr_r = requests.get(f"{NODE_URL}/qr-data", timeout=2)
         if qr_r.status_code == 200:
-            qr_data = qr_r.json()
-            # In a real app we might want to convert base64 to image,
-            # but for now let's just use the ascii in a pre tag
-            qr_img = f"<html><body style='background:white;padding:10px;'><pre style='font-size:6px;line-height:1;color:black;margin:0;'>{qr_data['ascii']}</pre></body></html>"
     except:
         pass
-    logs = "\n".join(status.get('logs', [])[-10:])
-    return msg, qr_img, logs
-def send_msg(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)}
 def send_poll(number, title, options):
     try:
@@ -52,37 +53,29 @@ def send_poll(number, title, options):
     except Exception as e:
         return {"error": str(e)}
-with gr.Blocks(title="WPPConnect Control Panel") as demo:
-    gr.Markdown("# 🤖 WPPConnect WhatsApp Control")
     with gr.Row():
-        with gr.Column(scale=1):
-            status_text = gr.Textbox(label="System Status", interactive=False)
-            refresh_btn = gr.Button("Refresh Status")
-        with gr.Column(scale=2):
             qr_display = gr.HTML(label="WhatsApp QR Code")
-    with gr.Tab("Send Message"):
-        msg_number = gr.Textbox(label="Phone Number (e.g. 5511999999999)")
-        msg_text = gr.Textbox(label="Message Text")
-        msg_send = gr.Button("Send Text")
-        msg_output = gr.JSON(label="Result")
     with gr.Tab("Send Poll"):
-        poll_number = gr.Textbox(label="Phone Number")
         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.Row():
-        log_display = gr.Code(label="Recent Logs", language="text")
-    refresh_btn.click(update_ui, outputs=[status_text, qr_display, log_display])
-    msg_send.click(send_msg, inputs=[msg_number, msg_text], outputs=msg_output)
     poll_send.click(send_poll, inputs=[poll_number, poll_title, poll_options], outputs=poll_output)
-    # Auto refresh every 5 seconds
     demo.load(update_ui, outputs=[status_text, qr_display, log_display], every=5)
 if __name__ == "__main__":

 import gradio as gr
 import requests
+import subprocess
 import time
 import os
 NODE_URL = "http://localhost:3000"
+# Start Node.js backend
+print("Starting Node.js backend...")
+subprocess.Popen(["node", "dist/app.js"])
 def get_status():
     try:
+        r = requests.get(f"{NODE_URL}/health", timeout=1)
+        return r.json()
     except:
+        return None
 def update_ui():
     status = get_status()
     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
+def trigger_init():
     try:
+        requests.post(f"{NODE_URL}/init")
+        return "Initialization triggered."
     except Exception as e:
+        return f"Error: {str(e)}"
 def send_poll(number, title, options):
     try:
     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="text", 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__":

src/app.ts CHANGED Viewed

@@ -1,7 +1,5 @@
 import express from 'express';
 import * as wppconnect from './index';
-import dns from 'dns';
-import { Resolver } from 'dns';
 import fs from 'fs';
 import path from 'path';
@@ -9,17 +7,16 @@ 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 = 'Initializing...';
 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 > 500) appLogs.shift();
 };
 app.use(express.json());
@@ -36,20 +33,14 @@ app.get('/health', (req, res) => {
 });
 app.get('/qr-data', (req, res) => {
-  if (lastQR) {
-    res.json(lastQR);
-  } else {
-    res.status(404).json({ error: 'No QR yet' });
-  }
 });
-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) => {
@@ -58,71 +49,45 @@ app.post('/send-poll', async (req, res) => {
   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 start() {
-  if (isInitializing) return;
-  isInitializing = true;
-  try {
-    const tokensDir = path.join(process.cwd(), 'tokens');
-    if (fs.existsSync(tokensDir)) {
-      fs.rmSync(tokensDir, { recursive: true, force: true });
-    }
-  } catch (e) {}
-  let waIP = '157.240.22.60';
-  try {
-    const resolver = new Resolver();
-    resolver.setServers(['8.8.8.8']);
-    const addrs = await new Promise<string[]>((resolve, reject) => {
-      resolver.resolve4('web.whatsapp.com', (err, addresses) => err ? reject(err) : resolve(addresses));
-    });
-    waIP = addrs[0];
-    log(`Resolved to ${waIP}`);
-  } catch (e) {
-    log(`DNS failed, using ${waIP}`);
-  }
-  statusMsg = 'Creating Client...';
   try {
     whatsappClient = await wppconnect.create({
       session: 'hf-session',
-      catchQR: (base64, ascii) => {
-        lastQR = { base64, ascii };
         statusMsg = 'Waiting for scan';
       },
       statusFind: (status) => {
         statusMsg = status;
         if (status === 'inChat') { isReady = true; lastQR = null; }
       },
       headless: true,
       useChrome: false,
       puppeteerOptions: {
         executablePath: '/usr/bin/chromium',
-        args: [
-          '--no-sandbox',
-          '--disable-setuid-sandbox',
-          '--disable-dev-shm-usage',
-          '--disable-gpu',
-          `--host-resolver-rules=MAP web.whatsapp.com ${waIP}`,
-        ],
       },
       autoClose: 0,
       updatesLog: false,
       waitForLogin: false,
     });
-    log('READY');
-    isInitializing = false;
   } catch (e) {
     log('Init Error: ' + e.message);
-    isInitializing = false;
-    setTimeout(start, 30000);
   }
 }
 app.listen(port, () => {
   log(`Node Backend started on port ${port}`);
-  start();
 });

 import express from 'express';
 import * as wppconnect from './index';
 import fs from 'fs';
 import path from 'path';
 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());
 });
 app.get('/qr-data', (req, res) => {
+  if (lastQR) res.json(lastQR);
+  else res.status(404).json({ error: 'No QR' });
 });
+app.post('/init', (req, res) => {
+  log('Manual init triggered');
+  startWPP();
+  res.json({ success: true });
 });
 app.post('/send-poll', async (req, res) => {
   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;
+        log('Status: ' + status);
         if (status === 'inChat') { isReady = true; lastQR = null; }
       },
       headless: true,
       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}`);
 });