Feedback changes

app.py

@@ -2,10 +2,10 @@ import streamlit as st
 import os
 import math
 import google.generativeai as genai
-GOOGLE_API_KEY= os.getenv('GEMINI_API_KEY')
+GOOGLE_API_KEY=os.getenv('GEMINI_API_KEY')
 genai.configure(api_key=GOOGLE_API_KEY)
 model = genai.GenerativeModel(model_name = "gemini-pro")
-from utils import findTop3MoneyNeeds, findTop3Topologies, findTop3CustomerExperienceNeeds, findTop3SustainabilityNeeds
+from utils import findTop3MoneyNeeds, findTop3Topologies, findTop3CustomerExperienceNeeds, findTop3SustainabilityNeeds, listNeeds, generatePropositionExample
 
 
 # Create a banner using Markdown
@@ -25,13 +25,42 @@ st.markdown(
 
 selectedCity = st.selectbox("Please select the City and the Bank Product for Your Proposition.", ["CharlesTown", "Limburg"])
 selectedProduct = st.selectbox("Please select the Product", ["Current", "Mortage", "Credit Card", "Crypto"])
-subscriberTakeOut = st.number_input("Please enter your subscriber take out")
-moneyNeeds = st.text_area("Describe money needs of your target audience. For example do they spend a lot on education, healthcare, gym, eating out etc.")
-customerExperience = st.text_area("Describe the customer experience needs of your target audience.")
-sutainabilityNeeds = st.text_area("Describe the sutainability needs of your target audience.")
+subscriberTakeOut = st.number_input("Please enter your subscriber take up. Enter the amount in multiples of 1000.", min_value=0, step=1)
+productName = st.text_input("Please enter your product name")
+
+moneyNeedsList, moneyNeedsDict = listNeeds('money_needs')
+customerExperienceList, customerExperienceDict = listNeeds('customer_exp')
+sutainabilityNeedsList, sutainabilityNeedsDict = listNeeds('sustainability')
+
+moneyNeeds = st.multiselect(
+    label="Select three money needs of your target audience which best describe their spending",
+    options=moneyNeedsList,
+    max_selections=3
+)
+
+customerExperience = st.multiselect(
+    label="Select three options which best describe the cuetomer experience you are targeting to provide",
+    options=customerExperienceList,
+    max_selections=3
+)
+
+sutainabilityNeeds = st.multiselect(
+    label="Select three options which best describe the sustainability needs of your customer",
+    options=sutainabilityNeedsList,
+    max_selections=3
+)
+
+
+#moneyNeeds = st.text_area("Describe money needs of your target audience. For example do they spend a lot on education, healthcare, gym, eating out etc.")
+#customerExperience = st.text_area("Describe the customer experience needs of your target audience.")
+#sutainabilityNeeds = st.text_area("Describe the sutainability needs of your target audience.")
+
+submit_button_details = st.button("Submit details to generate an example proposition")
+
+userProposal = st.text_area("Enter your final Proposition for Select City and Product", value=st.session_state.get("propositionVal", ""), disabled=st.session_state.get("disabled", True))
+
+submit_button_proposition = st.button("Submit Proposition")
 
-userProposal = st.text_area("Enter your final Proposition for Select City and Product")
-submit_button = st.button("Submit")
 pre_prompt = '''You are a business analyser bot. I will give you data and references and you will analyse the propositions and will give them scoring on particular criterias. An example is below
 
 Proposition - Culture bank Student – the Young Professionals Credit Card!
@@ -96,17 +125,24 @@ if selectedCity:
         demographic = LimburgTownDemographic
         population = LimburgTownPopulation
 
-if submit_button:
-    proposal = '''Given proposal is for the city {} with product {}. The propsal is as below.
-    {}'''
-    proposal = proposal.format(selectedCity, selectedProduct, userProposal)
+if submit_button_details:
+    st.session_state.disabled = False
+    
+
+    #topMoneyNeeds = moneyNeeds #findTop3MoneyNeeds(moneyNeeds)
+
+    #topCustomerExp = customerExperience #findTop3CustomerExperienceNeeds(customerExperience)
 
-   
-    topMoneyNeeds, topMoneyNeedsDict = findTop3MoneyNeeds(moneyNeeds)
+    #topSustainabilityNeeds = sutainabilityNeeds #findTop3SustainabilityNeeds(sutainabilityNeeds)
 
-    topCustomerExp, topCustomerExpDict = findTop3CustomerExperienceNeeds(customerExperience)
+    st.session_state.propositionVal = generatePropositionExample(productName, selectedProduct, moneyNeeds, customerExperience, sutainabilityNeeds)
 
-    topSustainabilityNeeds, topSustainabilityNeedsDict = findTop3SustainabilityNeeds(sutainabilityNeeds)
+    
+
+if submit_button_proposition:
+    proposal = '''Given proposal is for the city {} with product {}. The propsal is as below.
+    {}'''
+    proposal = proposal.format(selectedCity, selectedProduct, userProposal)
 
     matchingTopologies, topologyDetails = findTop3Topologies(proposal, demographic)
 
@@ -118,17 +154,17 @@ if submit_button:
 
     st.write("As for the money needs of this group your product is mostly targeting people with the spending behaviour described below:")
     
-    for idx, need in enumerate(topMoneyNeeds):
+    for idx, need in enumerate(moneyNeeds):
         st.write("{}. {}".format(idx+1, need))
 
     st.write("As for the customer experience needs of this group your product is mostly targeting people with the spending behaviour described below:",)
     
-    for idx, exp in enumerate(topCustomerExp):
+    for idx, exp in enumerate(customerExperience):
         st.write("{}. {}".format(idx+1, exp))
 
     st.write("As for the sustainability needs of this group your product is mostly targeting people with the spending behaviour described below:",)
     
-    for idx, need in enumerate(topSustainabilityNeeds):
+    for idx, need in enumerate(sutainabilityNeeds):
         st.write("{}. {}".format(idx+1, need))
 
 
@@ -141,14 +177,15 @@ if submit_button:
 
     for topology in matchingTopologies:
         sumTopology = 0
-        for moneyNeed in topMoneyNeedsDict:
-            sumTopology = sumTopology+int(moneyNeed[topology])
+        for moneyNeed in moneyNeeds:
+            #print(" Money need = {}, Topology is {}".format(moneyNeed, topology))
+            sumTopology = sumTopology+int(moneyNeedsDict[moneyNeed][topology])
         
-        for customerExp in topCustomerExpDict:
-            sumTopology = sumTopology+int(customerExp[topology])
+        for customerExp in customerExperience:
+            sumTopology = sumTopology+int(customerExperienceDict[customerExp][topology])
 
-        for sustainabilityNeed in topSustainabilityNeedsDict:
-            sumTopology = sumTopology+int(sustainabilityNeed[topology])
+        for sustainabilityNeed in sutainabilityNeeds:
+            sumTopology = sumTopology+int(sutainabilityNeedsDict[sustainabilityNeed][topology])
 
         topologySumDict[topology] = math.floor(sumTopology/3)
 
@@ -206,7 +243,6 @@ if submit_button:
     else:
         st.write("Congratulations!! Your proposition exceeds the target subscriber take out. Additional takeout = {}".format(totalSubscriberTakeOut- subscriberTakeOut))
 
-
 # 250 and below with a negative factor of 2.0
 # 260 with a negative factor of 1.8
 # 270 with a negative factor of 1.6

utils.py

@@ -76,6 +76,11 @@ def transform_topologies_to_dict(columns, rows):
 
     return result
 
+def listNeeds(tableName, dbName='data.sqlite'):
+    needs, rows = fetch_db_rows_as_dicts(dbName, tableName)
+    needsDict = transform_to_dict_of_dicts(needs, rows)
+    return list(needsDict.keys()), needsDict
+
 def findTop3MoneyNeeds(proposition):
     moneyNeeds, rows = fetch_db_rows_as_dicts('data.sqlite', 'money_needs')
     moneyNeedsDict = transform_to_dict_of_dicts(moneyNeeds, rows)
@@ -188,6 +193,26 @@ def findTop3Topologies(proposition, demographic):
     return obj['matches'], topologyDetails
 
 
+def generatePropositionExample(productName, selectedProduct, moneyNeeds, customerExperience, sutainabilityNeeds):
+
+    proposal = '''You are a business sales professional who can form propostion summary of 100 words based upon the details.
+    Please take the below details and summarize a propostion in less than 100 words.
+
+    product name = {}
+    
+    product type = {}
+
+    money needs of customer which this product is supposed to target = {}
+
+    Customer experience needs which our company will provide = {}
+
+    Sustainability needs which our product takes care of = {}
+    '''
+    proposal = proposal.format(productName, selectedProduct, moneyNeeds, customerExperience, sutainabilityNeeds)
+    response = model.generate_content([proposal])
+    return response.text
+
+
 # def findTop3Needs(proposition, moneyNeeds):
     
 #     moneyNeedsString = concatenate_keys(moneyNeeds)