Update app/main.py

app/main.py

@@ -2,82 +2,77 @@ from twilio.rest import Client
 import yaml
 import json
 import os
-
-import yaml
-import json
+import sys
+from pathlib import Path
+from flask import Flask, request, redirect
+from twilio.twiml.messaging_response import MessagingResponse
 from langchain.embeddings import OpenAIEmbeddings
 from langchain_community.vectorstores import Chroma
 from helper import retrieve_relevant_context, generate_response_with_context
+from pyngrok import ngrok
 
-import sys
-from pathlib import Path
+# Setting up paths
 file = Path(__file__).resolve()
 parent, root = file.parent, file.parents[1]
 sys.path.append(str(root))
-print("str(root) :",str(root))
-print("parent :",parent)
-
-print("CWD :",os.getcwd())
+print("str(root):", str(root))
+print("parent:", parent)
+print("CWD:", os.getcwd())
 
 # Load relevant API Keys
 file_path = parent / 'Config/API_KEYS.yml'
-
-# Define the persist directory
 persist_directory = str(parent / 'vector_db/chroma_v01/')
-
-print("file_path :",file_path)
-print("persist_directory :",str(persist_directory))
-
+print("file_path:", file_path)
+print("persist_directory:", str(persist_directory))
 
 with open(file_path, 'r') as file:
     api_keys = yaml.safe_load(file)
 
-
-# Extract openai username and key
+# Extract OpenAI key
 openai_key = api_keys['OPEN_AI']['Key']
-
 os.environ["OPENAI_API_KEY"] = openai_key
 
-
-
-# Extract openai username and key
+# Extract Twilio credentials
 account_sid = api_keys['TWILIO']['account_sid']
 auth_token = api_keys['TWILIO']['auth_token']
-
-account_sid = account_sid
-auth_token = auth_token
-
-print("====account_sid:=====",account_sid)
-# Define the persist directory
-# persist_directory = './vector_db/chroma_v01'
+twilio_whatsapp_number = api_keys['TWILIO']['whatsapp_number']
+print("====account_sid:=====", account_sid)
 
 # Initialize the embeddings model
 embedding_model = OpenAIEmbeddings()
 
-### Vectorstores
-from langchain_community.vectorstores import Chroma
-
 # Load the Chroma vector store
 vectordb = Chroma(persist_directory=persist_directory, embedding_function=embedding_model)
 
-
-#setup Twilio client
+# Setup Twilio client
 client = Client(account_sid, auth_token)
 
-# Find your Account SID and Auth Token at twilio.com/console
-message = client.messages.create(
-                              body='Hello there!',
-                              from_='whatsapp:+14155238886',
-                              to='whatsapp:+91-9108843322'
-                          )
-print("message  :",message )
-print("message.sid :",message.sid)
-
-
+# Example to send a WhatsApp message
+def send_whatsapp_message(to_number, message):
+    """
+    Send a WhatsApp message using Twilio.
+    
+    :param to_number: str, recipient's WhatsApp number in the format 'whatsapp:+1234567890'
+    :param message: str, message text to send
+    """
+    from_number = f'whatsapp:{twilio_whatsapp_number}'
+    to_number = f'whatsapp:{to_number}'
+    
+    message = client.messages.create(
+        body=message,
+        from_=from_number,
+        to=to_number
+    )
+    print(f"Message sent with SID: {message.sid}")
+
+# Example usage
+if __name__ == "__main__":
+    recipient_number = '+91-9108843322'  # Replace with the recipient's WhatsApp number
+    text_message = 'Hello from Twilio WhatsApp!'
+    send_whatsapp_message(recipient_number, text_message)
 
-from flask import Flask, request, redirect
-from twilio.twiml.messaging_response import MessagingResponse
-print("flask app is running")
+# Flask app setup
+print("Flask app is running")
 app = Flask(__name__)
 
 @app.route("/whatsapp", methods=['GET', 'POST'])
@@ -85,9 +80,8 @@ def incoming_sms():
     """Send a dynamic reply to an incoming text message"""
     # Get the message the user sent our Twilio number
     body = request.values.get('Body', None)
-    print("body :",body)
+    print("body:", body)
     
-    ##### Process incoming text #############
     incoming_msg = body.strip()
     if not incoming_msg:
         return str(MessagingResponse())
@@ -96,15 +90,20 @@ def incoming_sms():
     retrieved_texts = retrieve_relevant_context(vectordb, incoming_msg)
     context = "\n".join(retrieved_texts)
     response = generate_response_with_context(incoming_msg, context)
-    print("response :",response)
-    ##### Process incoming text Done #############
-    
+    print("response:", response)
     
     # Start our TwiML response
     resp = MessagingResponse()
-    print("TwiML resp :", resp)
+    print("TwiML resp:", resp)
     resp.message(response)
     return str(resp)
 
 if __name__ == "__main__":
-    app.run(port=5000, debug=True)
+    # Start ngrok tunnel
+    ngrok_tunnel = ngrok.connect(5000)
+    print(f"ngrok tunnel 'http' URL: {ngrok_tunnel.public_url}")
+    
+    # Print the ngrok URL so you can set it in Twilio webhook
+    print("ngrok URL:", ngrok_tunnel.public_url + "/whatsapp")
+    
+    app.run(host='0.0.0.0', port=5000, debug=True)