Update app.py

app.py

@@ -194,77 +194,78 @@ if QUESTION:
         print(d['url'], score, r['score'], meta_score)
         resplist.append(d)
 
-    sorted_indices = sorted(range(len(resplist)), key=lambda i: resplist[i]['score'], reverse=True)
-
-    # Get the elements with the top 2 highest values
-    top_2 =  [resplist[i] for i in sorted_indices[:2]]
-            
-    # covert to array
+    if len(resplist)>0:
+        sorted_indices = sorted(range(len(resplist)), key=lambda i: resplist[i]['score'], reverse=True)
     
-    json_data = json.dumps(top_2)
-
-    st.write(json_data)
-    goodmatch=False    
-    if resplist[sorted_indices[0]]['score']>.5:
-        goodmatch=True
-        mode = "two"  # two passages
-
-        client = OpenAI()
-
-        if mode=="one":
-            instr=system_instructions_text
-
-            out= resplist[sorted_indices[0]]['content']
-            content="""
-            <text>
-            {}
-            </text>
-            """.format(out)
-            
-        if mode=="two":
-            instr=json_instructions
-            content=json_data
-
-        response = client.chat.completions.create(
-          model="gpt-3.5-turbo",
-          messages=[
-            {"role": "system", "content":instr },
-            {"role": "user", "content": content},
-            {"role": "user", "content": "Question:"+QUESTION}
-          ]
-        )
-
-        ans= response.choices[0].message.content
-    else:
-        ans='Weak match to your query. Please try reframing your question'
-
-        #st.write("Matched URL:{}  Score:{}".format(url,score))
-        testing = False
-        if testing:
-            if len(resp)>=1:
-                st.write("2nd Matched URL:{}  Score:{}".format(resp[1]['id'],resp[1]['score']))
-            if len(resp)>=2:
-                st.write("3rd Matched URL:{}  Score:{}".format(resp[2]['id'],resp[2]['score']))
-
-        with st.chat_message("assistant"):
-            response = st.write_stream(response_generator(ans))
-            if goodmatch:
-                st.write('Resources:'+top_2[0]['url'])
-                st.write(top_2[1]['url'])
-            # Add assistant response to chat history
-            st.session_state.messages.append({"role": "assistant", "content": response})
-        #st.write(ans)
-
-        #st.write(' ----------------------')
-        #st.write(out)
-
-        now= str(datetime.utcnow())
-        df_log.loc[len(df_log)]=[QUESTION,url,score,ans,now]
-        write_log(QUESTION,url, score, ans, now)
-        #df.to_csv("hf://datasets/sujitb/data/test.csv")
-        #df_log.to_csv("hf://datasets/sujitb/data/"+logfile)
+        # Get the elements with the top 2 highest values
+        top_2 =  [resplist[i] for i in sorted_indices[:2]]
+                
+        # covert to array
+        
+        json_data = json.dumps(top_2)
+    
+        st.write(json_data)
+        goodmatch=False    
+        if resplist[sorted_indices[0]]['score']>.5:
+            goodmatch=True
+            mode = "two"  # two passages
+    
+            client = OpenAI()
+    
+            if mode=="one":
+                instr=system_instructions_text
+    
+                out= resplist[sorted_indices[0]]['content']
+                content="""
+                <text>
+                {}
+                </text>
+                """.format(out)
+                
+            if mode=="two":
+                instr=json_instructions
+                content=json_data
+    
+            response = client.chat.completions.create(
+              model="gpt-3.5-turbo",
+              messages=[
+                {"role": "system", "content":instr },
+                {"role": "user", "content": content},
+                {"role": "user", "content": "Question:"+QUESTION}
+              ]
+            )
+    
+            ans= response.choices[0].message.content
+        else:
+            ans='Weak match to your query. Please try reframing your question'
+    
+            #st.write("Matched URL:{}  Score:{}".format(url,score))
+            testing = False
+            if testing:
+                if len(resp)>=1:
+                    st.write("2nd Matched URL:{}  Score:{}".format(resp[1]['id'],resp[1]['score']))
+                if len(resp)>=2:
+                    st.write("3rd Matched URL:{}  Score:{}".format(resp[2]['id'],resp[2]['score']))
+    
+            with st.chat_message("assistant"):
+                response = st.write_stream(response_generator(ans))
+                if goodmatch:
+                    st.write('Resources:'+top_2[0]['url'])
+                    st.write(top_2[1]['url'])
+                # Add assistant response to chat history
+                st.session_state.messages.append({"role": "assistant", "content": response})
+            #st.write(ans)
+    
+            #st.write(' ----------------------')
+            #st.write(out)
+    
+            now= str(datetime.utcnow())
+            df_log.loc[len(df_log)]=[QUESTION,url,score,ans,now]
+            write_log(QUESTION,url, score, ans, now)
+            #df.to_csv("hf://datasets/sujitb/data/test.csv")
+            #df_log.to_csv("hf://datasets/sujitb/data/"+logfile)
 
-    else:
+    else:  ## Zero response from pinecone query 
         #st.write("No matches for query")
         ans= "No matches for query"
         response = st.write_stream(response_generator(ans))