Update app.py

app.py

@@ -13,7 +13,7 @@ from huggingface_hub import login
 # --- 1. Configuration and Authentication ---
 
 # IMPORTANT: Replace with your Google Cloud Project ID
-GCP_PROJECT_ID = "gen-lang-client-0193353123"
+GCP_PROJECT_ID = "YOUR_GCP_PROJECT_ID" # Make sure this is replaced!
 GCP_LOCATION = "us-central1"
 MODEL_ID = "veo-3.0-generate-preview"
 API_ENDPOINT = f"{GCP_LOCATION}-aiplatform.googleapis.com"
@@ -23,8 +23,7 @@ FETCH_URL = f"https://{API_ENDPOINT}/v1/projects/{GCP_PROJECT_ID}/locations/{GCP
 
 # --- Authentication Block ---
 
-# Part A: Hugging Face Hub Authentication (NEW)
-# This section looks for a secret named 'HF_TOKEN' to log into the Hub.
+# Part A: Hugging Face Hub Authentication
 hf_token = os.environ.get("HF_TOKEN")
 if hf_token:
     print("Hugging Face token found. Logging in.")
@@ -32,13 +31,10 @@ if hf_token:
 else:
     print("WARNING: Hugging Face token ('HF_TOKEN') not found. Hub-related features may be disabled.")
 
-
-# Part B: Google Cloud Authentication (Unchanged)
-# This section expects a secret named 'GOOGLE_APPLICATION_CREDENTIALS_JSON'
+# Part B: Google Cloud Authentication
 creds_json_str = os.environ.get("GOOGLE_APPLICATION_CREDENTIALS_JSON")
 if not creds_json_str:
     print("FATAL: 'GOOGLE_APPLICATION_CREDENTIALS_JSON' secret not found. App cannot authenticate with Google Cloud.")
-    # Define a dummy function to show an error in the UI
     def generate_video(prompt):
         raise gr.Error("Authentication failed. Server is missing Google Cloud credentials. Please check the Hugging Face Space secrets.")
 else:
@@ -51,17 +47,13 @@ else:
 
 
     def get_access_token():
-        """Generates a fresh short-lived access token for Google Cloud."""
         auth_req = google.auth.transport.requests.Request()
         credentials.refresh(auth_req)
         return credentials.token
 
-    # --- 2. Core Video Generation Logic (Unchanged) ---
+    # --- 2. Core Video Generation Logic (WITH FIX) ---
 
     def generate_video(prompt: str):
-        """
-        The main function to generate a video. It submits, polls, and returns the result.
-        """
         if not prompt:
             raise gr.Error("Prompt cannot be empty.")
             
@@ -111,15 +103,32 @@ else:
 
                 if poll_result.get("done"):
                     print("Job finished successfully.")
-                    video_base64 = poll_result["response"]["predictions"][0]["bytesBase64Encoded"]
-                    video_bytes = base64.b64decode(video_base64)
+                    # --- START: ROBUST RESPONSE HANDLING FIX ---
                     
-                    temp_video_path = "generated_video.mp4"
-                    with open(temp_video_path, "wb") as f:
-                        f.write(video_bytes)
+                    # For debugging, print the entire final response from Google
+                    print(f"Full successful response payload: {json.dumps(poll_result, indent=2)}")
+
+                    response_data = poll_result.get("response", {})
                     
-                    yield "Status: Done!", temp_video_path
-                    return
+                    # Check if the 'predictions' key exists and is not empty
+                    if "predictions" in response_data and response_data["predictions"]:
+                        video_base64 = response_data["predictions"][0]["bytesBase64Encoded"]
+                        video_bytes = base64.b64decode(video_base64)
+                        
+                        temp_video_path = "generated_video.mp4"
+                        with open(temp_video_path, "wb") as f:
+                            f.write(video_bytes)
+                        
+                        yield "Status: Done! Video generated successfully.", temp_video_path
+                        return
+                    else:
+                        # This branch handles cases where the job is "done" but no video was returned,
+                        # most likely due to safety filters.
+                        error_message = "Video generation finished, but the content was blocked by safety filters or another issue prevented video creation."
+                        print(f"ERROR: {error_message}")
+                        raise gr.Error(error_message)
+
+                    # --- END: ROBUST RESPONSE HANDLING FIX ---
                 
                 time.sleep(10)
 
@@ -129,8 +138,9 @@ else:
             print(f"HTTP Error: {e.response.text}")
             raise gr.Error(f"API Error: {e.response.status_code}. Details: {e.response.text}")
         except Exception as e:
-            print(f"An unexpected error occurred: {e}")
-            raise gr.Error(f"An unexpected error occurred: {str(e)}")
+            # Catch the specific Gradio error we raised, and any other unexpected errors.
+            print(f"An error occurred in the generation process: {e}")
+            raise gr.Error(str(e))
 
 
 # --- 3. Gradio User Interface (Unchanged) ---