Update app.py

app.py

@@ -303,7 +303,6 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                 generate_button = gr.Button("Generate", variant="primary")
             with gr.Column(scale=1):
                 output_image = gr.Image(label="Output", interactive=False, elem_id="output", type="filepath")
-                status_message = gr.Markdown("", visible=True)
                 use_image_button = gr.Button("♻️ Use this Image for Next Edit", variant="primary")
                 with gr.Row():
                     create_video_button = gr.Button("Create video between the two images 🎥", variant="secondary", visible=False)
@@ -311,7 +310,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                 with gr.Group(visible=False) as video_group:
                     video_output = gr.Video(label="Generated Video", show_download_button=True, autoplay=True)
                     gr.Markdown("Generate more with [Wan 2.2 first-last-frame](https://huggingface.co/spaces/multimodalart/wan-2-2-first-last-frame)", elem_id="wan_ad")
-                manual_token = gr.Textbox("Manual Token (to use with the API)", visible=False)
+        manual_token = gr.Textbox("Manual Token (to use with the API)", visible=False)
         gr.Markdown("<h2 style='text-align: center'>Thank you for being a PRO! 🤗</h2>")
 
     login_button = gr.LoginButton()
@@ -320,15 +319,15 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
         triggers=[generate_button.click, prompt_input.submit],
         fn=unified_image_generator,
         inputs=[prompt_input, image_input_gallery, previous_video_state, last_frame_of_video_state, manual_token],
-        outputs=[output_image, create_video_button, extend_video_button, video_group, status_message]
+        outputs=[output_image, create_video_button, extend_video_button, video_group]
     )
     use_image_button.click(
         fn=lambda img: (
             [img] if img else None, None, gr.update(visible=False),
-            gr.update(visible=False), gr.update(visible=False), ""
+            gr.update(visible=False), gr.update(visible=False)
         ),
         inputs=[output_image],
-        outputs=[image_input_gallery, output_image, create_video_button, extend_video_button, video_group, status_message]
+        outputs=[image_input_gallery, output_image, create_video_button, extend_video_button, video_group]
     )
     create_video_button.click(
         fn=lambda: gr.update(visible=True), outputs=[video_group]
@@ -346,12 +345,9 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     )
 
     def control_access(profile: Optional[gr.OAuthProfile] = None, oauth_token: Optional[gr.OAuthToken] = None):
-        if not profile: return gr.update(visible=False), gr.update(visible=False), ""
+        if not profile: return gr.update(visible=False), gr.update(visible=False)
         if verify_pro_status(oauth_token):
-            username = get_username(oauth_token)
-            remaining = get_remaining_generations(username) if username else 80
-            status = f"Welcome! You have {remaining} generations remaining today."
-            return gr.update(visible=True), gr.update(visible=False), status
+            return gr.update(visible=True), gr.update(visible=False)
         else:
             message = (
                 "## ✨ Exclusive Access for PRO Users\n\n"
@@ -359,8 +355,8 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
                 "To unlock this and many other cool stuff, please consider upgrading your account.\n\n"
                 "### [**Become a PRO Today!**](http://huggingface.co/subscribe/pro?source=nana_banana)"
             )
-            return gr.update(visible=False), gr.update(visible=True, value=message), ""
-    demo.load(control_access, inputs=None, outputs=[main_interface, pro_message, status_message])
+            return gr.update(visible=False), gr.update(visible=True, value=message)
+    demo.load(control_access, inputs=None, outputs=[main_interface, pro_message])
 
 if __name__ == "__main__":
     demo.queue(max_size=None, default_concurrency_limit=None).launch(show_error=True)