add llm output

app.py

@@ -1,3 +1,6 @@
+import os 
+
+import google.generativeai as palm
 import streamlit as st
 import pandas as pd
 import faiss
@@ -12,8 +15,10 @@ 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 = "Search for similar products"
+search_label = "Generate"
 
+def setup_palm():
+    palm.configure(api_key=os.environ.get('PALM_TOKEN'))
 
 @st.cache_data
 def load_data():
@@ -91,6 +96,11 @@ def handler_search():
     print('search done')
 
 
+def palm_handler():
+    response = palm.generate_text(prompt=st.session_state.user_prompt)
+    st.session_state.palm_output = response
+
+
 def render_search():
     """
     Render the search form in the sidebar.
@@ -102,10 +112,15 @@ def render_search():
             key="user_search_query",
         )
 
+        st.text_area(
+            placeholder="prompt here",
+            key="user_prompt"
+        )
+
         st.button(
             label=search_label,
             key="location_search",
-            on_click=handler_search)
+            on_click=palm_handler)
 
         st.write("---")
         render_cta_link(
@@ -117,10 +132,8 @@ def render_search():
 
 def render_results():
     # TODO: temporal
-    st.write("# Relevant reviews")
-    for rev in st.session_state.key_reviews:
-        st.write(" *", rev.split("\n")[0])
-
+    st.write("# PaLM outputs")
+    st.write(st.session_state.palm_output)
 
 # Execution start here!
 
@@ -131,11 +144,11 @@ st.set_page_config(
     initial_sidebar_state="expanded",
 )
 
-
+setup_palm()
 reviews, products = load_data()
 reviews_model, product_model, product_indexer = load_models()
 topic_extractor, clusterer = load_uncached_models()
 
 render_search()
-if "key_reviews" in st.session_state:
+if "palm_output" in st.session_state:
     render_results()