Create app.py

app.py

@@ -0,0 +1,144 @@
+import logging
+import time
+from threading import Thread
+from whatsapp_bridge.bot import ApplicationBuilder, MessageHandler, TypeHandler
+from whatsapp_bridge.bot import TextFilter
+from groq import Groq
+import os
+import uvicorn
+from fastapi import FastAPI
+
+# Configure logging
+logging.basicConfig(
+    format="%(asctime)s - %(name)s - %(levelname)s - %(message)s",
+    level=logging.INFO
+)
+logger = logging.getLogger(__name__)
+
+# Initialize Groq client
+client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+
+# System prompt to define assistant behavior
+system_prompt = """  
+You are Lucy, personal assistant of Naksh.  
+For the first incoming message reply,  
+Hi! I am Lucy, Personal Assistant of Naksh. He is currently busy and will message you soon.  
+The second message should be different.  
+Every message should be different and also always be respectful.  
+you can even chat with the sender  
+From the second message you can handle the conversation by yourself  
+If someone asks about me tell them he is busy in some work.  
+"""
+
+# Global sender instance
+sender = None
+
+class WhatsAppSender:
+    def __init__(self, application):
+        self.application = application
+        self.client = None
+
+    def get_client(self):
+        if self.client is None and hasattr(self.application, 'listener'):
+            self.client = self.application.listener.client
+        return self.client
+
+    def send_message(self, jid, message):
+        client = self.get_client()
+        if client is None:
+            return False, "WhatsApp client not available"
+        try:
+            for method_name in ['send_message', 'send_text', 'send']:
+                if hasattr(client, method_name):
+                    getattr(client, method_name)(jid, message)
+                    return True, "Message sent successfully"
+            return False, "No valid send method found on client"
+        except Exception as e:
+            return False, str(e)
+
+async def ai_handler(update, context):
+    global sender
+    if not (update.message and update.message.get("content")):
+        return
+
+    chat_jid = update.message.get("chat_jid")
+    incoming = update.message.get("content")
+    logger.info(f"Received from {chat_jid}: '{incoming}'")
+
+    messages = [
+        {"role": "system", "content": system_prompt},
+        {"role": "user",   "content": incoming}
+    ]
+
+    try:
+        response = client.chat.completions.create(
+            messages=messages,
+            model="llama-3.3-70b-versatile",
+            stream=False,
+        )
+        ai_text = response.choices[0].message.content
+    except Exception as e:
+        logger.error(f"Groq AI error: {e}")
+        ai_text = "Sorry, I'm having trouble responding right now."
+
+    success, result = sender.send_message(chat_jid, ai_text)
+    if success:
+        logger.info(f"Sent to {chat_jid}: '{ai_text}'")
+    else:
+        logger.error(f"Failed to send message to {chat_jid}: {result}")
+
+def cli_send_loop(sender_instance):
+    while True:
+        try:
+            command = input("\nEnter command (/send <jid> <message>): ").strip()
+        except EOFError:
+            break  # no CLI in non-interactive env
+
+        if command.startswith("/send "):
+            parts = command.split(" ", 2)
+            if len(parts) < 3:
+                print("Invalid format. Use: /send <jid> <message>")
+                continue
+            _, jid, message = parts
+            success, result = sender_instance.send_message(jid, message)
+            if success:
+                logger.info(f"Message sent to {jid}: '{message}'")
+            else:
+                logger.error(f"Failed to send: {result}")
+                print(f"Error: {result}")
+
+
+# FastAPI setup
+app = FastAPI()
+
+@app.get("/ping")
+async def ping():
+    return {"message": "pong"}
+
+
+def run_fastapi(port=8000):
+    uvicorn.run(app, host="0.0.0.0", port=port, log_level="info")
+
+
+def main():
+    global sender
+    logger.info("Starting WhatsApp AI bot...")
+
+    application = ApplicationBuilder().build()
+    sender = WhatsAppSender(application)
+
+    application.add_handler(TypeHandler(lambda u, c: logger.debug(f"Update: {u}")), group=-1)
+    application.add_handler(MessageHandler(TextFilter(), ai_handler))
+
+    cli_thread = Thread(target=cli_send_loop, args=(sender,), daemon=True)
+    cli_thread.start()
+
+    # Start FastAPI in a separate thread
+    fastapi_thread = Thread(target=run_fastapi, daemon=True)
+    fastapi_thread.start()
+
+    application.run_polling()
+
+
+if __name__ == "__main__":
+    main()