Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 import tempfile
 import shutil
-from typing import Optional, Tuple, Union
+from typing import Optional, Union
 from huggingface_hub import InferenceClient
 from pathlib import Path
 
@@ -176,7 +176,7 @@ def generate_video(
     duration: int = 8,
     size: str = "1280x720",
     api_key: Optional[str] = None
-) -> Tuple[Optional[str], str]:
+) -> Optional[str]:
     """Generate video using Sora-2 through Hugging Face Inference API with fal-ai provider."""
     cleanup_temp_files()
     try:
@@ -189,7 +189,7 @@ def generate_video(
         else:
             temp_client = client
             if not os.environ.get("HF_TOKEN") and not api_key:
-                return None, "❌ Please set HF_TOKEN environment variable."
+                return None
         
         video_bytes = temp_client.text_to_video(
             prompt,
@@ -204,19 +204,19 @@ def generate_video(
         finally:
             temp_file.close()
         
-        return video_path, "✅ Video generated successfully!"
+        return video_path
     except Exception as e:
-        return None, f"❌ Error generating video: {str(e)}"
+        return None
 
 def generate_video_from_image(
     image: Union[str, bytes],
     prompt: str,
     api_key: Optional[str] = None
-) -> Tuple[Optional[str], str]:
+) -> Optional[str]:
     """Generate a video from a single input image + prompt using Sora-2 image-to-video."""
     cleanup_temp_files()
     if not prompt or prompt.strip() == "":
-        return None, "❌ Please enter a prompt"
+        return None
     try:
         if api_key:
             temp_client = InferenceClient(
@@ -227,7 +227,7 @@ def generate_video_from_image(
         else:
             temp_client = client
             if not os.environ.get("HF_TOKEN") and not api_key:
-                return None, "❌ Please set HF_TOKEN environment variable."
+                return None
 
         if isinstance(image, str):
             with open(image, "rb") as f:
@@ -235,7 +235,7 @@ def generate_video_from_image(
         elif isinstance(image, (bytes, bytearray)):
             input_image = image
         else:
-            return None, "❌ Invalid image input. Please upload an image."
+            return None
 
         video_bytes = temp_client.image_to_video(
             input_image,
@@ -251,20 +251,20 @@ def generate_video_from_image(
         finally:
             temp_file.close()
 
-        return video_path, "✅ Video generated from image successfully!"
+        return video_path
     except Exception as e:
-        return None, f"❌ Error generating video from image: {str(e)}"
+        return None
 
 def generate_with_auth(
     prompt: str, 
     profile: gr.OAuthProfile | None
-) -> Tuple[Optional[str], str]:
+) -> Optional[str]:
     """Wrapper function that checks if user is logged in before generating video."""
     if profile is None:
         raise gr.Error("Click Sign in with Hugging Face button to use this app for free")
     
     if not prompt or prompt.strip() == "":
-        return None, "❌ Please enter a prompt"
+        return None
     
     return generate_video(
         prompt, 
@@ -277,12 +277,12 @@ def generate_with_auth_image(
     prompt: str,
     image_path: Optional[str],
     profile: gr.OAuthProfile | None
-) -> Tuple[Optional[str], str]:
+) -> Optional[str]:
     """Checks login status then calls image->video generator."""
     if profile is None:
         raise gr.Error("Click Sign in with Hugging Face button to use this app for free")
     if not image_path:
-        return None, "❌ Please upload an image"
+        return None
     return generate_video_from_image(image=image_path, prompt=prompt, api_key=None)
 
 def create_ui():
@@ -304,14 +304,14 @@ def create_ui():
     }
     #share-btn-container, #share-btn-container-img {
         display: flex; 
-        padding-left: 0.5rem !important; 
-        padding-right: 0.5rem !important; 
+        padding-left: 0.4rem !important; 
+        padding-right: 0.4rem !important; 
         background-color: #000000; 
         justify-content: center; 
         align-items: center; 
         border-radius: 9999px !important; 
-        width: 13rem;
-        margin-top: 10px;
+        width: 10rem;
+        margin-top: 8px;
         margin-left: auto;
         margin-right: auto;
     }
@@ -324,9 +324,10 @@ def create_ui():
         font-weight: 600; 
         cursor:pointer; 
         font-family: 'IBM Plex Sans', sans-serif; 
-        margin-left: 0.5rem !important; 
-        padding-top: 0.25rem !important; 
-        padding-bottom: 0.25rem !important;
+        margin-left: 0.4rem !important; 
+        padding-top: 0.2rem !important; 
+        padding-bottom: 0.2rem !important;
+        font-size: 0.85rem;
         right:0;
     }
     #share-btn *, #share-btn-img * {
@@ -376,7 +377,6 @@ def create_ui():
                     show_download_button=True,
                     elem_id="text-to-video"
                 )
-                status_output = gr.Textbox(label="Status", interactive=False, visible=True)
                 
                 # Share button for text-to-video
                 with gr.Group(elem_id="share-btn-container"):
@@ -387,7 +387,7 @@ def create_ui():
         generate_btn.click(
             fn=generate_with_auth,
             inputs=[prompt_input],
-            outputs=[video_output, status_output],
+            outputs=[video_output],
         )
         
         share_button.click(
@@ -422,7 +422,6 @@ def create_ui():
                     show_download_button=True,
                     elem_id="image-to-video"
                 )
-                status_output_img = gr.Textbox(label="Status", interactive=False, visible=True)
                 
                 # Share button for image-to-video
                 with gr.Group(elem_id="share-btn-container-img"):
@@ -433,7 +432,7 @@ def create_ui():
         generate_img_btn.click(
             fn=generate_with_auth_image,
             inputs=[img_prompt_input, image_input],
-            outputs=[video_output_img, status_output_img],
+            outputs=[video_output_img],
         )
         
         share_button_img.click(