Update app.py

app.py

@@ -163,7 +163,23 @@ def create_model():
         return model 
     except Exception as e: 
         print("❌ Failed to create any Gemini model:", e) 
-        raise 
+        raise
+
+def validate_uploaded_file(uploaded_file):
+    """
+    Validate that an uploaded file is accessible and hasn't expired.
+    """
+    try:
+        # Try to access the file info to check if it's valid
+        file_info = genai.get_file(uploaded_file.name)
+        if file_info.state.name == "ACTIVE":
+            return True
+        else:
+            print(f"⚠️ File {uploaded_file.name} is not active (state: {file_info.state.name})")
+            return False
+    except Exception as e:
+        print(f"❌ File validation failed: {e}")
+        return False 
 
 def merge_pdfs(paths, output_path): 
     writer = PdfWriter() 
@@ -542,11 +558,29 @@ def align_and_grade_pipeline(qp_path, ms_path, ans_path, imprint=False):
         merge_pdfs([qp_path, ms_path], merged_qpms_path)
         print("📎 Merged QP + MS ->", merged_qpms_path)
 
-        # Upload files to Gemini
+        # Upload files to Gemini with error handling and validation
         print("🔼 Uploading files to Gemini...")
-        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("✅ Upload complete.")
+        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")
+            
+            # 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")
+                
+            print("✅ Upload and validation complete.")
+        except Exception as e:
+            print(f"❌ File upload/validation failed: {e}")
+            print("🔄 Retrying file upload...")
+            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("✅ Retry upload successful.")
+            except Exception as retry_e:
+                print(f"❌ Retry failed: {retry_e}")
+                return f"❌ File upload error: {retry_e}", None, None, None, None
 
         # Create model and print which selected
         model = create_model()
@@ -638,8 +672,8 @@ def align_and_grade_pipeline(qp_path, ms_path, ans_path, imprint=False):
         return f"❌ Error: {e}", None, None, None, None
 
 # ---------------- GRADIO UI ---------------- 
-with gr.Blocks(title=" AI Grading (Final Flow )") as demo: 
-    gr.Markdown("## 📘  AI Grading — Final Flow") 
+with gr.Blocks(title="LeadIB AI Grading (Final Flow — Verbose)") as demo: 
+    gr.Markdown("## 📘 LeadIB AI Grading — Final Flow\nUpload **Question Paper**, **Markscheme**, and **Student Answer Sheet**.\nFlow: merge QP+MS -> transcribe QP+MS (questions first, full markscheme) -> extract IDs -> transcribe AS with expected IDs -> grade -> (optional) imprint. Console prints show progress.") 
 
     with gr.Row(): 
         qp_file = gr.File(label="📄 Upload Question Paper (PDF)")