File size: 4,508 Bytes
5a2f268 6457f9c 5a2f268 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 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=7860):
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() |