Update app.py

app.py

@@ -1,4 +1,4 @@
-# app.py (Final Polished Version)
+# app.py
 
 import os
 import gradio as gr
@@ -11,10 +11,12 @@ import google.auth.transport.requests
 from huggingface_hub import login
 
 # --- 1. Configuration and Authentication ---
+
 GCP_PROJECT_ID = os.environ.get("GCP_PROJECT_ID")
 GCP_LOCATION = os.environ.get("GCP_LOCATION")
 
 # --- Authentication and Sanity Checks Block ---
+
 hf_token = os.environ.get("HF_TOKEN")
 if hf_token:
     print("Hugging Face token found. Logging in.")
@@ -25,7 +27,11 @@ else:
 creds_json_str = os.environ.get("GOOGLE_APPLICATION_CREDENTIALS_JSON")
 
 if not all([GCP_PROJECT_ID, GCP_LOCATION, creds_json_str]):
-    missing_secrets = [s for s, v in {"GCP_PROJECT_ID": GCP_PROJECT_ID, "GCP_LOCATION": GCP_LOCATION, "GOOGLE_APPLICATION_CREDENTIALS_JSON": creds_json_str}.items() if not v]
+    missing_secrets = [s for s, v in {
+        "GCP_PROJECT_ID": GCP_PROJECT_ID,
+        "GCP_LOCATION": GCP_LOCATION,
+        "GOOGLE_APPLICATION_CREDENTIALS_JSON": creds_json_str
+    }.items() if not v]
     error_message = f"FATAL: Missing required secrets: {', '.join(missing_secrets)}."
     print(error_message)
     def generate_video(prompt):
@@ -68,6 +74,7 @@ else:
                 poll_result = poll_response.json()
                 if poll_result.get("done"):
                     print("Job finished.")
+                    print(f"Full response payload: {json.dumps(poll_result, indent=2)}") # For debugging
                     response_data = poll_result.get("response", {})
                     if "videos" in response_data and response_data["videos"]:
                         video_base64 = response_data["videos"][0]["bytesBase64Encoded"]
@@ -76,23 +83,21 @@ else:
                         yield "Status: Done!", "generated_video.mp4"
                         return
                     else:
+                        # <<< START: IMPROVED ERROR HANDLING >>>
                         error_message = "Video generation failed."
+                        # Check for a specific error message in the operation response
                         if "error" in poll_result:
                             error_details = poll_result["error"].get("message", "No details provided.")
-                            # <<< START: FINAL UX IMPROVEMENT >>>
-                            if "Internal error" in error_details:
-                                error_message += "\nReason: The Google API reported a temporary internal error. Please wait and try again."
-                            elif "Deadline exceeded" in error_details:
-                                error_message += "\nReason: The job took too long on Google's servers and timed out. This can happen during peak load. Please try again."
-                            else:
-                                error_message += f"\nAPI Error: {error_details}"
-                            # <<< END: FINAL UX IMPROVEMENT >>>
+                            error_message += f"\nAPI Error: {error_details}"
+                        # Check for a specific RAI reason
                         elif "raiResult" in response_data:
                             rai_reason = response_data.get("raiMediaFilteredReason", "Unknown reason.")
                             error_message += f"\nReason: Content was blocked by safety filters ({rai_reason})."
                         else:
                             error_message += "\nReason: The API did not return a video or a specific error."
+                        
                         raise gr.Error(error_message)
+                        # <<< END: IMPROVED ERROR HANDLING >>>
                 time.sleep(10)
             raise gr.Error("Operation timed out.")
         except Exception as e: