augmented generation

app.py

@@ -13,9 +13,16 @@ from src.modelling.topics.topic_extractor import (
 from src.modelling.topics.class_tf_idf import ClassTfidfTransformer
 from src import deploy_utils
 
-semantic_search_header = "What kind of product are you trying to sell?"
-semantic_search_placeholder = "Your magic idea goes here ✨"
-search_label = "Generate"
+
+def get_prompt(title, reviews):
+    return f"""We are doing a marketing research analysis, in particular we are trying to understand what users thing about a particular market in order to generate tips for future sellers.
+In particular, we are interesting to analyze the market for "{title}"
+
+This is what amazon customers are saying about similar products:
+{reviews}
+
+Can you write some recomendations about how can we disrupt this market? Try to propose the necesary methodology to create a breaking product."""
+
 
 def setup_palm():
     palm.configure(api_key=os.environ.get('PALM_TOKEN'))
@@ -65,7 +72,7 @@ def render_cta_link(url, label, font_awesome_icon):
     return st.markdown(button_code, unsafe_allow_html=True)
 
 
-def handler_search():
+def handler_review_query():
     relevant_products = deploy_utils.query_relevant_documents(
         product_model=product_model,
         indexer=product_indexer,
@@ -92,8 +99,10 @@ def handler_search():
         extracted_topics,
     )
 
-    st.session_state.key_reviews = key_reviews
-    print('search done')
+    reviews_prompt = deploy_utils.key_reviews_to_prompt(key_reviews)
+    prompt = get_prompt(st.session_state.user_search_query, reviews_prompt)
+    st.session_state.user_prompt = prompt
+
 
 
 def palm_handler():
@@ -106,12 +115,15 @@ def render_search():
     Render the search form in the sidebar.
     """
     with st.sidebar:
-        st.text_input(
-            label=semantic_search_header,
-            placeholder=semantic_search_placeholder,
+        query = st.text_input(
+            label="What kind of product are you trying to sell?",
+            placeholder="Your magic idea goes here ✨",
             key="user_search_query",
         )
 
+        if query:
+            handler_review_query()
+
         st.text_area(
             label="test env",
             placeholder="prompt here",
@@ -119,7 +131,7 @@ def render_search():
         )
 
         st.button(
-            label=search_label,
+            label="Generate",
             key="location_search",
             on_click=palm_handler)
 
@@ -131,9 +143,9 @@ def render_search():
         )
 
 
-def render_results():
+def render_palm_results():
     # TODO: temporal
-    st.write("# PaLM outputs")
+    st.write("# ALMond recommendations")
     st.write(st.session_state.palm_output.result)
 
 # Execution start here!
@@ -152,4 +164,4 @@ topic_extractor, clusterer = load_uncached_models()
 
 render_search()
 if "palm_output" in st.session_state:
-    render_results()
+    render_palm_results()

src/deploy_utils.py

@@ -95,4 +95,15 @@ def get_key_reviews(
         for idx in indices
     }
 
-    return list(top_rated_reviews | representative_reviews)
+    return list(top_rated_reviews | representative_reviews)
+
+
+def _format_review(x):
+    single_line = x.split("\n")[0]
+    return f' - {single_line.strip()}'
+
+
+def key_reviews_to_prompt(reviews):
+    return '\n'.join([
+        _format_review(i) for i in reviews
+    ])