Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import os
 import json
+import time # We need this to wait
 from dotenv import load_dotenv
 
 # Your requested imports
@@ -23,12 +24,28 @@ def analyze_device_condition(video_file_path):
 
     uploaded_file = None
     try:
-        # Upload the file
+        # 1. Upload the file
         print(f"Log: Uploading file: {video_file_path}...")
         uploaded_file = client.files.upload(file=video_file_path)
-        print("Log: File uploaded successfully.")
+        print(f"Log: File upload initiated. File name: {uploaded_file.name}, State: {uploaded_file.state.name}")
         
-        # Prepare the prompt
+        # 2. --- THIS IS THE CRITICAL POLLING LOOP ---
+        #    Wait for the file to become ACTIVE.
+        while uploaded_file.state.name == "PROCESSING":
+            print("Log: File is processing, waiting 5 seconds...")
+            time.sleep(5)
+            # Get the latest status of the file.
+            uploaded_file = client.files.get(name=uploaded_file.name)
+            print(f"Log: Current file state: {uploaded_file.state.name}")
+
+        # If the file failed processing, stop here.
+        if uploaded_file.state.name != "ACTIVE":
+            raise Exception(f"File processing failed. Final state: {uploaded_file.state.name}")
+        # --- END OF CORRECTION ---
+        
+        print("Log: File is now ACTIVE and ready for use.")
+        
+        # 3. Prepare the prompt
         prompt = """
         Analyze the provided video. Respond ONLY with a valid JSON object with three keys:
         1. "device_type": A short string identifying the device.
@@ -36,27 +53,24 @@ def analyze_device_condition(video_file_path):
         3. "reason": A brief string explaining the condition.
         """
         
-        # Use gemini-2.5-flash
+        # 4. Use gemini-2.5-flash
         model_name = "gemini-2.5-flash"
         generate_content_config = types.GenerateContentConfig(
             temperature=0.2,
             response_mime_type="application/json"
         )
         
-        # Call the API with the simplified contents list
+        # 5. Call the API with the now-active file
         contents = [uploaded_file, prompt]
         
-        # --- THIS IS THE CORRECTED LINE ---
-        # The keyword argument is 'config', not 'generation_config'.
         print(f"Log: Sending request to model: {model_name}...")
         response = client.models.generate_content(
             model=f"models/{model_name}",
             contents=contents,
             config=generate_content_config,
         )
-        # --- END OF CORRECTION ---
 
-        # Parse the final JSON response
+        # 6. Parse the final JSON response
         parsed_json = json.loads(response.text)
         device_type = parsed_json.get("device_type", "N/A")
         condition = parsed_json.get("condition", "N/A")
@@ -70,7 +84,7 @@ def analyze_device_condition(video_file_path):
         return error_message, "", ""
     
     finally:
-        # Cleanup: Delete the file from Google's servers
+        # 7. Cleanup: Delete the file from Google's servers
         if uploaded_file:
             print(f"Log: Deleting uploaded file: {uploaded_file.name}")
             client.files.delete(name=uploaded_file.name)