Update app.py

app.py

@@ -3,11 +3,16 @@ import hashlib
 import datetime
 import streamlit as st
 import pandas as pd
+import openai
+import os
 
 # Hotelbeds API credentials
 hotelbeds_api_key = "95ce3e45a02fc6fd9720ecc013a6f674"
 hotelbeds_api_secret = "785150201f"
 
+# OpenAI API credentials
+openai.api_key = os.getenv("sk-proj-ai6i8e4j43sJxWlG8y5XT3BlbkFJA7w5uANWtfvRchfBdlgy")
+
 # Function to generate the API signature
 def generate_signature():
     local_timestamp = int(datetime.datetime.now().timestamp())
@@ -15,6 +20,19 @@ def generate_signature():
     signature = hashlib.sha256(assemble.encode()).hexdigest()
     return signature
 
+# Function to get geo-coordinates from a location name using OpenAI API
+def get_coordinates(location_name):
+    prompt = f"Provide the latitude and longitude for {location_name}."
+    response = openai.Completion.create(
+        engine="text-davinci-003",
+        prompt=prompt,
+        max_tokens=10,
+        temperature=0
+    )
+    result = response.choices[0].text.strip()
+    coordinates = result.split(",")
+    return float(coordinates[0]), float(coordinates[1])
+
 # Function to make the POST request to the Hotelbeds API and handle pagination
 def fetch_activities(latitude, longitude, from_date, to_date, language="en", paxes=[{"age": 5}, {"age": 70}], order="DEFAULT"):
     url = "https://api.test.hotelbeds.com/activity-api/3.0/activities/availability"
@@ -74,14 +92,16 @@ def fetch_activities(latitude, longitude, from_date, to_date, language="en", pax
 # Streamlit app
 st.title("Hotelbeds Activity Availability Checker")
 
-latitude = st.number_input("Latitude", value=40.730610)
-longitude = st.number_input("Longitude", value=-73.93524)
+location_name = st.text_input("Location Name", value="New York City")
 from_date = st.date_input("From Date", value=pd.to_datetime("2024-06-20"))
 to_date = st.date_input("To Date", value=pd.to_datetime("2024-06-24"))
 
 if st.button("Check Availability"):
+    latitude, longitude = get_coordinates(location_name)
     activities = fetch_activities(latitude, longitude, from_date.isoformat(), to_date.isoformat())
     if activities:
-        st.json(activities)
+        activity_names = [activity['content']['name'] for activity in activities]
+        st.json(activity_names)
+        st.write(f"Total number of activities: {len(activity_names)}")
     else:
         st.error("No activities found or an error occurred.")