Update app.py

app.py

@@ -633,38 +633,250 @@ def align_and_grade_pipeline(qp_path, ms_path, ans_path, imprint=False):
         print("❌ Pipeline error:", e)
         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.Row(): 
-        qp_file = gr.File(label="📄 Upload Question Paper (PDF)") 
-        ms_file = gr.File(label="📄 Upload Markscheme (PDF)") 
-        ans_file = gr.File(label="📝 Upload Student Answer Sheet (PDF)") 
-
-    imprint_toggle = gr.Checkbox(label="✍ Imprint Marks on Student Answer Sheet", value=False) 
-    run_button = gr.Button("🚀 Run Pipeline") 
-
-    with gr.Row(): 
-        grading_pdf_file = gr.File(label="📥 Download Grading PDF") 
-        imprint_pdf_file = gr.File(label="📥 Download Imprinted PDF (Optional)") 
-
-    def run_pipeline(qp_file_obj, ms_file_obj, ans_file_obj, imprint_flag): 
-        qp_path = qp_file_obj.name 
-        ms_path = ms_file_obj.name 
-        ans_path = ans_file_obj.name 
-
-        qpms_text, as_text, grading_text, grading_pdf_path, imprinted_pdf_path = align_and_grade_pipeline( 
-            qp_path, ms_path, ans_path, imprint=imprint_flag 
-        ) 
-
-        return grading_pdf_path, imprinted_pdf_path 
-
-    run_button.click( 
-        fn=run_pipeline, 
-        inputs=[qp_file, ms_file, ans_file, imprint_toggle], 
-        outputs=[grading_pdf_file, imprint_pdf_file] 
-    ) 
+# ---------------- MASS GRADING HELPERS ----------------
+def process_single_triplet(triplet_data, triplet_index):
+    """Process a single QP+MS+AS triplet and return results."""
+    try:
+        qp_path = triplet_data["qp_file"]
+        ms_path = triplet_data["ms_file"] 
+        ans_path = triplet_data["ans_file"]
+        imprint_flag = triplet_data["imprint"]
+        
+        print(f"🔄 Processing triplet {triplet_index + 1}: {os.path.basename(ans_path)}")
+        
+        qpms_text, as_text, grading_text, grading_pdf_path, imprinted_pdf_path = align_and_grade_pipeline(
+            qp_path, ms_path, ans_path, imprint=imprint_flag
+        )
+        
+        return {
+            "triplet_index": triplet_index,
+            "status": "✅ Complete",
+            "grading_pdf": grading_pdf_path,
+            "imprinted_pdf": imprinted_pdf_path if imprint_flag else None,
+            "error": None
+        }
+    except Exception as e:
+        print(f"❌ Error processing triplet {triplet_index + 1}: {e}")
+        return {
+            "triplet_index": triplet_index,
+            "status": f"❌ Error: {str(e)[:50]}...",
+            "grading_pdf": None,
+            "imprinted_pdf": None,
+            "error": str(e)
+        }
+
+def process_all_triplets(triplets_data):
+    """Process all triplets sequentially and return results."""
+    results = []
+    for i, triplet in enumerate(triplets_data):
+        if triplet["qp_file"] and triplet["ms_file"] and triplet["ans_file"]:
+            result = process_single_triplet(triplet, i)
+            results.append(result)
+            yield results  # Yield intermediate results for progress updates
+    return results
+
+# ---------------- GRADIO UI ----------------
+with gr.Blocks(title="🎓 AI Mass Grading System", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🎓 AI Mass Grading System")
+    gr.Markdown("Upload multiple sets of Question Papers, Markschemes, and Answer Sheets for batch processing.")
+    
+    # State to store triplets
+    triplets_state = gr.State([])
+    
+    # Dynamic triplet container
+    triplets_container = gr.Column()
+    
+    # Add triplet button
+    with gr.Row():
+        add_triplet_btn = gr.Button("➕ Add New Triplet", variant="secondary")
+        start_grading_btn = gr.Button("🚀 Start Mass Grading", variant="primary", visible=False)
+    
+    # Status table
+    status_table = gr.Dataframe(
+        headers=["Set", "QP File", "MS File", "Answer Sheets", "Status"],
+        datatype=["number", "str", "str", "str", "str"],
+        label="📊 Grading Status",
+        visible=False
+    )
+    
+    # Results section
+    results_section = gr.Column(visible=False)
+    with results_section:
+        gr.Markdown("## 📥 Download Results")
+        results_files = gr.Column()
+    
+    def create_triplet_row(triplet_index):
+        """Create a new triplet upload row."""
+        with gr.Row() as row:
+            with gr.Column(scale=2):
+                qp_file = gr.File(label=f"📄 QP {triplet_index + 1}", file_types=[".pdf"])
+            with gr.Column(scale=2):
+                ms_file = gr.File(label=f"📋 MS {triplet_index + 1}", file_types=[".pdf"])
+            with gr.Column(scale=2):
+                ans_file = gr.File(label=f"📝 AS {triplet_index + 1}", file_types=[".pdf"])
+            with gr.Column(scale=1):
+                imprint_check = gr.Checkbox(label="✅ Imprint", value=False)
+            with gr.Column(scale=1):
+                remove_btn = gr.Button("🗑️", variant="stop", size="sm")
+        
+        return row, qp_file, ms_file, ans_file, imprint_check, remove_btn
+    
+    def add_triplet(current_triplets):
+        """Add a new triplet to the list."""
+        new_triplet = {
+            "qp_file": None,
+            "ms_file": None, 
+            "ans_file": None,
+            "imprint": False
+        }
+        current_triplets.append(new_triplet)
+        return current_triplets, gr.update(visible=len(current_triplets) > 0)
+    
+    def update_status_table(triplets):
+        """Update the status table with current triplets."""
+        if not triplets:
+            return gr.update(visible=False, value=[])
+        
+        table_data = []
+        for i, triplet in enumerate(triplets):
+            qp_name = os.path.basename(triplet["qp_file"]) if triplet["qp_file"] else "Not uploaded"
+            ms_name = os.path.basename(triplet["ms_file"]) if triplet["ms_file"] else "Not uploaded"
+            ans_name = os.path.basename(triplet["ans_file"]) if triplet["ans_file"] else "Not uploaded"
+            
+            # Determine status
+            if triplet["qp_file"] and triplet["ms_file"] and triplet["ans_file"]:
+                status = "📋 Ready"
+            else:
+                status = "⏳ Pending uploads"
+            
+            table_data.append([i + 1, qp_name, ms_name, ans_name, status])
+        
+        return gr.update(visible=True, value=table_data)
+    
+    def start_mass_grading(triplets):
+        """Start processing all triplets."""
+        if not triplets:
+            return gr.update(), gr.update(visible=False)
+        
+        # Filter only complete triplets
+        complete_triplets = [t for t in triplets if t["qp_file"] and t["ms_file"] and t["ans_file"]]
+        
+        if not complete_triplets:
+            return gr.update(value=[["No complete triplets to process"]]), gr.update(visible=False)
+        
+        # Process triplets and update table progressively
+        results = []
+        table_data = []
+        
+        for i, triplet in enumerate(complete_triplets):
+            # Update status to processing
+            table_row = [
+                i + 1,
+                os.path.basename(triplet["qp_file"]),
+                os.path.basename(triplet["ms_file"]),
+                os.path.basename(triplet["ans_file"]),
+                "🔄 Processing..."
+            ]
+            table_data.append(table_row)
+            
+            # Process the triplet
+            result = process_single_triplet(triplet, i)
+            results.append(result)
+            
+            # Update status
+            table_data[i][4] = result["status"]
+        
+        # Create download links
+        download_components = []
+        for i, result in enumerate(results):
+            if result["grading_pdf"]:
+                download_components.append(
+                    gr.File(value=result["grading_pdf"], label=f"📄 Set {i+1} - Grading PDF", visible=True)
+                )
+            if result["imprinted_pdf"]:
+                download_components.append(
+                    gr.File(value=result["imprinted_pdf"], label=f"✍️ Set {i+1} - Imprinted PDF", visible=True)
+                )
+        
+        return gr.update(value=table_data), gr.update(visible=True)
+    
+    # Event handlers
+    add_triplet_btn.click(
+        fn=add_triplet,
+        inputs=[triplets_state],
+        outputs=[triplets_state, start_grading_btn]
+    ).then(
+        fn=update_status_table,
+        inputs=[triplets_state],
+        outputs=[status_table]
+    )
+    
+    # File upload handlers to update triplets state
+    def create_file_update_handler(triplet_idx, file_type):
+        def update_triplet_file(file_obj, current_triplets):
+            if triplet_idx < len(current_triplets):
+                current_triplets[triplet_idx][file_type] = file_obj.name if file_obj else None
+            return current_triplets
+        return update_triplet_file
+    
+    def create_imprint_update_handler(triplet_idx):
+        def update_triplet_imprint(imprint_val, current_triplets):
+            if triplet_idx < len(current_triplets):
+                current_triplets[triplet_idx]["imprint"] = imprint_val
+            return current_triplets
+        return update_triplet_imprint
+    
+    # Dynamic UI management
+    @gr.render(inputs=[triplets_state])
+    def render_triplets(triplets):
+        if not triplets:
+            return
+        
+        for i in range(len(triplets)):
+            row, qp_file, ms_file, ans_file, imprint_check, remove_btn = create_triplet_row(i)
+            
+            # Set up file upload handlers
+            qp_file.change(
+                fn=create_file_update_handler(i, "qp_file"),
+                inputs=[qp_file, triplets_state],
+                outputs=[triplets_state]
+            ).then(
+                fn=update_status_table,
+                inputs=[triplets_state],
+                outputs=[status_table]
+            )
+            
+            ms_file.change(
+                fn=create_file_update_handler(i, "ms_file"),
+                inputs=[ms_file, triplets_state],
+                outputs=[triplets_state]
+            ).then(
+                fn=update_status_table,
+                inputs=[triplets_state],
+                outputs=[status_table]
+            )
+            
+            ans_file.change(
+                fn=create_file_update_handler(i, "ans_file"),
+                inputs=[ans_file, triplets_state],
+                outputs=[triplets_state]
+            ).then(
+                fn=update_status_table,
+                inputs=[triplets_state],
+                outputs=[status_table]
+            )
+            
+            imprint_check.change(
+                fn=create_imprint_update_handler(i),
+                inputs=[imprint_check, triplets_state],
+                outputs=[triplets_state]
+            )
+    
+    start_grading_btn.click(
+        fn=start_mass_grading,
+        inputs=[triplets_state],
+        outputs=[status_table, results_section]
+    )
 
-if __name__ == "__main__": 
+if __name__ == "__main__":
     demo.launch()