Update app.py

app.py

@@ -165,6 +165,38 @@ def create_model():
         print("❌ Failed to create any Gemini model:", e) 
         raise
 
+def wait_for_file_active(uploaded_file, timeout=300):
+    """
+    Wait for an uploaded file to become ACTIVE before proceeding.
+    Polls the file status every 2 seconds until ACTIVE or timeout.
+    """
+    import time
+    start_time = time.time()
+    
+    while time.time() - start_time < timeout:
+        try:
+            file_info = genai.get_file(uploaded_file.name)
+            state = file_info.state.name
+            print(f"📊 File {uploaded_file.name} status: {state}")
+            
+            if state == "ACTIVE":
+                print(f"✅ File {uploaded_file.name} is now ACTIVE and ready to use")
+                return True
+            elif state == "FAILED":
+                print(f"❌ File {uploaded_file.name} processing failed")
+                return False
+            
+            # Still processing, wait a bit
+            print(f"⏳ File still processing, waiting 2 seconds...")
+            time.sleep(2)
+            
+        except Exception as e:
+            print(f"❌ Error checking file status: {e}")
+            time.sleep(2)
+    
+    print(f"⏰ Timeout waiting for file {uploaded_file.name} to become active")
+    return False
+
 def validate_uploaded_file(uploaded_file):
     """
     Validate that an uploaded file is accessible and hasn't expired.
@@ -562,21 +594,28 @@ def align_and_grade_pipeline(qp_path, ms_path, ans_path, imprint=False):
         print("🔼 Uploading files to Gemini...")
         try:
             merged_uploaded = genai.upload_file(path=merged_qpms_path, display_name="QP+MS (merged)")
-            ans_uploaded = genai.upload_file(path=ans_path, display_name="Answer Sheet")
+            print("📤 QP+MS file uploaded, waiting for processing...")
+            if not wait_for_file_active(merged_uploaded):
+                raise Exception("QP+MS file failed to become active")
             
-            # Validate uploads
-            if not validate_uploaded_file(merged_uploaded):
-                raise Exception("QP+MS file upload validation failed")
-            if not validate_uploaded_file(ans_uploaded):
-                raise Exception("Answer Sheet file upload validation failed")
+            ans_uploaded = genai.upload_file(path=ans_path, display_name="Answer Sheet")
+            print("📤 Answer Sheet uploaded, waiting for processing...")
+            if not wait_for_file_active(ans_uploaded):
+                raise Exception("Answer Sheet file failed to become active")
                 
-            print("✅ Upload and validation complete.")
+            print("✅ All files uploaded and active.")
         except Exception as e:
-            print(f"❌ File upload/validation failed: {e}")
+            print(f"❌ File upload/processing failed: {e}")
             print("🔄 Retrying file upload...")
             try:
                 merged_uploaded = genai.upload_file(path=merged_qpms_path, display_name="QP+MS (merged)")
+                if not wait_for_file_active(merged_uploaded):
+                    raise Exception("QP+MS retry failed to become active")
+                    
                 ans_uploaded = genai.upload_file(path=ans_path, display_name="Answer Sheet")
+                if not wait_for_file_active(ans_uploaded):
+                    raise Exception("Answer Sheet retry failed to become active")
+                    
                 print("✅ Retry upload successful.")
             except Exception as retry_e:
                 print(f"❌ Retry failed: {retry_e}")