Update app.py

app.py

@@ -59,10 +59,15 @@ with gr.Blocks() as demo:
     def _retrieve(message, chat_history):
         idx = int(message)
         for _, m in chat_history[::-1]:
+            print("chat message: ", m)
             if m.startswith("results:\n"):
+                print("found results")
                 for n in m.split("\n")[1:]:
+                    print("result item: ", n)
                     if str(idx) == n.split(".")[0]:
+                        print("found the number")
                         df_idx = int(n.split(" // ")[-1])
+                        print("df_idx", df_idx)
                         chat_history.append((message, f"contents of {n}:\n{df[df_idx]}"))
                         return "", chat_history
     
@@ -72,7 +77,7 @@ with gr.Blocks() as demo:
             int(message)
             print(f"retrieving #{message}")
             return _retrieve(message, chat_history)
-        except:
+        except ValueError:
             print(traceback.format_exc())
             print(f"searching for {message}")
             return _search(message, chat_history)