Update app.py

app.py

@@ -6,7 +6,6 @@ from word2number import w2n
 import re
 from typing import Tuple, List, Dict
 from simple_salesforce import Salesforce
-import os
 import base64
 from io import BytesIO
 import uuid
@@ -16,20 +15,13 @@ from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
 from reportlab.lib.units import inch
 import time
+import tempfile
+import os
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
-# Configuration for public URL base
-CONFIG = {
-    "PUBLIC_URL_BASE": "https://huggingface.co/spaces/YourUsername/YourSpace/resolve/main/static/output/",
-    "OUTPUT_DIR": "static/output"
-}
-
-# Ensure output directory exists
-os.makedirs(CONFIG["OUTPUT_DIR"], exist_ok=True)
-
 # Custom CSS for styling with dark mode compatibility
 css = """
 :root {
@@ -223,8 +215,16 @@ body {
     transition: all 0.2s ease;
 }
 
+.count-item {
+    display: flex;
+    justify-content: flex-start;
+    align-items: center;
+    padding: 10px 0;
+    gap: 10px;
+}
+
 .count-item:hover {
-    background-color: rgba(0,0,0,0.05);
+    background-color: rgba(0,255,255,0.05);
     transform: translateX(5px);
 }
 
@@ -256,20 +256,20 @@ button {
 button:hover {
     background: linear-gradient(135deg, var(--secondary-color), var(--primary-color)) !important;
     transform: translateY(-2px) !important;
-    box-shadow: 0 6px 12px rgba(0,0,0,0.15) !important;
+    box-shadow: 0 6px 12px rgba(0,0,0,0.15)!important;
 }
 
 .upload-area {
     border: 2px dashed var(--primary-color) !important;
-    background-color: rgba(240, 248, 255, 0.3) !important;
+    background-color: rgba(240, 248, 255, 0.3)!important;
     border-radius: 10px !important;
     padding: 30px !important;
     transition: all 0.3s ease !important;
 }
 
 .upload-area:hover {
-    background-color: rgba(224, 255, 255, 0.3) !important;
-    border-color: var(--secondary-color) !important;
+    background-color: rgba(224,255,255,0.3)!important;
+    border-color: var(--secondary-color)!important;
 }
 
 .risk-meter {
@@ -392,11 +392,9 @@ def get_hugging_face_sentiment(text: str) -> float:
         logger.error(f"Hugging Face sentiment analysis failed: {str(e)}. Using fallback score.")
         return 0.5
 
-def generate_sentiment_pdf(sentiment_score: float, output_dir: str) -> tuple[str, str, BytesIO]:
-    """Generate a PDF with sentiment analysis results and return path, URL, and file buffer"""
+def generate_sentiment_pdf(sentiment_score: float) -> BytesIO:
+    """Generate a PDF with sentiment analysis results in memory and return BytesIO buffer"""
     try:
-        pdf_filename = f"sentiment_analysis_{int(time.time())}.pdf"
-        pdf_path = os.path.join(output_dir, pdf_filename)
         pdf_file = BytesIO()
         c = canvas.Canvas(pdf_file, pagesize=letter)
 
@@ -428,16 +426,11 @@ def generate_sentiment_pdf(sentiment_score: float, output_dir: str) -> tuple[str
 
         c.save()
         pdf_file.seek(0)
-
-        with open(pdf_path, "wb") as f:
-            f.write(pdf_file.getvalue())
-
-        public_url = f"{CONFIG['PUBLIC_URL_BASE']}{pdf_filename}"
-        logger.info(f"PDF generated: {public_url}")
-        return pdf_path, public_url, pdf_file
+        logger.info("PDF generated in memory")
+        return pdf_file
     except Exception as e:
         logger.error(f"Error generating PDF: {str(e)}")
-        return "", "", None
+        return None
 
 def save_to_salesforce(sf: Salesforce, data: Dict) -> str:
     """Save analysis results to Salesforce, return record ID"""
@@ -721,10 +714,16 @@ def analyze_pdf(file_obj) -> List:
             salesforce_id = "N/A"
 
         try:
-            pdf_path, public_url, pdf_file = generate_sentiment_pdf(sentiment_score, CONFIG["OUTPUT_DIR"])
+            pdf_buffer = generate_sentiment_pdf(sentiment_score)
+            if pdf_buffer is None:
+                raise Exception("Failed to generate PDF")
+            # Save to a temporary file for Gradio to serve
+            with tempfile.NamedTemporaryFile(suffix=".pdf", delete=False) as temp_file:
+                temp_file.write(pdf_buffer.getvalue())
+                temp_file_path = temp_file.name
         except Exception as e:
             logger.error(f"PDF generation failed: {str(e)}")
-            public_url = None
+            temp_file_path = None
 
         box_class = "success-box" if risk_level == "Low" else "warning-box" if risk_level == "Medium" else "danger-box"
         risk_icon = "✅" if risk_level == "Low" else "⚠" if risk_level == "Medium" else "🚨"
@@ -734,7 +733,7 @@ def analyze_pdf(file_obj) -> List:
             "High": "This contract is high risk! Immediate legal review required."
         }
         
-        # Add sentiment analysis PDF download link
+        # Sentiment analysis output with PDF download prompt
         sentiment_analysis_output = f"""
         <div class='result-box'>
             <div class='section-title'>📊 Sentiment Analysis</div>
