close loop

app.py

@@ -1,11 +1,13 @@
 from gradio import Interface, Dropdown, CheckboxGroup
-from data import get_prompt, get_cities
+
+from data import get_cities
+from llm import get_response
 
 cities = get_cities()
 
 
 def hotel_recommender(city, preferences) -> dict:
-    return get_prompt(city, preferences)
+    return get_response(city, preferences)
 
 
 demo = Interface(

data.py

@@ -36,7 +36,7 @@ hotel_id_to_name_map = get_hotel_id_to_name_map()
 hotel_id_to_review_map = get_hotel_id_to_review_map()
 
 
-def get_prompt(city, preferences) -> dict:
+def get_reviews_for_prompt(city, preferences) -> dict:
 
     for hotel_id in perm(city_to_hotel_id_map[city]):
         hotel_id = str(hotel_id)
@@ -51,6 +51,8 @@ def get_prompt(city, preferences) -> dict:
             if (review['score'] <= 2) & (len(res['negative']) < 1):
                 res['negative'].append(review)
             if (len(res['positive']) >= 3) & (len(res['negative']) >= 1):
-                break
+                return res
+
+    return None
+
 
-    return res

llm.py

@@ -0,0 +1,16 @@
+import openai
+from os import environ
+
+from prompt import get_prompt
+
+openai.api_key = environ.get("OPENAI_API_KEY")
+
+
+def get_response(city, preferences):
+    response = openai.Completion.create(
+        model="text-davinci-003",
+        prompt=get_prompt(city, preferences),
+        temperature=0.6,
+        max_tokens=200
+    )
+    return response["choices"][0]["text"]

prompt.py

@@ -0,0 +1,19 @@
+from data import get_reviews_for_prompt
+
+
+def get_text_from_reviews(reviews, kind):
+    return [review["text"] for review in reviews[kind]]
+
+
+def get_prompt(city, preferences):
+    reviews = get_reviews_for_prompt(city, preferences)
+    hotel_name = reviews["hotel_name"]
+    positive_text, negative_text = [[review["text"] for review in reviews[kind]]
+                                    for kind in ["positive", "negative"]]
+    return f"You are an expert travel agent. A customer is looking for a hotel in the city of {city}. " \
+           f"You need to convince the customer to book a stay in {hotel_name}. " \
+           f"Rely on the following positive and negative reviews. " \
+           f"Make your argument compelling yet honest. " \
+           f"User reviews: " \
+           f"{positive_text}" \
+           f"{negative_text}."