Update app.py

app.py

@@ -114,4 +114,62 @@ def generate_heatmap(risk_scores):
     buffer.seek(0)
     return buffer
 
-def generate_pdf_report(results, heatmap
+def generate_pdf_report(results, heatmap_buffer):
+    """Generate PDF report with summary and heatmap."""
+    buffer = BytesIO()
+    c = canvas.Canvas(buffer, pagesize=letter)
+    c.setFont("Helvetica", 12)
+    c.drawString(50, 750, "Contract Risk Analysis Report")
+    
+    # Summary
+    c.drawString(50, 720, "Summary of Risk-Prone Clauses:")
+    y = 700
+    for result in results[:5]:  # Limit to top 5 for brevity
+        text = f"Clause {result['clause_id']}: {result['clause_type'].capitalize()} (Risk: {result['risk_score']})"
+        c.drawString(50, y, text[:80] + "..." if len(text) > 80 else text)
+        y -= 20
+
+    # Embed heatmap
+    if heatmap_buffer:
+        c.drawImage(BytesIO(heatmap_buffer.read()), 50, y-200, width=500, height=100)
+    
+    c.showPage()
+    c.save()
+    buffer.seek(0)
+    return buffer
+
+def process_contract(pdf_file):
+    """Main function to process uploaded contract."""
+    # Extract text
+    text = extract_text_from_pdf(pdf_file)
+    if "Error" in text:
+        return text, None, None, {"Error": text}
+
+    # Parse and classify
+    results, risk_scores = parse_contract(text)
+    if not results:
+        return "No relevant clauses detected.", None, None, {"Summary": "No risk-prone clauses found."}
+
+    # Generate outputs
+    json_output = json.dumps(results, indent=2)
+    heatmap_buffer = generate_heatmap(risk_scores)
+    pdf_report = generate_pdf_report(results, heatmap_buffer)
+
+    return json_output, heatmap_buffer, pdf_report, {"Summary": f"Detected {len(results)} risk-prone clauses."}
+
+# Gradio interface
+iface = gr.Interface(
+    fn=process_contract,
+    inputs=gr.File(label="Upload Contract PDF"),
+    outputs=[
+        gr.Textbox(label="JSON Output"),
+        gr.Image(label="Risk Heatmap"),
+        gr.File(label="Download PDF Report"),
+        gr.JSON(label="Summary")
+    ],
+    title="Contract Risk Analyzer",
+    description="Upload a contract PDF to analyze risk-prone clauses and visualize results."
+)
+
+if __name__ == "__main__":
+    iface.launch()