Update app.py

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import os
-from io import BytesIO
+import json
 from utils import (
     get_salesforce_client, get_salesforce_objects, get_object_fields,
     extract_text_from_pdf, extract_key_value_pairs, map_fields,
@@ -10,7 +10,8 @@ from utils import (
 # Ensure uploads directory exists
 os.makedirs("uploads", exist_ok=True)
 
-# Gradio Interface Functions
+# --- Gradio Interface Functions ---
+
 def upload_pdfs(files):
     uploaded_files = []
     if isinstance(files, list):
@@ -65,19 +66,19 @@ def fetch_fields(object_name):
 def process_pdf(pdf_paths):
     if not pdf_paths or not isinstance(pdf_paths, list) or not pdf_paths[0]:
         return "Error: No valid PDF file provided"
-    pdf_path = pdf_paths[0]  # Take the first file from the list
+    pdf_path = pdf_paths[0]
     text_data, error = extract_text_from_pdf(pdf_path)
     if error:
         return f"Error: {error}"
     kv_pairs, error = extract_key_value_pairs(pdf_path)
     if error:
         return f"Error: {error}"
-    return f"Text: {text_data}\nKey-Value Pairs: {kv_pairs}"
+    return f"Text:\n{text_data}\n\nKey-Value Pairs:\n{kv_pairs}"
 
 def display_mappings(pdf_paths, object_name):
     if not pdf_paths or not isinstance(pdf_paths, list) or not pdf_paths[0]:
         return "Error: No valid PDF file provided"
-    pdf_path = pdf_paths[0]  # Take the first file from the list
+    pdf_path = pdf_paths[0]
     sf, error = get_salesforce_client()
     if error:
         return f"Error: {error}"
@@ -98,7 +99,7 @@ def display_mappings(pdf_paths, object_name):
 def migrate_to_salesforce(pdf_paths, object_name):
     if not pdf_paths or not isinstance(pdf_paths, list) or not pdf_paths[0]:
         return "Error: No valid PDF file provided"
-    pdf_path = pdf_paths[0]  # Take the first file from the list
+    pdf_path = pdf_paths[0]
     sf, error = get_salesforce_client()
     if error:
         log_failure(pdf_path, object_name, error)
@@ -124,7 +125,7 @@ def migrate_to_salesforce(pdf_paths, object_name):
     if error:
         log_failure(pdf_path, object_name, error)
         return f"Error: {error}"
-    return f"Record Created: {record_id}\n{attach_status}"
+    return f"✅ Record Created: {record_id}\n📎 Attachment: {attach_status}"
 
 def display_failures():
     try:
@@ -138,57 +139,58 @@ def display_failures():
         return "No failures logged"
 
 def retry_migration(pdf_path, object_name):
-    return migrate_to_salesforce([pdf_path], object_name)  # Wrap single path in list for consistency
-
-# Gradio UI
-with gr.Blocks() as app:
-    gr.Markdown("# Smart Contract Migrator")
-
-    # PDF Upload Section
-    gr.Markdown("## Upload PDFs")
-    pdf_upload = gr.File(label="Upload Contract PDFs", file_count="multiple", file_types=[".pdf"])
-    upload_status = gr.Textbox(label="Upload Status")
-    pdf_upload.change(upload_pdfs, inputs=pdf_upload, outputs=upload_status)
-
-    # Salesforce Object Selection
-    gr.Markdown("## Salesforce Object Selection")
-    object_dropdown = gr.Dropdown(label="Select Salesforce Object")
-    fetch_objects_button = gr.Button("Fetch Objects")
-    object_status = gr.Textbox(label="Object Fetch Status")
-    fetch_objects_button.click(fetch_objects, outputs=[object_dropdown, object_status])
-
-    # Field Selection
-    field_checkboxes = gr.CheckboxGroup(label="Select Fields")
-    fetch_fields_button = gr.Button("Fetch Fields")
-    field_status = gr.Textbox(label="Field Fetch Status")
-    fetch_fields_button.click(fetch_fields, inputs=object_dropdown, outputs=[field_checkboxes, field_status])
-
-    # PDF Processing
-    gr.Markdown("## Process PDF")
-    process_button = gr.Button("Process PDF")
-    process_output = gr.Textbox(label="Extracted Data")
-    process_button.click(process_pdf, inputs=pdf_upload, outputs=process_output)
-
-    # Field Mapping
-    gr.Markdown("## Field Mapping")
-    map_button = gr.Button("Map Fields")
-    mapping_output = gr.Textbox(label="Field Mappings")
-    map_button.click(display_mappings, inputs=[pdf_upload, object_dropdown], outputs=mapping_output)
-
-    # Salesforce Migration
-    gr.Markdown("## Migrate to Salesforce")
-    migrate_button = gr.Button("Migrate to Salesforce")
-    migrate_output = gr.Textbox(label="Migration Status")
-    migrate_button.click(migrate_to_salesforce, inputs=[pdf_upload, object_dropdown], outputs=migrate_output)
-
-    # Reconciliation Dashboard
-    gr.Markdown("## Reconciliation Dashboard")
-    show_failures_button = gr.Button("Show Failures")
-    failures_output = gr.Textbox(label="Failed Migrations")
-    pdf_path_input = gr.Textbox(label="PDF Path to Retry")
-    retry_button = gr.Button("Retry Migration")
-    retry_output = gr.Textbox(label="Retry Status")
-    show_failures_button.click(display_failures, outputs=failures_output)
-    retry_button.click(retry_migration, inputs=[pdf_path_input, object_dropdown], outputs=retry_output)
-
-app.launch()
+    return migrate_to_salesforce([pdf_path], object_name)
+
+# --- Gradio App Layout ---
+
+with gr.Blocks(css="footer {display: none !important;}") as app:
+    gr.Markdown("# 🧠 Smart Contract Migrator")
+
+    with gr.Tab("📄 Upload & Process PDF"):
+        with gr.Row():
+            pdf_upload = gr.File(label="Upload Contract PDFs", file_count="multiple", file_types=[".pdf"])
+            upload_status = gr.Textbox(label="Status", interactive=False)
+        pdf_upload.change(upload_pdfs, inputs=pdf_upload, outputs=upload_status)
+
+        with gr.Row():
+            process_button = gr.Button("Extract Info")
+            process_output = gr.Textbox(label="Text & Key-Value Output", lines=6, interactive=False)
+        process_button.click(process_pdf, inputs=pdf_upload, outputs=process_output)
+
+    with gr.Tab("🔁 Salesforce Integration"):
+        with gr.Row():
+            fetch_objects_button = gr.Button("🔄 Get Objects")
+            object_dropdown = gr.Dropdown(label="Salesforce Object")
+            object_status = gr.Textbox(label="Status", interactive=False)
+        fetch_objects_button.click(fetch_objects, outputs=[object_dropdown, object_status])
+
+        with gr.Row():
+            fetch_fields_button = gr.Button("🔄 Get Fields")
+            field_checkboxes = gr.CheckboxGroup(label="Fields")
+            field_status = gr.Textbox(label="Status", interactive=False)
+        fetch_fields_button.click(fetch_fields, inputs=object_dropdown, outputs=[field_checkboxes, field_status])
+
+        with gr.Row():
+            map_button = gr.Button("🔗 Map Fields")
+            mapping_output = gr.Textbox(label="Mappings", lines=6, interactive=False)
+        map_button.click(display_mappings, inputs=[pdf_upload, object_dropdown], outputs=mapping_output)
+
+        with gr.Row():
+            migrate_button = gr.Button("🚀 Migrate to Salesforce")
+            migrate_output = gr.Textbox(label="Migration Result", lines=4, interactive=False)
+        migrate_button.click(migrate_to_salesforce, inputs=[pdf_upload, object_dropdown], outputs=migrate_output)
+
+    with gr.Tab("🧾 Reconciliation"):
+        with gr.Row():
+            show_failures_button = gr.Button("📋 Show Failures")
+            failures_output = gr.Textbox(label="Failures", lines=6, interactive=False)
+        show_failures_button.click(display_failures, outputs=failures_output)
+
+        with gr.Row():
+            pdf_path_input = gr.Textbox(label="PDF Path to Retry")
+            retry_button = gr.Button("🔁 Retry")
+            retry_output = gr.Textbox(label="Retry Status", interactive=False)
+        retry_button.click(retry_migration, inputs=[pdf_path_input, object_dropdown], outputs=retry_output)
+
+# --- Launch App ---
+app.launch()