| import gradio as gr | |
| import time | |
| from utils import ( | |
| extract_text_from_pdf, | |
| summarize_text, | |
| generate_pdf, | |
| extract_fields_from_summary, | |
| send_to_salesforce | |
| ) | |
| def process_and_summarize(file): | |
| start_time = time.time() | |
| try: | |
| text = extract_text_from_pdf(file) | |
| char_count = len(text) | |
| summary = summarize_text(text) | |
| fields = extract_fields_from_summary(summary) | |
| sf_response = send_to_salesforce(fields) | |
| pdf_path = generate_pdf(summary) | |
| return summary, pdf_path, char_count, round(time.time() - start_time, 2), fields, sf_response | |
| except Exception as e: | |
| return f"Error: {str(e)}", None, 0, 0.0, {}, {} | |
| iface = gr.Interface( | |
| fn=process_and_summarize, | |
| inputs=gr.File(label="📄 Upload Contract (PDF)"), | |
| outputs=[ | |
| gr.Textbox(label="📝 Contract Summary", lines=15, elem_id="summary-box"), | |
| gr.File(label="📥 Download PDF Summary"), | |
| gr.Number(label="🧮 Character Count"), | |
| gr.Number(label="⏱️ Processing Time (s)"), | |
| gr.JSON(label="📤 Fields Sent to Salesforce"), | |
| gr.JSON(label="📥 Salesforce Response"), | |
| gr.HTML( | |
| """ | |
| 🖨️ Print Summary | |
| """ | |
| ) | |
| ], | |
| title="🧠 AI Contract Summarizer + Salesforce", | |
| description="Upload a contract to summarize, download, print, and sync with Salesforce.", | |
| allow_flagging="never" | |
| ) | |
| if __name__ == "__main__": | |
| iface.launch() |