Update app.py

app.py

@@ -5,6 +5,9 @@ import streamlit as st
 import pandas as pd
 import openai
 import os
+from langchain.llms import OpenAI
+from langchain.chains import LLMChain
+from langchain.prompts import PromptTemplate
 
 # Hotelbeds API credentials
 hotelbeds_api_key = "95ce3e45a02fc6fd9720ecc013a6f674"
@@ -20,16 +23,17 @@ def generate_signature():
     signature = hashlib.sha256(assemble.encode()).hexdigest()
     return signature
 
-# Function to get geo-coordinates from a location name using OpenAI API
+# Function to get geo-coordinates from a location name using OpenAI API through Langchain
 def get_coordinates(location_name):
-    response = openai.ChatCompletion.create(
-        model="gpt-3.5-turbo",
-        messages=[
-            {"role": "system", "content": "You are a helpful assistant."},
-            {"role": "user", "content": f"Provide the latitude and longitude for {location_name}."}
-        ]
+    prompt_template = PromptTemplate(
+        input_variables=["location"],
+        template="Provide the latitude and longitude for {location}."
     )
-    result = response.choices[0].message['content'].strip()
+    
+    llm = OpenAI(model_name="gpt-4", openai_api_key=openai.api_key)
+    chain = LLMChain(llm=llm, prompt=prompt_template)
+    
+    result = chain.run(location_name).strip()
     coordinates = result.split(",")
     return float(coordinates[0]), float(coordinates[1])