augmented generation

app.py

@@ -1,4 +1,4 @@
-import os 
+import os
 
 import google.generativeai as palm
 import streamlit as st
@@ -24,8 +24,51 @@ This is what amazon customers are saying about similar products:
 Can you write some recomendations about how can we disrupt this market? Try to propose the necesary methodology to create a breaking product."""
 
 
+def get_prompt_without_reviews(title):
+    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}"
+
+Can you write some recomendations about how can we disrupt this market? Try to propose the necesary methodology to create a breaking product."""
+
+
+no_electronics_message = """
+Sorry, we are currently only recommending business that operate around electronics. Would you like to input another search?
+
+
+This doesn't mean you make a mistake, I search amazon products and try to extract relevant reviews from similar products and we didn't find relevant products for your search.
+
+#### Maybe you are way ahead of the market!
+
+
+```
+.                                           
+                                      ___,,,
+                                      \_[o o]
+     Errare humanum est!              C\  _\/
+             /                     _____),_/__
+        ________                  /     \/   /
+      _|       .|                /      o   /
+     | |       .|               /          /
+      \|       .|              /          /
+       |________|             /_        \/
+       __|___|__             _//\        \\
+ _____|_________|____       \  \ \        \\
+                    _|       ///  \        \\
+                   |               \       /
+                   |               /      /
+                   |              /      /
+ ________________  |             /__    /_
+              ...|_|.............. /______\.......
+```
+"""
+
+TEST_MODE = False
+
+
 def setup_palm():
-    palm.configure(api_key=os.environ.get('PALM_TOKEN'))
+    # palm.configure(api_key=os.environ.get('PALM_TOKEN'))
+    palm.configure(api_key="AIzaSyCbbwbNqjqPYzLOx9-5iO9zaYeHG5CGNxI")
+
 
 @st.cache_data
 def load_data():
@@ -81,12 +124,17 @@ def handler_review_query():
     )
 
     # TODO: check if there are relevant products
+    if len(relevant_products) == 0:
+        st.session_state.user_prompt = None
+        st.session_state.palm_output = no_electronics_message
+        return
 
     relevant_reviews = deploy_utils.get_relevant_reviews(
         relevant_products, reviews)
 
     raw_topic_assigment = deploy_utils.clusterize_reviews(
         relevant_reviews, reviews_model, clusterer)
+
     relevant_reviews["topic"] = raw_topic_assigment
     reviews_with_topics = relevant_reviews[relevant_reviews["topic"] != -1]
 
@@ -104,10 +152,14 @@ def handler_review_query():
     st.session_state.user_prompt = prompt
 
 
+def handler_product_without_reviews():
+    st.session_state.user_prompt = get_prompt_without_reviews(
+        st.session_state.user_search_query)
+
 
 def palm_handler():
     response = palm.generate_text(prompt=st.session_state.user_prompt)
-    st.session_state.palm_output = response
+    st.session_state.palm_output = response.result
 
 
 def render_search():
@@ -122,18 +174,20 @@ def render_search():
         )
 
         if query:
-            handler_review_query()
+            try:
+                handler_review_query()
+            except:
+                handler_product_without_reviews()
 
-        st.text_area(
-            label="test env",
-            placeholder="prompt here",
-            key="user_prompt"
-        )
+        if TEST_MODE:
+            _ = st.text_area(
+                label="test env",
+                placeholder="prompt here",
+                key="user_prompt"
+            )
 
-        st.button(
-            label="Generate",
-            key="location_search",
-            on_click=palm_handler)
+        if "user_prompt" in st.session_state and st.session_state.user_prompt:
+            palm_handler()
 
         st.write("---")
         render_cta_link(
@@ -146,7 +200,7 @@ def render_search():
 def render_palm_results():
     # TODO: temporal
     st.write("# ALMond recommendations")
-    st.write(st.session_state.palm_output.result)
+    st.write(st.session_state.palm_output)
 
 # Execution start here!