ok

paper_chat_tab.py

@@ -257,7 +257,7 @@ def create_chat_interface(provider_dropdown, model_dropdown, paper_content, hf_t
         except Exception as ex:
             yield f"Unexpected error: {ex}"
 
-    chatbot = gr.Chatbot(label="Chatbot", scale=1, height=400, autoscroll=True)
+    chatbot = gr.Chatbot(label="Chatbot", scale=1, height=800, autoscroll=True)
     chat_interface = gr.ChatInterface(
         fn=get_fn,
         chatbot=chatbot,
@@ -268,13 +268,15 @@ def create_chat_interface(provider_dropdown, model_dropdown, paper_content, hf_t
 
 
 def paper_chat_tab(paper_id, paper_from, paper_central_df):
+    # A top-level button to "Chat with another paper" (visible only if paper_id is set)
+    # We'll place it above everything
+    chat_another_button = gr.Button("Chat with another paper", variant="primary", visible=False)
+
     # First row with two columns
     with gr.Row():
         # Left column: Paper selection and display
         with gr.Column(scale=1):
-            gr.Markdown("### Select a Paper")
             todays_date = datetime.today().strftime('%Y-%m-%d')
-
             # Filter papers for today's date and having a paper_page
             selectable_papers = paper_central_df.df_prettified
             selectable_papers = selectable_papers[
@@ -290,10 +292,17 @@ def paper_chat_tab(paper_id, paper_from, paper_central_df):
                 paper_choices = [("No available papers for today", "")]
 
             paper_select = gr.Dropdown(
-                label="Select a paper to chat with:",
+                label="Select a paper to chat with: (from today's 🤗 hugging face paper page)",
                 choices=[p[0] for p in paper_choices],
                 value=paper_choices[0][0] if paper_choices else None
             )
+
+            # Add a textbox for user to enter a paper_id (arxiv_id)
+            paper_id_input = gr.Textbox(
+                label="Or enter a 🤗 paper_id (arxiv_id) directly",
+                placeholder="e.g. 1234.56789"
+            )
+
             select_paper_button = gr.Button("Load this paper")
 
             # Paper info display
@@ -345,7 +354,6 @@ def paper_chat_tab(paper_id, paper_from, paper_central_df):
         provider_info = PROVIDERS[selected_provider]
         models = provider_info['models']
         logo_url = provider_info['logo']
-        chatbot_message_type = provider_info['type']
         max_total_tokens = provider_info['max_total_tokens']
 
         model_dropdown_choices = gr.update(choices=models, value=models[0])
@@ -357,7 +365,8 @@ def paper_chat_tab(paper_id, paper_from, paper_central_df):
             placeholder=f"Enter your {selected_provider} API token to avoid rate limits"
         )
         chatbot_reset = []
-        return model_dropdown_choices, logo_html_update, note_markdown_update, hf_token_input_update, chatbot_message_type, max_total_tokens, chatbot_reset
+        return model_dropdown_choices, logo_html_update, note_markdown_update, hf_token_input_update, provider_info[
+            'type'], max_total_tokens, chatbot_reset
 
     provider_dropdown.change(
         fn=update_provider,
@@ -394,19 +403,44 @@ def paper_chat_tab(paper_id, paper_from, paper_central_df):
 
         return gr.update(value=card_html), text, []
 
-    def select_paper(paper_title):
-        for t, ppage in paper_choices:
-            if t == paper_title:
-                return ppage, "paper_page"
-        return "", ""
+    def select_paper(paper_title, paper_id_val):
+        # If user provided a paper_id_val (arxiv_id), use that
+        if paper_id_val and paper_id_val.strip():
+            # Check if it exists in df as a paper with paper_page not None
+            df = paper_central_df.df_raw
+            # We assume `arxiv_id` column exists in df (the user requested checking arxiv_id)
+            # If not present, you must ensure `paper_central_df` has `arxiv_id` column.
+            if 'arxiv_id' not in df.columns:
+                return gr.update(value="<div>arxiv_id column not found in dataset</div>"), None
+
+            found = df[
+                (df['arxiv_id'] == paper_id_val.strip()) &
+                df['paper_page'].notna() & (df['paper_page'] != "")
+                ]
+
+            if len(found) > 0:
+                # We found a matching paper
+                return paper_id_val.strip(), "paper_page"
+            else:
+                # Not found, show error in content
+                # We can't directly show error from here. We'll return something that doesn't update states and rely on error message
+                # Let's return empty paper_id and paper_from but we must also show error in content after this call
+                return "", ""
+        else:
+            # fallback to dropdown selection
+            for t, ppage in paper_choices:
+                if t == paper_title:
+                    return ppage, "paper_page"
+            return "", ""
 
     select_paper_button.click(
         fn=select_paper,
-        inputs=[paper_select],
+        inputs=[paper_select, paper_id_input],
         outputs=[paper_id, paper_from]
     )
 
-    paper_id.change(
+    # After the paper_id/paper_from are set, we update paper info
+    paper_id_update = paper_id.change(
         fn=update_paper_info,
         inputs=[paper_id, paper_from, model_dropdown, content],
         outputs=[content, paper_content, chatbot]
@@ -418,20 +452,60 @@ def paper_chat_tab(paper_id, paper_from, paper_central_df):
         else:
             return gr.update(visible=False)
 
-    # Toggle provider section visibility
     paper_id.change(
         fn=toggle_provider_visibility,
         inputs=[paper_id],
         outputs=[provider_section]
     )
 
-    # Toggle chat row visibility
     paper_id.change(
         fn=toggle_provider_visibility,
         inputs=[paper_id],
         outputs=[chat_row]
     )
 
+    # Show/hide the "Chat with another paper" button
+    # If paper_id is set, show it. If not, hide it.
+    def toggle_chat_another_button(paper_id_value):
+        if paper_id_value and paper_id_value.strip():
+            return gr.update(visible=True)
+        else:
+            return gr.update(visible=False)
+
+    paper_id.change(
+        fn=toggle_chat_another_button,
+        inputs=[paper_id],
+        outputs=[chat_another_button]
+    )
+
+    # Button action to reset paper_id to None
+    def reset_paper_id():
+        # reset paper_id to ""
+        return "", "neurips", gr.update(value="<div></div>")
+
+    # When this button is clicked, we reset the paper_id and content
+    chat_another_button.click(
+        fn=reset_paper_id,
+        outputs=[paper_id, paper_from, content]
+    )
+
+    # If user tried an invalid paper_id_input, no error was shown yet:
+    # Actually we can show error message if no paper selected by updating after select_paper_button
+    # The select_paper returns paper_id/paper_from. If empty means error:
+    def check_paper_id_error(p_id, p_from):
+        # If p_id is empty after clicking load, show error message
+        if not p_id:
+            return gr.update(value="<div style='color:red;'>No valid paper found for the given input.</div>")
+        else:
+            return gr.update()
+
+    select_paper_button.click(
+        fn=check_paper_id_error,
+        inputs=[paper_id, paper_from],
+        outputs=[content],
+        queue=False
+    )
+
 
 def main():
     with gr.Blocks(css_paths="style.css") as demo:
@@ -448,6 +522,7 @@ def main():
                 data = {
                     'date': [datetime.today().strftime('%Y-%m-%d')],
                     'paper_page': ['1234.56789'],
+                    'arxiv_id': ['1234.56789'],  # adding arxiv_id column as user requested
                     'title': ['An Example Paper']
                 }
                 self.df_prettified = pd.DataFrame(data)