bug fixes. initial like buttons

app.py

@@ -80,7 +80,8 @@ def card(company_id, name, description, score, data_type, region, country, metad
                     <span>{business_model}</span>
              </div>
              <div  class="col-md-1 col-sm-1">
-                    <button onclick="like_company({name});">
+                    <button type='button' onclick="like_company({company_id});">Like</button>
+                    <button type='button' onclick="dislike_company({company_id});">DisLike</button>
              </div>
     """             
 
@@ -91,7 +92,7 @@ def card(company_id, name, description, score, data_type, region, country, metad
             <span>[Score: {score}</span>
         </div>        
         """             
-    markdown = markdown + "</div></div>"
+    markdown = markdown + "</div>"
     return st.markdown(markdown, unsafe_allow_html=True)
 
 
@@ -117,10 +118,7 @@ def call_openai(prompt, engine="gpt-3.5-turbo", temp=0, top_p=1.0, max_tokens=40
     try:
         response = openai.ChatCompletion.create(
             model=engine,
-            messages=[
-                {"role": "system", "content": "You are an assistant analyzing startup companies for investments."},
-                {"role": "user", "content": prompt}
-            ],
+            messages=[{"role": "user", "content": prompt}],
             temperature=temp,
             max_tokens=max_tokens
         )
@@ -166,11 +164,13 @@ def run_query(query, prompt, scrape_boost, top_k , regions, countries, is_debug,
             answer['id'] = match['id']
         answer["name"] = match["metadata"]['company_name']
         answer["description"] = match["metadata"]['description'] if "description" in match['metadata'] else ""
-        data = {"Summary": match["metadata"]["summary"]}
-        try:                
-            data = json.loads(match["metadata"]["summary"])
-        except Exception as e:
-            pass            
+        data = None
+        data = {"Summary": match["metadata"]["summary"]} 
+        if 'summary' in match['metadata']:
+            try:                
+                data = json.loads(match["metadata"]["summary"])
+            except Exception as e:
+                pass                
         answer['data'] = data
         
         results.append(answer)
@@ -187,8 +187,11 @@ def run_query(query, prompt, scrape_boost, top_k , regions, countries, is_debug,
     User query: {query} 
     """    
     prompt_template = PromptTemplate(template=prompt_txt, input_variables=["descriptions", "query"])
-    prompt = prompt_template.format(descriptions = [f"{res['name']}: {res['data']['Summary']}" for res in results[:10]], query = query)
-    m_text = call_openai(prompt, engine="gpt-3.5-turbo-16k", temp=0, top_p=1.0, max_tokens=16192)
+    descriptions = str([f"{res['name']}: {res['data']['Summary']}" for res in results[:20] if 'Summary' in res['data']])
+    ntokens = len(descriptions.split(" "))
+    print(f"#Tokens {ntokens}:\n {descriptions}")    
+    prompt = prompt_template.format(descriptions = descriptions, query = query)
+    m_text = call_openai(prompt, engine="gpt-3.5-turbo-16k", temp=0, top_p=1.0)
 
     m_text
     
@@ -297,7 +300,15 @@ if check_password():
 
 
     st.markdown(
-        f'''
+        '''
+            <script>
+                function like_company(company_id) {
+                    console.log("Company " + company_id + " Liked!");
+                    }            
+                function dislike_company(company_id) {
+                    console.log("Company " + company_id + " Disliked!");
+                    }            
+            </script>                    
             <style>
                 .sidebar .sidebar-content {{
                     width: 375px;
@@ -321,6 +332,8 @@ if check_password():
         top_k = st.number_input('# Top Results', value=20)
         is_debug = st.checkbox("Debug output", value = False, key="debug")
         index_namespace = st.selectbox(label="Data Type", options=["websummarized", "web", "cbli", "all"], index=0)
+        liked_companies = st.text_input(label="liked companies", key='liked_companies')
+        disliked_companies = st.text_input(label="disliked companies", key='disliked_companies')