logout

app.py

@@ -10,7 +10,7 @@ import spaces
 import requests
 from dotenv import load_dotenv
 import gradio as gr
-from gradio.components import LoginButton
+from gradio.components import LoginButton, LogoutButton
 
 import data_manager
 
@@ -707,6 +707,7 @@ 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")
@@ -735,6 +736,7 @@ 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),
             )
@@ -752,6 +754,7 @@ 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),
             )
@@ -764,6 +767,7 @@ with gr.Blocks(css=APP_CSS, title="LLM GeoGuessr") as demo:
             gr.update(visible=False),
             gr.update(visible=True, value=welcome_message),
             gr.update(visible=True),
+            gr.update(visible=True),
         ]
 
         if has_played:
@@ -789,6 +793,7 @@ 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,
@@ -803,6 +808,22 @@ 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,