Update app.py

app.py

@@ -58,17 +58,19 @@ with gr.Blocks() as demo:
 
     def _retrieve(message, chat_history):
         idx = int(message)
-        try:
-            for _, m in chat_history[::-1]:
-                if m.startswith("results:\n"):
-                    for n in m.split("\n")[1:]:
-                        if str(idx) == n.split(".")[0]:
-                            df_idx = int(n.split(" // ")[-1])
-                            article = df.iloc[df_idx]['text'].replace("==", "\n\n==")
-                            chat_history.append((message, f"contents of {n}:\n{article}"))
-                            return "", chat_history
-        except:
-            print(traceback.format_exc())
+        for _, m in chat_history[::-1]:
+            if m.startswith("results:\n"):
+                for n in m.split("\n")[1:]:
+                    print(n)
+                    if str(idx) == n.split(".")[0]:
+                        df_idx = int(n.split(" // ")[-1])
+                        print(df_idx)
+                        article = df.iloc[df_idx]['text'].replace("==", "\n\n==")
+                        chat_history.append((message, f"contents of {n}:\n{article}"))
+                        return "", chat_history
+        print("nothing found")
+        return chat_history.append((message, "article not found"))
+    
     def respond(message, chat_history):
         print(f"received input '{message}'")
         try: