update

app.py

@@ -24,8 +24,10 @@ custom_css = """
 div#component-4 #chatbot {
     height: 800px !important;
 }
+
 """
 
+
 def replace_texttt(text):
     return re.sub(r"\\texttt\{(.*?)\}", r"*\1*", text)
 
@@ -41,8 +43,8 @@ def get_paper_info(paper_id):
         # Return the paper's title and abstract
         #  remove new lines
         title_ = paper.title.replace("\n", " ").replace("\r", " ")
-        summary_  = paper.summary.replace("\n", " ").replace("\r", " ")
-        return  title_, summary_
+        summary_ = paper.summary.replace("\n", " ").replace("\r", " ")
+        return title_, summary_
     else:
         return None, None
 
@@ -51,7 +53,7 @@ def get_paper_from_huggingface(paper_id):
     try:
         url = f"https://huggingface.co/datasets/taesiri/arxiv_db/raw/main/papers/{paper_id}.tex"
         response = requests.get(url)
-        response.raise_for_status() 
+        response.raise_for_status()
         return response.text
     except Exception as e:
         return None
@@ -214,7 +216,9 @@ def clear_context():
     return []
 
 
-with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
+with gr.Blocks(
+    theme=gr.themes.Soft(), css=custom_css, title="ArXiv QA with Claude"
+) as demo:
     gr.HTML(
         """
     <h1 style='text-align: center; font-size: 24px;'>
@@ -246,7 +250,10 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
 
     with gr.Row().style(equal_height=False):
         with gr.Column(scale=2, emem_id="column-flex"):
-            chatbot = gr.Chatbot(elem_id="chatbot")
+            chatbot = gr.Chatbot(
+                elem_id="chatbot",
+                avatar_images=("./assets/user.png", "./assets/Claude.png"),
+            )
 
         with gr.Column(scale=1):
             paper_id_input = gr.Textbox(label="Enter Paper ID", value="2310.12103")
@@ -260,8 +267,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
             btn_answer = gr.Button("Answer Question")
             btn_clear = gr.Button("Clear Chat")
 
-
-
     gr.HTML(
         """<center>All the inputs are being sent to Anthropic's Claude endpoints. Please refer to <a href="https://legal.anthropic.com/#privacy">this link</a> for privacy policy.</center>"""
     )
@@ -288,5 +293,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css=custom_css) as demo:
 
     btn_clear.click(clear_context, outputs=[chatbot])
 
+
 app.mount("/js", StaticFiles(directory="js"), name="js")
 gr.mount_gradio_app(app, demo, path="/")