logout + better image

app.py

@@ -11,7 +11,6 @@ import requests
 from dotenv import load_dotenv
 import gradio as gr
 from gradio.components import LoginButton
-from gradio import LogoutButton
 
 import data_manager
 
@@ -716,7 +715,6 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
     login_button = LoginButton(visible=True)
     
     logged_in_md = gr.Markdown(visible=False)
-    logout_button = LogoutButton(visible=False)
 
     with gr.Group(visible=False, elem_id="lobby_group") as lobby_group:
         start_btn = gr.Button("Start Game", variant="primary", elem_id="start_btn")
@@ -745,7 +743,6 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
                 gr.update(visible=True),
                 gr.update(visible=False),
                 gr.update(visible=False),
-                gr.update(visible=False),
                 gr.update(),
                 gr.update(visible=False),
             )
@@ -763,7 +760,6 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
                 gr.update(visible=True),
                 gr.update(visible=False),
                 gr.update(visible=False),
-                gr.update(visible=False),
                 gr.update(),
                 gr.update(visible=False),
             )
@@ -773,9 +769,8 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
         updates = [
             profile,
             gr.update(visible=False),
-            gr.update(visible=False),
-            gr.update(visible=True, value=welcome_message),
             gr.update(visible=True),
+            gr.update(visible=True, value=welcome_message),
             gr.update(visible=True),
         ]
 
@@ -802,7 +797,6 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
             login_prompt_md,
             login_button,
             logged_in_md,
-            logout_button,
             lobby_group,
             start_btn,
             limit_msg,
@@ -817,22 +811,6 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
             login_prompt_md,
             login_button,
             logged_in_md,
-            logout_button,
-            lobby_group,
-            start_btn,
-            limit_msg,
-        ],
-    )
-    
-    # Handle logout
-    logout_button.click(
-        on_login,
-        outputs=[
-            user_profile,
-            login_prompt_md,
-            login_button,
-            logged_in_md,
-            logout_button,
             lobby_group,
             start_btn,
             limit_msg,