Update app.py

app.py

@@ -11,6 +11,7 @@ import base64
 from io import BytesIO
 import uuid
 import logging
+import textwrap
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -27,136 +28,236 @@ body {
     min-height: 100vh;
     margin: 0;
     padding: 0;
-    font-family: Arial, sans-serif;
+    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
 }
 .gradio-container {
-    background-color: rgba(240, 248, 255, 0.95) !important;
-    border-radius: 10px;
-    padding: 20px;
-    margin: 20px;
-    max-width: 100%;
-    box-shadow: 0 4px 12px rgba(0,0,0,0.15);
+    background-color: rgba(255, 255, 255, 0.97) !important;
+    border-radius: 15px;
+    padding: 25px;
+    margin: 20px auto;
+    max-width: 1200px;
+    box-shadow: 0 8px 24px rgba(0,0,0,0.12);
     min-height: 90vh;
     border: 1px solid #1e90ff !important;
 }
 .risk-low { color: #28a745; font-weight: bold; }
-.risk-medium { color: #ffc107; font-weight: bold; }
-.risk-high { color: #dc3545; font-weight: bold; }
+.risk-medium { color: #ff9800; font-weight: bold; }
+.risk-high { color: #f44336; font-weight: bold; }
 .result-box {
     padding: 20px;
-    border-radius: 5px;
-    margin-bottom: 20px;
-    background-color: rgba(240, 248, 255, 0.9);
+    border-radius: 10px;
+    margin-bottom: 25px;
+    background-color: white;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.08);
     border-left: 5px solid #1e90ff !important;
 }
 .penalty-box {
-    padding: 15px;
-    border-radius: 5px;
-    margin-bottom: 15px;
-    border-left: 5px solid #dc3545;
-    background-color: rgba(255, 245, 245, 0.9);
+    padding: 20px;
+    border-radius: 10px;
+    margin-bottom: 20px;
+    border-left: 5px solid #f44336;
+    background-color: white;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.08);
 }
 .obligation-box {
-    padding: 15px;
-    border-radius: 5px;
-    margin-bottom: 15px;
-    border-left: 5px solid #ffc107;
-    background-color: rgba(255, 249, 230, 0.9);
+    padding: 20px;
+    border-radius: 10px;
+    margin-bottom: 20px;
+    border-left: 5px solid #ff9800;
+    background-color: white;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.08);
 }
 .delay-box {
-    padding: 15px;
-    border-radius: 5px;
-    margin-bottom: 15px;
-    border-left: 5px solid #17a2b8;
-    background-color: rgba(230, 249, 255, 0.9);
+    padding: 20px;
+    border-radius: 10px;
+    margin-bottom: 20px;
+    border-left: 5px solid #2196F3;
+    background-color: white;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.08);
 }
 .combined-risk-container {
     display: flex;
     flex-direction: column;
-    gap: 10px;
-    margin-bottom: 20px;
+    gap: 15px;
+    margin-bottom: 25px;
+    background-color: white;
+    padding: 20px;
+    border-radius: 10px;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.08);
 }
 .risk-row {
     display: flex;
     align-items: center;
-    gap: 15px;
-    padding: 10px;
-    border-radius: 5px;
-    background-color: rgba(240, 248, 255, 0.8);
-    border: 1px solid #add8e6 !important;
+    gap: 20px;
+    padding: 15px;
+    border-radius: 8px;
+    background-color: #f8f9fa;
+    border: 1px solid #e0e0e0 !important;
+    transition: all 0.3s ease;
+}
+.risk-row:hover {
+    transform: translateY(-2px);
+    box-shadow: 0 4px 12px rgba(0,0,0,0.1);
 }
 .risk-label {
-    width: 120px;
-    font-weight: bold;
+    width: 150px;
+    font-weight: 600;
     font-size: 16px;
-    color: #1e4b8f !important;
+    color: #2c3e50 !important;
 }
 .risk-score {
-    width: 100px;
-    font-size: 18px;
+    width: 120px;
+    font-size: 20px;
     text-align: center;
+    padding: 8px 12px;
+    border-radius: 6px;
 }
 .heatmap-wrapper {
     flex-grow: 1;
+    margin-top: 15px;
 }
 .warning-box {
-    padding: 15px;
-    border-radius: 5px;
-    margin: 10px 0;
-    background-color: rgba(255, 243, 205, 0.9);
-    border-left: 5px solid #ffc107;
-    font-weight: bold;
+    padding: 18px;
+    border-radius: 8px;
+    margin: 15px 0;
+    background-color: #fff3e0;
+    border-left: 5px solid #ff9800;
+    font-weight: 600;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.05);
 }
 .danger-box {
-    padding: 15px;
-    border-radius: 5px;
-    margin: 10px 0;
-    background-color: rgba(248, 215, 218, 0.9);
-    border-left: 5px solid #dc3545;
-    font-weight: bold;
+    padding: 18px;
+    border-radius: 8px;
+    margin: 15px 0;
+    background-color: #ffebee;
+    border-left: 5px solid #f44336;
+    font-weight: 600;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.05);
 }
 .success-box {
-    padding: 15px;
-    border-radius: 5px;
-    margin: 10px 0;
-    background-color: rgba(212, 237, 218, 0.9);
-    border-left: 5px solid #28a745;
-    font-weight: bold;
+    padding: 18px;
+    border-radius: 8px;
+    margin: 15px 0;
+    background-color: #e8f5e9;
+    border-left: 5px solid #4CAF50;
+    font-weight: 600;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.05);
+}
+.info-box {
+    padding: 18px;
+    border-radius: 8px;
+    margin: 15px 0;
+    background-color: #e3f2fd;
+    border-left: 5px solid #2196F3;
+    font-weight: 600;
+    box-shadow: 0 2px 8px rgba(0,0,0,0.05);
 }
 .section-title {
-    font-size: 20px;
-    font-weight: bold;
-    margin-bottom: 15px;
+    font-size: 22px;
+    font-weight: 700;
+    margin-bottom: 18px;
     color: #1e4b8f !important;
+    display: flex;
+    align-items: center;
+    gap: 10px;
 }
 .count-item {
     display: flex;
     justify-content: space-between;
-    padding: 10px 0;
-    border-bottom: 1px solid #add8e6 !important;
+    padding: 12px 0;
+    border-bottom: 1px solid #e0e0e0 !important;
+    transition: all 0.2s ease;
+}
+.count-item:hover {
+    background-color: #f5f5f5;
+    transform: translateX(5px);
 }
 .count-label {
-    font-weight: bold;
-    color: #1e4b8f !important;
+    font-weight: 600;
+    color: #2c3e50 !important;
+    display: flex;
+    align-items: center;
+    gap: 8px;
 }
 .count-value {
     color: #4169e1 !important;
+    font-weight: 600;
+    font-size: 16px;
 }
 button {
     background: linear-gradient(135deg, #1e90ff, #4169e1) !important;
     border: none !important;
     color: white !important;
-    font-weight: bold !important;
+    font-weight: 600 !important;
+    padding: 12px 24px !important;
+    border-radius: 8px !important;
+    transition: all 0.3s ease !important;
+    box-shadow: 0 4px 6px rgba(0,0,0,0.1) !important;
 }
 button:hover {
     background: linear-gradient(135deg, #4169e1, #1e90ff) !important;
+    transform: translateY(-2px) !important;
+    box-shadow: 0 6px 12px rgba(0,0,0,0.15) !important;
 }
 .upload-area {
     border: 2px dashed #1e90ff !important;
     background-color: rgba(240, 248, 255, 0.7) !important;
+    border-radius: 10px !important;
+    padding: 30px !important;
+    transition: all 0.3s ease !important;
 }
 .upload-area:hover {
     background-color: rgba(224, 255, 255, 0.7) !important;
+    border-color: #4169e1 !important;
+}
+.risk-meter {
+    width: 100%;
+    height: 20px;
+    background: linear-gradient(90deg, #4CAF50, #FFC107, #F44336);
+    border-radius: 10px;
+    margin: 15px 0;
+    position: relative;
+}
+.risk-meter-indicator {
+    position: absolute;
+    top: -5px;
+    width: 3px;
+    height: 30px;
+    background-color: #2c3e50;
+    transform: translateX(-50%);
+}
+.risk-meter-labels {
+    display: flex;
+    justify-content: space-between;
+    margin-top: 5px;
+    font-size: 12px;
+    color: #666;
+}
+.clause-example {
+    background-color: #f5f7fa;
+    padding: 15px;
+    border-radius: 8px;
+    margin-bottom: 10px;
+    border-left: 3px solid #1e90ff;
+    font-family: 'Courier New', monospace;
+    line-height: 1.5;
+}
+.clause-number {
+    font-weight: bold;
+    color: #1e90ff;
+    margin-right: 8px;
+}
+.sentiment-meter {
+    width: 100%;
+    height: 20px;
+    background: linear-gradient(90deg, #F44336, #FFC107, #4CAF50);
+    border-radius: 10px;
+    margin: 15px 0;
+}
+.sentiment-score {
+    height: 100%;
+    border-radius: 10px;
+    background-color: rgba(255,255,255,0.3);
 }
 /* Hide elements */
 footer, .gradio-footer, .hide, [data-testid="Use via API"], [data-testid="mmsettings"] {
@@ -272,6 +373,34 @@ def calculate_risk_score(penalty_count: int, penalty_values: List[float], obliga
     else:
         return score, "High"
 
+def generate_risk_meter(risk_score: float) -> str:
+    """Generate a visual risk meter with indicator"""
+    position = risk_score
+    return f"""
+    <div class="risk-meter">
+        <div class="risk-meter-indicator" style="left: {position}%"></div>
+    </div>
+    <div class="risk-meter-labels">
+        <span>Low (0-30)</span>
+        <span>Medium (31-69)</span>
+        <span>High (70-100)</span>
+    </div>
+    """
+
+def generate_sentiment_meter(sentiment_score: float) -> str:
+    """Generate a visual sentiment meter"""
+    width = sentiment_score * 100
+    return f"""
+    <div class="sentiment-meter">
+        <div class="sentiment-score" style="width: {width}%"></div>
+    </div>
+    <div style="display: flex; justify-content: space-between; margin-top: 5px;">
+        <span>Negative</span>
+        <span>Neutral</span>
+        <span>Positive</span>
+    </div>
+    """
+
 def generate_heatmap(risk_level: str):
     """Generate a simple heatmap based on risk level"""
     try:
@@ -279,10 +408,13 @@ def generate_heatmap(risk_level: str):
        
         if risk_level == "Low":
             cmap = plt.cm.Blues
+            color = '#4CAF50'
         elif risk_level == "Medium":
             cmap = plt.cm.Oranges
+            color = '#FF9800'
         else:
             cmap = plt.cm.Reds
+            color = '#F44336'
        
         gradient = np.linspace(0, 1, 256).reshape(1, -1)
         gradient = np.vstack((gradient, gradient))
@@ -322,11 +454,22 @@ def save_to_salesforce(sf: Salesforce, data: Dict):
 def format_warning_message(count: int, item_type: str, emoji: str) -> str:
     """Format warning message based on count with appropriate color coding"""
     if count == 0:
-        return f"""<div class="success-box">✅ {emoji} No {item_type} clauses detected - Good!</div>"""
+        return f"""<div class="success-box">✅ {emoji} No {item_type} clauses detected - This is excellent!</div>"""
+    elif count < 3:
+        return f"""<div class="info-box">ℹ️ {emoji} {count} {item_type} clauses detected - Standard contract language</div>"""
     elif count < 5:
         return f"""<div class="warning-box">⚠️ {emoji} {count} {item_type} clauses detected - Review recommended</div>"""
     else:
-        return f"""<div class="danger-box">🚨 {emoji} {count} {item_type} clauses detected - High Risk!</div>"""
+        return f"""<div class="danger-box">🚨 {emoji} {count} {item_type} clauses detected - High Risk! Needs immediate attention</div>"""
+
+def format_clause_example(example: str, index: int) -> str:
+    """Format a clause example with proper wrapping and styling"""
+    wrapped_text = textwrap.fill(example, width=80)
+    return f"""
+    <div class="clause-example">
+        <span class="clause-number">{index}.</span> {wrapped_text}
+    </div>
+    """
 
 def analyze_pdf(file_obj) -> List:
     """Main analysis function for Gradio interface"""
@@ -380,15 +523,17 @@ def analyze_pdf(file_obj) -> List:
 
         try:
             heatmap = generate_heatmap(risk_level)
+            risk_meter = generate_risk_meter(risk_score)
+            sentiment_meter = generate_sentiment_meter(sentiment_score)
         except Exception as e:
-            raise Exception(f"Heatmap generation failed: {str(e)}")
+            raise Exception(f"Visual generation failed: {str(e)}")
 
         # Format details with warning messages and emojis
         penalty_details = f"""
         {penalty_warning}
         <div class='penalty-box'>
             <div class='section-title'>💰 Penalty Clause Details</div>
-            {"".join([f"<div class='count-item'><span class='count-label'>{kw}</span><span class='count-value'>{count}</span></div>" for kw, count in penalty_counts.items()])}
+            {"".join([f"<div class='count-item'><span class='count-label'><span style='color: #f44336'>•</span> {kw}</span><span class='count-value'>{count}</span></div>" for kw, count in penalty_counts.items()])}
         </div>
         """
 
@@ -396,7 +541,7 @@ def analyze_pdf(file_obj) -> List:
         {obligation_warning}
         <div class='obligation-box'>
             <div class='section-title'>📝 Obligation Clause Details</div>
-            {"".join([f"<div class='count-item'><span class='count-label'>{kw}</span><span class='count-value'>{count}</span></div>" for kw, count in obligation_counts.items()])}
+            {"".join([f"<div class='count-item'><span class='count-label'><span style='color: #ff9800'>•</span> {kw}</span><span class='count-value'>{count}</span></div>" for kw, count in obligation_counts.items()])}
         </div>
         """
 
@@ -404,18 +549,18 @@ def analyze_pdf(file_obj) -> List:
         {delay_warning}
         <div class='delay-box'>
             <div class='section-title'>⏱️ Delay Clause Details</div>
-            {"".join([f"<div class='count-item'><span class='count-label'>{kw}</span><span class='count-value'>{count}</span></div>" for kw, count in delay_counts.items()])}
+            {"".join([f"<div class='count-item'><span class='count-label'><span style='color: #2196F3'>•</span> {kw}</span><span class='count-value'>{count}</span></div>" for kw, count in delay_counts.items()])}
         </div>
         """
 
-        penalty_amounts = "\n".join([f"<div class='count-item'><span class='count-label'>💰 Amount</span><span class='count-value'>${amt:,.2f}</span></div>" for amt in penalty_values[:5]]) if penalty_values else "<div class='success-box'>✅ No specific penalty amounts found</div>"
+        penalty_amounts = "\n".join([f"<div class='count-item'><span class='count-label'>💰 Amount</span><span class='count-value'>${amt:,.2f}</span></div>" for amt in penalty_values[:5]]) if penalty_values else "<div class='success-box'>✅ No specific penalty amounts found - This is good news!</div>"
        
         penalty_sentences = []
         for sentence in re.split(r'(?<=[.!?])\s+', text):
             if any(kw.lower() in sentence.lower() for kw in penalty_keywords):
                 penalty_sentences.append(sentence.strip())
        
-        penalty_examples = "\n\n".join([f"{i+1}. {sent}" for i, sent in enumerate(penalty_sentences[:3])]) if penalty_sentences else "<div class='success-box'>✅ No penalty clauses found</div>"
+        penalty_examples = "\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())
         sf_data = {
@@ -437,69 +582,119 @@ def analyze_pdf(file_obj) -> List:
             logger.error(f"Salesforce record creation failed: {str(e)}")
 
         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 "🚨"
+        risk_advice = {
+            "Low": "This contract appears to be low risk. Standard review recommended.",
+            "Medium": "This contract has moderate risk. Careful review advised.",
+            "High": "This contract is high risk! Immediate legal review required."
+        }
        
         return [
-            f"<div class='risk-row'><span class='risk-label'>Risk Score</span><span class='risk-score risk-{risk_level.lower()}'>{risk_score:.1f}/100</span></div>",
-            f"<div class='risk-row'><span class='risk-label'>Risk Level</span><span class='risk-score risk-{risk_level.lower()}'>{risk_level}</span></div>",
-            heatmap,
+            f"""
+            <div class='result-box'>
+                <div class='section-title'>{risk_icon} Contract Risk Summary</div>
+                <div class='risk-row'>
+                    <span class='risk-label'>Overall Risk Score</span>
+                    <span class='risk-score risk-{risk_level.lower()}'>{risk_score:.1f}/100</span>
+                </div>
+                {risk_meter}
+                <div style='margin-top: 15px; font-size: 16px;'>
+                    <strong>Assessment:</strong> {risk_advice[risk_level]}
+                </div>
+            </div>
+            """,
+            f"""
+            <div class='result-box'>
+                <div class='section-title'>📊 Risk Visualization</div>
+                {heatmap}
+            </div>
+            """,
             penalty_details,
-            f"<div class='penalty-box'><div class='section-title'>💰 Penalty Amounts</div>{penalty_amounts}</div>",
+            f"<div class='penalty-box'><div class='section-title'>💰 Penalty Amounts Found</div>{penalty_amounts}</div>",
             obligation_details,
             delay_details,
             f"<div class='result-box'><div class='section-title'>📜 Example Penalty Clauses</div>{penalty_examples}</div>",
-            f"<div class='{box_class}'><div class='section-title'>😊 Sentiment Analysis</div>Sentiment Score: {sentiment_score}</div>"
+            f"""
+            <div class='result-box'>
+                <div class='section-title'>😊 Contract Sentiment Analysis</div>
+                <div style='margin-bottom: 10px;'>Score: {sentiment_score:.2f} (0 = Negative, 1 = Positive)</div>
+                {sentiment_meter}
+                <div style='margin-top: 15px;'>
+                    <strong>Interpretation:</strong> {
+                        "Positive tone detected" if sentiment_score > 0.7 
+                        else "Neutral tone detected" if sentiment_score > 0.4 
+                        else "Negative tone detected"
+                    }
+                </div>
+            </div>
+            """
         ]
     except Exception as e:
         logger.error(f"Analysis failed: {str(e)}")
-        error_message = f"<div class='danger-box'>❌ Error: {str(e)}</div>"
-        return [error_message] * 9
+        error_message = f"""
+        <div class='danger-box'>
+            <div style='display: flex; align-items: center; gap: 10px;'>
+                <span style='font-size: 24px;'>❌</span>
+                <span style='font-size: 18px; font-weight: bold;'>Analysis Error</span>
+            </div>
+            <div style='margin-top: 10px;'>{str(e)}</div>
+            <div style='margin-top: 15px; font-size: 14px;'>
+                Please ensure you've uploaded a valid PDF document with selectable text.
+            </div>
+        </div>
+        """
+        return [error_message] * 8
 
 # Create Gradio interface with blue theme and hidden elements
 with gr.Blocks(css=css, title="PDF Contract Risk Analyzer", theme=gr.themes.Default(primary_hue="blue")) as demo:
     gr.Markdown("""
-        <div style='text-align: center;'>
-            <h1 style='color: #1e4b8f;'>📄 PDF Contract Risk Analyzer</h1>
-            <p style='color: #4169e1;'>Upload a contract PDF to analyze penalties, obligations, delays, and sentiment.</p>
+        <div style='text-align: center; margin-bottom: 30px;'>
+            <h1 style='color: #1e4b8f; margin-bottom: 10px;'>📄 PDF Contract Risk Analyzer</h1>
+            <p style='color: #4169e1; font-size: 16px;'>
+                Upload a contract PDF to analyze risks, obligations, and sentiment. Get instant insights into potential issues.
+            </p>
         </div>
     """)
    
     with gr.Row():
         with gr.Column(scale=1):
-            file_input = gr.File(label="📤 Upload PDF", file_types=[".pdf"], elem_classes="upload-area")
-            submit_btn = gr.Button("🔍 Analyze PDF", variant="primary")
+            file_input = gr.File(
+                label="📤 Upload Contract PDF", 
+                file_types=[".pdf"], 
+                elem_classes="upload-area",
+                info="Drag and drop your contract PDF file here"
+            )
+            submit_btn = gr.Button("🔍 Analyze Contract", variant="primary")
        
         with gr.Column(scale=3):
-            gr.Markdown("<div class='section-title'>🔍 Overall Risk Assessment</div>")
-            with gr.Group(elem_classes="combined-risk-container"):
-                risk_score = gr.HTML(label="Risk Score")
-                risk_level = gr.HTML(label="Risk Level")
-                heatmap = gr.Plot(label="Risk Heatmap", elem_classes="heatmap-wrapper")
+            risk_summary = gr.HTML(label="Contract Risk Summary")
+            risk_visualization = gr.HTML(label="Risk Visualization")
    
     with gr.Row():
         with gr.Column():
-            penalty_count = gr.HTML(label="💰 Penalty Clauses")
-            penalty_amounts = gr.HTML(label="💰 Penalty Amounts")
+            penalty_count = gr.HTML(label="Penalty Clauses Analysis")
+            penalty_amounts = gr.HTML(label="Penalty Amounts Found")
        
         with gr.Column():
-            obligation_count = gr.HTML(label="📝 Obligation Clauses")
+            obligation_count = gr.HTML(label="Obligation Clauses Analysis")
        
         with gr.Column():
-            delay_count = gr.HTML(label="⏱️ Delay Clauses")
+            delay_count = gr.HTML(label="Delay Clauses Analysis")
    
     with gr.Row():
-        penalty_examples = gr.HTML(label="📜 Example Penalty Clauses")
+        penalty_examples = gr.HTML(label="Example Penalty Clauses")
    
     with gr.Row():
-        additional_results = gr.HTML(label="😊 Sentiment Analysis")
+        sentiment_analysis = gr.HTML(label="Contract Sentiment Analysis")
    
     submit_btn.click(
         fn=analyze_pdf,
         inputs=file_input,
         outputs=[
-            risk_score, risk_level, heatmap,
+            risk_summary, risk_visualization,
             penalty_count, penalty_amounts,
             obligation_count, delay_count,
-            penalty_examples, additional_results
+            penalty_examples, sentiment_analysis
         ]
     )