Update app.py

app.py

@@ -3,30 +3,6 @@ from utils import extract_text_from_pdf, simple_clause_split
 from model import classify_clauses
 from report import generate_pdf
 from salesforce_stub import send_to_salesforce
-import plotly.graph_objects as go
-
-# Function to generate heatmap
-def generate_heatmap(results):
-    risk_levels = {"High": 3, "Medium": 2, "Low": 1}
-    
-    clauses = [r['clause'] for r in results]
-    risk_values = [risk_levels.get(r['risk_level'], 1) for r in results]
-
-    fig = go.Figure(data=go.Heatmap(
-        z=[risk_values],
-        x=clauses,
-        y=["Risk Levels"],
-        colorscale='YlOrRd',  # You can change the color scale
-        colorbar=dict(title="Risk Level")
-    ))
-
-    fig.update_layout(
-        title="Contract Risk Heatmap",
-        xaxis_title="Clause",
-        yaxis_title="Risk Level"
-    )
-
-    return fig.to_html(full_html=False)
 
 # Initialize Gradio interface
 def analyze_contract(file):
@@ -38,33 +14,21 @@ def analyze_contract(file):
     overall_score = sum(r['risk_score'] for r in results) / len(results) if results else 0
     report_path = generate_pdf(results, overall_score)
 
-    # Create clause-level risk highlights
     highlight_output = ""
-    high_risk_clauses = ""
-    
     for r in results:
         color = (
             "red" if r['risk_level'] == "High" else
             "orange" if r['risk_level'] == "Medium" else
             "green"
         )
-
-        # High-risk clauses separately
-        if r['risk_level'] == "High":
-            high_risk_clauses += f"<div style='color:{color}'><b>{r['risk_level']}</b>: {r['clause']}</div><br>"
-
         highlight_output += f"<div style='color:{color}'><b>{r['risk_level']}</b>: {r['clause']}</div><br>"
 
-    # Generate heatmap HTML
-    heatmap_html = generate_heatmap(results)
-
-    # Send results to Salesforce (if necessary)
     send_to_salesforce({
         "clauses": results,
         "overall_score": overall_score
     })
 
-    return f"Overall Risk Score: {overall_score:.2f}", high_risk_clauses, highlight_output, heatmap_html, report_path
+    return f"Overall Risk Score: {overall_score:.2f}", highlight_output, report_path
 
 # Create Gradio Interface
 iface = gr.Interface(
@@ -72,9 +36,7 @@ iface = gr.Interface(
     inputs=gr.File(label="Upload Contract PDF"),
     outputs=[
         gr.Textbox(label="Overall Risk Score"),
-        gr.HTML(label="High-Risk Clauses"),  # Display high-risk clauses here
-        gr.HTML(label="Clause Risk Highlight"),  # Heatmap-style highlighting
-        gr.HTML(label="Risk Heatmap"),  # Display heatmap here
+        gr.HTML(label="Clause Risk Highlight"),
         gr.File(label="Download Risk Report (PDF)")
     ],
     title="📜 Contract Risk Heatmap Generator",
@@ -83,5 +45,4 @@ iface = gr.Interface(
 
 # Ensure the Gradio app starts properly
 if _name_ == "_main_":
-    iface.launch()
-    
+    iface.launch()