Upload folder using huggingface_hub

frontend.py

@@ -14,17 +14,14 @@ def get_status():
     try:
         r = requests.get(f"{NODE_URL}/health", timeout=2)
         return r.json()
-    except:
-        return None
+    except Exception as e:
+        return {"whatsappStatus": "Offline", "connected": False, "ready": False, "logs": [str(e)]}
 
 def update_ui():
     status = get_status()
-    if not status:
-        return "Backend Offline", "Waiting for Node.js...", "", None
-    
     msg = f"Status: {status.get('whatsappStatus', 'Unknown')}\nConnected: {status.get('connected', False)}\nReady: {status.get('ready', False)}"
     
-    qr_html = "<div style='text-align:center; padding:20px; color:#666;'>QR code will appear here when ready.</div>"
+    qr_html = "<div style='text-align:center; padding:20px; color:#666;'>QR code will appear here. If it takes too long, check logs.</div>"
     if status.get('hasQR'):
         try:
             qr_r = requests.get(f"{NODE_URL}/qr-data", timeout=1)
@@ -34,33 +31,24 @@ def update_ui():
                 qr_html = f"""
                 <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; letter-spacing:0;'>{ascii_qr}</pre>
-                    <p style='margin-top:10px; font-size:12px; font-family:sans-serif;'>Scan this with your WhatsApp mobile app</p>
+                    <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:20px; color:#4caf50; font-weight:bold;'>WhatsApp is connected and ready!</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', []))
     
-    screenshot_path = None
-    try:
-        sc_r = requests.get(f"{NODE_URL}/screenshot", timeout=5)
-        if sc_r.status_code == 200:
-            with open("current_screen.png", "wb") as f:
-                f.write(sc_r.content)
-            screenshot_path = "current_screen.png"
-    except:
-        pass
+    screenshot_url = f"{NODE_URL}/screenshot?t={int(time.time())}"
     
-    return msg, qr_html, logs, screenshot_path
+    return msg, qr_html, logs, screenshot_url
 
 def trigger_init():
     try:
         requests.post(f"{NODE_URL}/init")
-        return "Initialization triggered. Check logs."
+        return "Initialization triggered."
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -72,13 +60,13 @@ def send_poll(number, title, options):
     except Exception as e:
         return {"error": str(e)}
 
-with gr.Blocks(title="WPPConnect Dashboard", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🤖 WPPConnect WhatsApp Control Panel")
+with gr.Blocks(title="WPPConnect Dashboard") as demo:
+    gr.Markdown("# 🤖 WPPConnect Control Panel")
     
     with gr.Row():
         with gr.Column(scale=1):
             status_display = gr.Label(label="System Status")
-            init_btn = gr.Button("🚀 Start/Restart Session", variant="primary")
+            init_btn = gr.Button("🚀 Start/Retry Session", variant="primary")
             refresh_btn = gr.Button("🔄 Refresh UI")
         
         with gr.Column(scale=2):
@@ -87,27 +75,24 @@ with gr.Blocks(title="WPPConnect Dashboard", theme=gr.themes.Soft()) as demo:
     with gr.Tabs():
         with gr.TabItem("📊 Send Poll"):
             with gr.Row():
-                poll_num = gr.Textbox(label="Target Phone Number", placeholder="5511999999999")
-                poll_title = gr.Textbox(label="Poll Question", placeholder="What do you think?")
+                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="Yes, No, Maybe")
             poll_send = gr.Button("Send Poll Now", variant="primary")
             poll_out = gr.JSON(label="Result")
             
         with gr.TabItem("📋 System Logs"):
-            log_display = gr.Code(label="Recent Internal Logs", lines=15, language="markdown")
+            log_display = gr.Code(label="Recent Logs", lines=15, language="markdown")
             
         with gr.TabItem("🔍 Browser Debug"):
-            screenshot_display = gr.Image(label="Live Browser Screenshot")
+            screenshot_display = gr.Image(label="Live Browser View")
 
     init_btn.click(trigger_init, 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)
     
-    # Try to auto-refresh every 10s
-    try:
-        demo.load(update_ui, outputs=[status_display, qr_display, log_display, screenshot_display], every=10)
-    except:
-        demo.load(update_ui, outputs=[status_display, qr_display, log_display, screenshot_display])
+    # Simple recurring update
+    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)

src/app.ts

@@ -60,6 +60,15 @@ 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('/send-poll', async (req, res) => {
   const { telnumber, name, choices } = req.body;
   if (!whatsappClient || !isReady) return res.status(503).json({ error: 'WhatsApp not ready' });
@@ -75,7 +84,7 @@ async function startWPP() {
   statusMsg = 'Initializing...';
   log('Starting WPPConnect init...');
   
-  let waIP = '157.240.22.60'; 
+  let waIP = '31.13.66.56'; // Use the IP that worked in logs
   try {
     const resolver = new Resolver();
     resolver.setServers(['8.8.8.8', '1.1.1.1']);
@@ -87,10 +96,10 @@ async function startWPP() {
         log(`Resolved web.whatsapp.com to ${waIP}`);
     }
   } catch (e) {
-    log(`DNS failure, using fallback IP ${waIP}`);
+    log(`DNS failed, using ${waIP}`);
   }
 
-  // Fresh start every time for debugging
+  // Cleanup tokens to avoid locks
   try {
     const tokensDir = path.join(process.cwd(), 'tokens');
     if (fs.existsSync(tokensDir)) fs.rmSync(tokensDir, { recursive: true, force: true });
@@ -106,7 +115,7 @@ async function startWPP() {
       },
       statusFind: (status) => { 
         statusMsg = status; 
-        log('Status: ' + status);
+        log('Status Change: ' + status);
         if (status === 'inChat') { isReady = true; lastQR = null; } 
       },
       headless: true,