Update app.py

app.py

@@ -186,7 +186,7 @@ QUESTION = st.chat_input('Ask a question -e.g How to prepare for Verbal section
 #QUESTION=st.text_area('Ask a question -e.g How to prepare for Verbal section for CAT?')  ##' How to prepare for Verbal section ?'
 score=0
 testing=True
-
+ext_url=''
 
 if QUESTION:
 
@@ -278,6 +278,25 @@ if QUESTION:
         else:
             ans='Weak match to your query. Please try reframing your question'
 
+            ## Call Tavily
+            tavily_key= os.environ["TAVILY_KEY"]
+
+            tavily = TavilyClient(api_key=tavily_key)
+
+            success= 0
+            while success<3:
+                success+=1
+                try:
+                    resp=tavily.search(query=QUESTION)
+                    with st.chat_message("assistant"):
+                        ans=resp['results'][0]['content']
+                        
+                    ext_url= resp['results'][0]['url']
+                    break
+                except:
+                    pass
+                        
+
             #st.write("Matched URL:{}  Score:{}".format(url,score))
             testing = False
             if testing:
@@ -285,6 +304,8 @@ if QUESTION:
                     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']))
+
+                    
         ##  Send RESPONSE
         with st.chat_message("assistant"):
             response = st.write_stream(response_generator(ans))
@@ -301,9 +322,11 @@ if QUESTION:
                             break
                     st.write("["+disp_title+"]("+disp_url+")")
 
-            else:
-                st.write('Resources:',top_2[0]['score'], ' confidence')
-                st.write(top_2[0]['url'])                
+            else: # not a good match
+                
+                if len(ext_url)>5:    
+                    st.write('External Site:',ext_url)
+                #st.write(top_2[0]['url'])                
             # Add assistant response to chat history
             st.session_state.messages.append({"role": "assistant", "content": response})
         #st.write(ans)