from flask import Flask, request, jsonify
from twilio.rest import Client
import snowflake.connector
from groq import Groq
from prosus import Prosus
from serpapi import GoogleSearch
import coral_protocol as coral
import tavily
import json

app = Flask(__name__)

# Setup for APIs and Agents
twilio_client = Client()
groq_client = Groq()
prosus_client = Prosus()
snowflake_client = snowflake.connector.connect(user='YOUR_USER', password='YOUR_PASSWORD', account='YOUR_ACCOUNT')
coral_client = coral.Coral()

# Function to get restaurant details from SERP API
def get_restaurant_info(query):
    params = {
        'q': query,
        'api_key': 'YOUR_SERP_API_KEY'
    }
    search = GoogleSearch(params)
    results = search.get_dict()
    return results

# Function to send message using Twilio
def send_sms(to, message):
    message = twilio_client.messages.create(
        body=message,
        from_='+19379392457',  
        to=to
    )
    return message.sid

# Function to interact with the Prosus API
def create_ai_agent_for_store(store_name):
    agent_response = prosus_client.create_agent(store_name)
    return agent_response

# Function to store user profile in a knowledge graph (Coral Protocol)
def store_user_profile(user_id, profile_data):
    coral_client.store_profile(user_id, profile_data)
    return "Profile stored successfully!"

@app.route("/place_order", methods=["POST"])
def place_order():
    order_data = request.json
    restaurant = order_data.get("restaurant")
    food_items = order_data.get("items")

    # Process order (could involve AI-based recommendation here)
    order_info = get_restaurant_info(restaurant)
    send_sms(order_data.get("user_phone"), "Your order has been placed successfully!")

    return jsonify({"message": "Order placed successfully", "order_info": order_info})

@app.route("/book_travel", methods=["POST"])
def book_travel():
    travel_data = request.json
    destination = travel_data.get("destination")
    flight_info = get_flight_info(destination)  # Assume get_flight_info is another function
    hotel_info = get_hotel_info(destination)    # Assume get_hotel_info is another function

    return jsonify({"message": "Travel booking details", "flights": flight_info, "hotels": hotel_info})

@app.route("/create_store_agent", methods=["POST"])
def create_store_agent():
    store_name = request.json.get("store_name")
    agent_response = create_ai_agent_for_store(store_name)
    return jsonify({"message": "AI agent created for the store", "agent_response": agent_response})

@app.route("/store_profile", methods=["POST"])
def store_profile():
    user_id = request.json.get("user_id")
    profile_data = request.json.get("profile_data")
    store_user_profile = store_user_profile(user_id, profile_data)
    return jsonify({"message": store_user_profile})

if __name__ == "__main__":
    app.run(debug=True)