Update app.py

app.py

@@ -14,7 +14,6 @@ import logging
 import textwrap
 import tempfile
 
-
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
@@ -562,7 +561,8 @@ def save_to_salesforce(sf: Salesforce, data: Dict):
             'Penalty_Details__c': data['penalty_details'][:131072],
             'Penalty_Amounts__c': data['penalty_amounts'][:131072],
             'Obligation_Details__c': data['obligation_details'][:131072],
-            'Delay_Details__c': data['delay_details'][:131072]
+            'Delay_Details__c': data['delay_details'][:131072],
+            'PDF_URL__c': data['pdf_url']
         }
         result = sf.Custom_Risk_Analysis__c.create(record)
         logger.info(f"Successfully created Salesforce record: {result['id']}")
@@ -593,6 +593,11 @@ def format_clause_example(example: str, index: int) -> str:
 
 def analyze_pdf(file_obj) -> List:
     """Main analysis function for Gradio interface"""
+    # Initialize variables that might be used in error handling
+    pdf_url = "Not uploaded"
+    project_title = "Unknown"
+    record_id = str(uuid.uuid4())
+    
     try:
         if not file_obj:
             raise Exception("No PDF file uploaded. Please upload a valid PDF file.")
@@ -682,7 +687,6 @@ def analyze_pdf(file_obj) -> List:
        
         extracted_data = "\n".join([format_clause_example(sent, i+1) for i, sent in enumerate(penalty_sentences[:3])]) if penalty_sentences else "<div class='success-box'>✅ No penalty clauses found - Excellent contract terms!</div>"
        
-        record_id = str(uuid.uuid4())
         project_title = os.path.splitext(os.path.basename(file_obj.name))[0]
         
         # Generate PDF report
@@ -715,20 +719,12 @@ def analyze_pdf(file_obj) -> List:
             logger.error(f"PDF report generation failed: {str(e)}")
             pdf_report = None
         
-        # Save PDF to temporary directory if generated
-        pdf_path = None
+        # Upload PDF to Salesforce if generated
         if pdf_report:
-            with tempfile.TemporaryDirectory() as temp_dir:
-                pdf_path = os.path.join(temp_dir, f"{project_title}_report.pdf")
-                with open(pdf_path, "wb") as f:
-                    f.write(pdf_report.getvalue())
-                
-                # Upload PDF to Salesforce
-                pdf_content_id, pdf_url = None, None
-                if pdf_report:
-                    pdf_content_id, pdf_url = upload_pdf_to_salesforce(pdf_report, project_title, record_id)
-                    if not pdf_content_id:
-                        logger.warning("Failed to upload PDF, but proceeding with record creation")
+            pdf_content_id, pdf_url = upload_pdf_to_salesforce(pdf_report, project_title, record_id)
+            if not pdf_content_id:
+                logger.warning("Failed to upload PDF to Salesforce")
+                pdf_url = "Not uploaded"
 
         sf_data = {
             'sentiment_score': sentiment_score,
@@ -740,7 +736,7 @@ def analyze_pdf(file_obj) -> List:
             'penalty_amounts': "\n".join([f"${amt:,.2f}" for amt in penalty_values[:5]]) if penalty_values else "No specific penalty amounts found",
             'obligation_details': "\n".join([f"{kw}: {count}" for kw, count in obligation_counts.items()]),
             'delay_details': "\n".join([f"{kw}: {count}" for kw, count in delay_counts.items()]),
-            'pdf_url': pdf_url if pdf_url else "Not uploaded"
+            'pdf_url': pdf_url
         }
        
         try: