| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051 |
- from flask import Flask, request
- from twilio.twiml.messaging_response import MessagingResponse
- from embedchain import App
- app = Flask(__name__)
- chat_bot = App()
- @app.route("/chat", methods=["POST"])
- def chat():
- incoming_message = request.values.get("Body", "").lower()
- response = handle_message(incoming_message)
- twilio_response = MessagingResponse()
- twilio_response.message(response)
- return str(twilio_response)
- def handle_message(message):
- if message.startswith("add "):
- response = add_sources(message)
- else:
- response = query(message)
- return response
- def add_sources(message):
- message_parts = message.split(" ", 2)
- if len(message_parts) == 3:
- data_type = message_parts[1]
- url_or_text = message_parts[2]
- try:
- chat_bot.add(data_type, url_or_text)
- response = f"Added {data_type}: {url_or_text}"
- except Exception as e:
- response = f"Failed to add {data_type}: {url_or_text}.\nError: {str(e)}"
- else:
- response = "Invalid 'add' command format.\nUse: add <data_type> <url_or_text>"
- return response
- def query(message):
- try:
- response = chat_bot.chat(message)
- except Exception:
- response = "An error occurred. Please try again!"
- return response
- if __name__ == "__main__":
- app.run(host="0.0.0.0", port=5000, debug=False)
|
