cosmetic

app.py

@@ -186,8 +186,7 @@ with gr.Blocks(css_paths="style.css") as demo:
             with gr.Tab("Contributors"):
                 author_resource_leaderboard_tab()
 
-
-        with gr.Tab("Chat With Paper", id="tab-chat-with-paper"):
+        with gr.Tab("Chat With Paper", id="tab-chat-with-paper", visible=False) as tab_chat_paper:
             gr.Markdown("## Chat with Paper")
             arxiv_id = gr.State(value=None)
             paper_chat_tab(arxiv_id)
@@ -481,6 +480,7 @@ with gr.Blocks(css_paths="style.css") as demo:
         hf_options = gr.update(value=[])
         selected_tab = gr.Tabs()
         paper_id = gr.update(value=None)
+        tab_chat_paper = gr.update(visible=False)
 
         if request:
             # print("Request headers dictionary:", dict(request.headers))
@@ -519,8 +519,9 @@ with gr.Blocks(css_paths="style.css") as demo:
                     selected_tab = gr.Tabs(selected="tab-chat-with-paper")
                     if "paper_id" in request.query_params:
                         paper_id = request.query_params['paper_id']
+                        tab_chat_paper = gr.update(visible=True)
 
-        return calendar, date_range, conferences, hf_options, selected_tab, paper_id
+        return calendar, date_range, conferences, hf_options, selected_tab, paper_id, tab_chat_paper
 
 
     demo.load(
@@ -530,7 +531,7 @@ with gr.Blocks(css_paths="style.css") as demo:
         api_name="update_data",
     ).then(
         fn=echo,
-        outputs=[calendar, date_range_radio, conference_options, hf_options, tabs, arxiv_id],
+        outputs=[calendar, date_range_radio, conference_options, hf_options, tabs, arxiv_id, tab_chat_paper],
         api_name=False,
     ).then(
         # New then to handle LoginButton and HTML components

df/PaperCentral.py

@@ -483,7 +483,7 @@ class PaperCentral:
                     neurips_id = re.search(r'id=([^&]+)', row["proceedings"])
                     if neurips_id:
                         neurips_id = neurips_id.group(1)
-                        return f'<a href="/?tab=tab-chat-with-paper&paper_id={neurips_id}" id="custom_button" target="_blank" rel="noopener noreferrer" aria-disabled="false">✨ Chat with paper</a>'
+                        return f'<a href="/?tab=tab-chat-with-paper&paper_id={neurips_id}" id="custom_button" target="_blank">✨ Chat with paper</a>'
                     else:
                         return ""
 

paper_chat_tab.py

@@ -167,7 +167,7 @@ def create_chat_interface(model_name, paper_content, hf_token_input):
         # Load the tokenizer from Hugging Face
         # tokenizer = AutoTokenizer.from_pretrained(model_name)
         tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-1B-Instruct",
-                                                  token=os.getenv("HF_TOKEN"))
+                                                  token=os.environ.get("HF_TOKEN"))
         tokenizer_cache[model_name] = tokenizer
     else:
         tokenizer = tokenizer_cache[model_name]
@@ -241,7 +241,7 @@ def create_chat_interface(model_name, paper_content, hf_token_input):
         final_messages.extend(messages)
 
         # Use the Hugging Face token if provided
-        api_key = hf_token_value or os.getenv("SAMBANOVA_API_KEY")
+        api_key = hf_token_value or os.environ.get("SAMBANOVA_API_KEY")
         if not api_key:
             raise ValueError("API token is not provided.")