Update app.py

app.py

@@ -680,16 +680,30 @@ with gr.Blocks(title="🎓 AI Mass Grading System", theme=gr.themes.Soft()) as d
     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([])
+    # Initialize with one triplet
+    initial_triplet = {
+        "qp_file": None,
+        "ms_file": None, 
+        "ans_file": None,
+        "imprint": False
+    }
+    triplets_state = gr.State([initial_triplet])
     
-    # Dynamic triplet container
-    triplets_container = gr.Column()
+    # Initial triplet row (always visible)
+    with gr.Row():
+        qp_file_1 = gr.File(label="📄 Upload Question Paper (PDF)", file_types=[".pdf"])
+        ms_file_1 = gr.File(label="📋 Upload Markscheme (PDF)", file_types=[".pdf"])
+        ans_file_1 = gr.File(label="📝 Upload Student Answer Sheet (PDF)", file_types=[".pdf"])
+    
+    imprint_check_1 = gr.Checkbox(label="✍ Imprint Marks on Student Answer Sheet", value=False)
+    
+    # Dynamic container for additional triplets
+    additional_triplets = gr.Column()
     
-    # Add triplet button
+    # Buttons row
     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)
+        add_triplet_btn = gr.Button("➕ Add Another Set", variant="secondary")
+        start_grading_btn = gr.Button("🚀 Run Pipeline", variant="primary")
     
     # Status table
     status_table = gr.Dataframe(
@@ -705,21 +719,25 @@ with gr.Blocks(title="🎓 AI Mass Grading System", theme=gr.themes.Soft()) as d
         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 update_first_triplet_qp(file_obj, current_triplets):
+        """Update QP file for first triplet."""
+        current_triplets[0]["qp_file"] = file_obj.name if file_obj else None
+        return current_triplets
+    
+    def update_first_triplet_ms(file_obj, current_triplets):
+        """Update MS file for first triplet."""
+        current_triplets[0]["ms_file"] = file_obj.name if file_obj else None
+        return current_triplets
+    
+    def update_first_triplet_ans(file_obj, current_triplets):
+        """Update AS file for first triplet."""
+        current_triplets[0]["ans_file"] = file_obj.name if file_obj else None
+        return current_triplets
+    
+    def update_first_triplet_imprint(imprint_val, current_triplets):
+        """Update imprint setting for first triplet."""
+        current_triplets[0]["imprint"] = imprint_val
+        return current_triplets
     
     def add_triplet(current_triplets):
         """Add a new triplet to the list."""
@@ -730,11 +748,17 @@ with gr.Blocks(title="🎓 AI Mass Grading System", theme=gr.themes.Soft()) as d
             "imprint": False
         }
         current_triplets.append(new_triplet)
-        return current_triplets, gr.update(visible=len(current_triplets) > 0)
+        return current_triplets
+    
+    def remove_triplet(triplet_idx, current_triplets):
+        """Remove a triplet from the list."""
+        if triplet_idx < len(current_triplets) and len(current_triplets) > 1:
+            current_triplets.pop(triplet_idx)
+        return current_triplets
     
     def update_status_table(triplets):
         """Update the status table with current triplets."""
-        if not triplets:
+        if len(triplets) <= 1:
             return gr.update(visible=False, value=[])
         
         table_data = []
@@ -786,58 +810,84 @@ with gr.Blocks(title="🎓 AI Mass Grading System", theme=gr.themes.Soft()) as d
             # 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,
+    # Event handlers for first triplet
+    qp_file_1.change(
+        fn=update_first_triplet_qp,
+        inputs=[qp_file_1, triplets_state],
+        outputs=[triplets_state]
+    ).then(
+        fn=update_status_table,
         inputs=[triplets_state],
-        outputs=[triplets_state, start_grading_btn]
+        outputs=[status_table]
+    )
+    
+    ms_file_1.change(
+        fn=update_first_triplet_ms,
+        inputs=[ms_file_1, triplets_state],
+        outputs=[triplets_state]
     ).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
+    ans_file_1.change(
+        fn=update_first_triplet_ans,
+        inputs=[ans_file_1, triplets_state],
+        outputs=[triplets_state]
+    ).then(
+        fn=update_status_table,
+        inputs=[triplets_state],
+        outputs=[status_table]
+    )
     
-    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
+    imprint_check_1.change(
+        fn=update_first_triplet_imprint,
+        inputs=[imprint_check_1, triplets_state],
+        outputs=[triplets_state]
+    )
     
-    # Dynamic UI management
+    # Dynamic UI management for additional triplets
     @gr.render(inputs=[triplets_state])
-    def render_triplets(triplets):
-        if not triplets:
+    def render_additional_triplets(triplets):
+        if len(triplets) <= 1:
             return
         
-        for i in range(len(triplets)):
-            row, qp_file, ms_file, ans_file, imprint_check, remove_btn = create_triplet_row(i)
+        for i in range(1, len(triplets)):  # Start from index 1 (skip first triplet)
+            with gr.Row():
+                qp_file = gr.File(label=f"📄 QP Set {i + 1}", file_types=[".pdf"])
+                ms_file = gr.File(label=f"📋 MS Set {i + 1}", file_types=[".pdf"])
+                ans_file = gr.File(label=f"📝 AS Set {i + 1}", file_types=[".pdf"])
+            
+            with gr.Row():
+                imprint_check = gr.Checkbox(label=f"✍ Imprint Set {i + 1}", value=False)
+                remove_btn = gr.Button("🗑️ Remove", variant="stop", size="sm")
+            
+            # File update handlers
+            def create_file_handler(idx, file_type):
+                def handler(file_obj, current_triplets):
+                    if idx < len(current_triplets):
+                        current_triplets[idx][file_type] = file_obj.name if file_obj else None
+                    return current_triplets
+                return handler
             
-            # Set up file upload handlers
+            def create_imprint_handler(idx):
+                def handler(imprint_val, current_triplets):
+                    if idx < len(current_triplets):
+                        current_triplets[idx]["imprint"] = imprint_val
+                    return current_triplets
+                return handler
+            
+            def create_remove_handler(idx):
+                def handler(current_triplets):
+                    return remove_triplet(idx, current_triplets)
+                return handler
+            
+            # Set up event handlers
             qp_file.change(
-                fn=create_file_update_handler(i, "qp_file"),
+                fn=create_file_handler(i, "qp_file"),
                 inputs=[qp_file, triplets_state],
                 outputs=[triplets_state]
             ).then(
@@ -847,7 +897,7 @@ with gr.Blocks(title="🎓 AI Mass Grading System", theme=gr.themes.Soft()) as d
             )
             
             ms_file.change(
-                fn=create_file_update_handler(i, "ms_file"),
+                fn=create_file_handler(i, "ms_file"),
                 inputs=[ms_file, triplets_state],
                 outputs=[triplets_state]
             ).then(
@@ -857,7 +907,7 @@ with gr.Blocks(title="🎓 AI Mass Grading System", theme=gr.themes.Soft()) as d
             )
             
             ans_file.change(
-                fn=create_file_update_handler(i, "ans_file"),
+                fn=create_file_handler(i, "ans_file"),
                 inputs=[ans_file, triplets_state],
                 outputs=[triplets_state]
             ).then(
@@ -867,11 +917,33 @@ with gr.Blocks(title="🎓 AI Mass Grading System", theme=gr.themes.Soft()) as d
             )
             
             imprint_check.change(
-                fn=create_imprint_update_handler(i),
+                fn=create_imprint_handler(i),
                 inputs=[imprint_check, triplets_state],
                 outputs=[triplets_state]
             )
+            
+            remove_btn.click(
+                fn=create_remove_handler(i),
+                inputs=[triplets_state],
+                outputs=[triplets_state]
+            ).then(
+                fn=update_status_table,
+                inputs=[triplets_state],
+                outputs=[status_table]
+            )
+    
+    # Add triplet button handler
+    add_triplet_btn.click(
+        fn=add_triplet,
+        inputs=[triplets_state],
+        outputs=[triplets_state]
+    ).then(
+        fn=update_status_table,
+        inputs=[triplets_state],
+        outputs=[status_table]
+    )
     
+    # Start grading button handler
     start_grading_btn.click(
         fn=start_mass_grading,
         inputs=[triplets_state],