@@ -744,8 +743,7 @@ def analyze_pdf(file_obj) -> List:
             </div>
             {sentiment_meter}
             <div style='margin-top: 15px;'>
-                <strong>Download Sentiment Report:</strong>
-                {'<a href="' + public_url + '" target="_blank" style="color: var(--primary-color); text-decoration: underline;">Download PDF</a>' if public_url else 'Failed to generate PDF download link'}
+                <strong>Sentiment Report:</strong> {'' if temp_file_path else 'Failed to generate PDF'}
             </div>
         </div>
         """
@@ -770,7 +768,8 @@ def analyze_pdf(file_obj) -> List:
             obligation_details,
             delay_details,
             f"<div class='result-box'><div class='section-title'>📜 Extracted Data</div>{extracted_data}</div>",
-            sentiment_analysis_output
+            sentiment_analysis_output,
+            temp_file_path  # Return temporary file path for PDF download
         ]
     except Exception as e:
         logger.error(f"Analysis failed: {str(e)}")
@@ -786,7 +785,14 @@ def analyze_pdf(file_obj) -> List:
             </div>
         </div>
         """
-        return [error_message] * 8
+        return [error_message] * 9
+    finally:
+        if temp_file_path and os.path.exists(temp_file_path):
+            try:
+                os.remove(temp_file_path)
+                logger.info(f"Cleaned up temporary PDF file: {temp_file_path}")
+            except Exception as e:
+                logger.error(f"Failed to clean up temporary PDF file {temp_file_path}: {e}")
 
 # Create Gradio interface with dark mode compatibility
 with gr.Blocks(css=css, title="PDF Contract Risk Analyzer", theme=gr.themes.Default(primary_hue="blue")) as demo:
@@ -833,6 +839,7 @@ with gr.Blocks(css=css, title="PDF Contract Risk Analyzer", theme=gr.themes.Defa
     
     with gr.Row():
         sentiment_analysis = gr.HTML(label="Sentiment Analysis")
+        pdf_output = gr.File(label="Download Sentiment Report PDF", file_types=[".pdf"])
     
     submit_btn.click(
         fn=analyze_pdf,
@@ -841,7 +848,7 @@ with gr.Blocks(css=css, title="PDF Contract Risk Analyzer", theme=gr.themes.Defa
             risk_summary, risk_visualization,
             penalty_count, penalty_amounts,
             obligation_count, delay_count,
-            extracted_data, sentiment_analysis
+            extracted_data, sentiment_analysis, pdf_output
         ]
     )