Update app.py

app.py

@@ -106,14 +106,14 @@ def _generate_video_segment(input_image_path: str, output_image_path: str, promp
     )
     return result[0]["video"]
 
-def unified_image_generator(prompt: str, images: Optional[List[str]], previous_video_path: Optional[str], last_frame_path: Optional[str], oauth_token: Optional[gr.OAuthToken]) -> tuple:
-    if not verify_pro_status(oauth_token): raise gr.Error("Access Denied.")
+def unified_image_generator(prompt: str, images: Optional[List[str]], previous_video_path: Optional[str], last_frame_path: Optional[str], manual_token: str, oauth_token: Optional[gr.OAuthToken]) -> tuple:
+    if not (verify_pro_status(oauth_token) or verify_pro_status(manual_token)): raise gr.Error("Access Denied.")
     try:
         contents = [Image.open(image_path[0]) for image_path in images] if images else []
         contents.append(prompt)
         response = client.models.generate_content(model=GEMINI_MODEL_NAME, contents=contents)
         image_data = _extract_image_data_from_response(response)
-        if not image_data: raise ValueError("No image data in response.")
+        if not image_data: raise gr.Error("No image data in response.")
         with tempfile.NamedTemporaryFile(delete=False, suffix=".png") as tmp:
             Image.open(BytesIO(image_data)).save(tmp.name)
             output_path = tmp.name
@@ -195,6 +195,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)
         gr.Markdown("<h2 style='text-align: center'>Thank you for being a PRO! 🤗</h2>")
 
     login_button = gr.LoginButton()
@@ -202,7 +203,7 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=css) as demo:
     gr.on(
         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],
+        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]
     )
     use_image_button.click(