Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import fal_client
+from huggingface_hub import whoami
 import os
 from typing import Optional
 
@@ -109,12 +110,16 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                     outputs=[multi_image_output],
                 )
 
-    def show_app_on_login(profile: Optional[gr.OAuthProfile] = None):
+    def show_app_on_login(profile: Optional[gr.OAuthProfile] = None, oauth_token: gr.OAuthToken = None):
         """
         Controls the visibility of the main UI.
         Gradio automatically injects `gr.OAuthProfile` if the user is logged in.
         """
+        
         if profile:
+            print(profile)
+            user = whoami(token=oauth_token.token)
+            print(user)
             return gr.update(visible=True)
         return gr.update(visible=False)