Update app.py

app.py

@@ -58,15 +58,17 @@ with gr.Blocks() as demo:
 
     def _retrieve(message, chat_history):
         idx = int(message)
-        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
-    
+        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())
     def respond(message, chat_history):
         print(f"received input '{message}'")
         try: