Update app.py

app.py

@@ -177,13 +177,12 @@ with gr.Blocks(title="Smart Contract Migrator (Local Mode)") as demo:
         status, ai_result, updated_records, progress = process_contract(files, obj_name, manual_mappings_dict)
         global failed_records
         failed_records = updated_records
-        upload_progress.update(value=progress)
-        return status, ai_result if ai_result else {}
+        return status, ai_result if ai_result else {}, gr.update(value=progress)
 
     process_button.click(
         fn=process_and_display,
         inputs=[uploaded_files, object_name] + [comp for comp in manual_mapping_inputs.value.values()],
-        outputs=[status_output, ai_result_output]
+        outputs=[status_output, ai_result_output, upload_progress]
     )
 
     # EPIC 6: Reconciliation & Retry
@@ -202,8 +201,7 @@ with gr.Blocks(title="Smart Contract Migrator (Local Mode)") as demo:
             result, updated_records, progress = retry_failed_record(int(index), obj_name, manual_mappings_dict)
             global failed_records
             failed_records = updated_records
-            upload_progress.update(value=progress)
-            return result, update_reconciliation()
+            return result, update_reconciliation(), gr.update(value=progress)
 
         retry_index = gr.Number(label="Select Failed Record Index", interactive=False)
         retry_manual_inputs = gr.State(value={field: gr.Textbox(label=f"{field} (Retry)", interactive=True, value="") for field in manual_mapping_inputs.value.keys()})
@@ -213,7 +211,7 @@ with gr.Blocks(title="Smart Contract Migrator (Local Mode)") as demo:
         retry_button.click(
             fn=retry_and_update,
             inputs=[retry_index, object_name] + [comp for comp in retry_manual_inputs.value.values()],
-            outputs=[retry_status, failed_records_output]
+            outputs=[retry_status, failed_records_output, upload_progress]
         )
 
 demo.launch()