Update app.py

app.py

@@ -97,25 +97,21 @@ if input:
 
     model_answer = find_sentences("How can I extract information from invoices?")
 
-
-    if resp.status_code == 200:
+    key_exists = False
+    for key in link_dict:
+        if key in model_answer:
+            key_exists = True
+            url = link_dict[key]
+            response_templates = [f"I think that {model_answer} is the best task for this 🤩 Check out the page 👉🏼 {url}", f"I think you should use {model_answer} 🪄 Check it out here 👉🏼 {url}", f"I think {model_answer} should work for you 🤓 Check out the page 👉🏼 {url}"]
     
-
-        key_exists = False
-        for key in link_dict:
-            if key in model_answer:
-                key_exists = True
-                url = link_dict[key]
-                response_templates = [f"I think that {model_answer} is the best task for this 🤩 Check out the page 👉🏼 {url}", f"I think you should use {model_answer} 🪄 Check it out here 👉🏼 {url}", f"I think {model_answer} should work for you 🤓 Check out the page 👉🏼 {url}"]
-        
-                bot_answer = random.choice(response_templates)
-                message_history.append({"text":bot_answer, "is_user" : False})
-        if key_exists == False:
-            fallback_template = ["I didn't get the question 🧐 Could you please ask again? Try 'What should I use for detecting masks in an image?'",
-                                     "Hmm, not sure I know the answer, maybe you could ask differently? 🤓",
-                                     "Sorry, I didn't understand you, maybe you could ask differently? 🤓 Try asking 'What should I use to extract name in a document' 🤗"]
-            bot_answer = random.choice(fallback_template)
+            bot_answer = random.choice(response_templates)
             message_history.append({"text":bot_answer, "is_user" : False})
+    if key_exists == False:
+        fallback_template = ["I didn't get the question 🧐 Could you please ask again? Try 'What should I use for detecting masks in an image?'",
+                                 "Hmm, not sure I know the answer, maybe you could ask differently? 🤓",
+                                 "Sorry, I didn't understand you, maybe you could ask differently? 🤓 Try asking 'What should I use to extract name in a document' 🤗"]
+        bot_answer = random.choice(fallback_template)
+        message_history.append({"text":bot_answer, "is_user" : False})
     
     with placeholder.container():
         last_message = message_history[-1]