Update app.py

app.py

@@ -12,6 +12,11 @@ from io import BytesIO
 import uuid
 import logging
 import textwrap
+from reportlab.lib.pagesizes import letter
+from reportlab.platypus import SimpleDocTemplate, Paragraph, Spacer, Image
+from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
+from reportlab.lib.enums import TA_CENTER
+from reportlab.lib import colors
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -332,6 +337,10 @@ mark {
     border-radius: 3px;
     font-weight: bold;
 }
+.download-btn {
+    margin-top: 20px !important;
+    background: linear-gradient(135deg, #4CAF50, #2E7D32) !important;
+}
 /* Hide elements */
 footer, .gradio-footer, .hide, [data-testid="Use via API"], [data-testid="mmsettings"], 
 .sentiment-analysis, .risk-visualization {
@@ -492,7 +501,13 @@ def generate_heatmap(risk_level: str):
        
         ax.set_axis_off()
         plt.tight_layout()
-        return fig
+        
+        # Save the figure to a BytesIO object
+        buf = BytesIO()
+        plt.savefig(buf, format='png', bbox_inches='tight')
+        plt.close(fig)
+        buf.seek(0)
+        return buf
     except Exception as e:
         logger.error(f"Heatmap generation failed: {str(e)}")
         raise Exception(f"Heatmap generation failed: {str(e)}")
@@ -570,6 +585,78 @@ def format_clause_example(example: str, index: int) -> str:
     </div>
     """
 
+def generate_pdf_report(analysis_results: dict, heatmap_buffer: BytesIO) -> BytesIO:
+    """Generate a PDF report from the analysis results"""
+    buffer = BytesIO()
+    doc = SimpleDocTemplate(buffer, pagesize=letter)
+    
+    styles = getSampleStyleSheet()
+    styles.add(ParagraphStyle(name='Center', alignment=TA_CENTER))
+    styles.add(ParagraphStyle(name='RiskHigh', textColor=colors.red, fontSize=14))
+    styles.add(ParagraphStyle(name='RiskMedium', textColor=colors.orange, fontSize=14))
+    styles.add(ParagraphStyle(name='RiskLow', textColor=colors.green, fontSize=14))
+    styles.add(ParagraphStyle(name='NormalBold', fontName='Helvetica-Bold', parent=styles['Normal']))
+    
+    story = []
+    
+    # Title
+    story.append(Paragraph("Contract Risk Analysis Report", styles['Title']))
+    story.append(Spacer(1, 12))
+    
+    # Risk Summary
+    story.append(Paragraph("<b>Risk Summary</b>", styles['Heading2']))
+    risk_style = 'RiskHigh' if analysis_results['risk_level'] == 'High' else 'RiskMedium' if analysis_results['risk_level'] == 'Medium' else 'RiskLow'
+    story.append(Paragraph(f"Risk Score: {analysis_results['risk_score']:.1f}/100 - <b>{analysis_results['risk_level']} Risk</b>", styles[risk_style]))
+    story.append(Spacer(1, 12))
+    
+    # Add heatmap to PDF
+    if heatmap_buffer:
+        story.append(Image(heatmap_buffer, width=400, height=100))
+        story.append(Spacer(1, 24))
+    
+    # Key Findings
+    story.append(Paragraph("<b>Key Findings</b>", styles['Heading2']))
+    story.append(Paragraph(f"<b>Penalty Clauses:</b> {analysis_results['penalty_count']}", styles['Normal']))
+    story.append(Paragraph(f"<b>Obligation Clauses:</b> {analysis_results['obligation_count']}", styles['Normal']))
+    story.append(Paragraph(f"<b>Delay Clauses:</b> {analysis_results['delay_count']}", styles['Normal']))
+    story.append(Spacer(1, 12))
+    
+    if analysis_results['penalty_values']:
+        story.append(Paragraph("<b>Penalty Amounts Found</b>", styles['Heading3']))
+        for amount in analysis_results['penalty_values'][:5]:
+            story.append(Paragraph(f"${amount:,.2f}", styles['Normal']))
+        story.append(Spacer(1, 12))
+    
+    # Detailed Findings
+    story.append(Paragraph("<b>Detailed Findings</b>", styles['Heading2']))
+    
+    # Penalty Details
+    if analysis_results['penalty_details']:
+        story.append(Paragraph("<b>Penalty Clauses</b>", styles['Heading3']))
+        story.append(Paragraph(analysis_results['penalty_details'], styles['Normal']))
+        story.append(Spacer(1, 12))
+    
+    # Obligation Details
+    if analysis_results['obligation_details']:
+        story.append(Paragraph("<b>Obligation Clauses</b>", styles['Heading3']))
+        story.append(Paragraph(analysis_results['obligation_details'], styles['Normal']))
+        story.append(Spacer(1, 12))
+    
+    # Delay Details
+    if analysis_results['delay_details']:
+        story.append(Paragraph("<b>Delay Clauses</b>", styles['Heading3']))
+        story.append(Paragraph(analysis_results['delay_details'], styles['Normal']))
+        story.append(Spacer(1, 12))
+    
+    # Sentiment Analysis
+    story.append(Paragraph("<b>Sentiment Analysis</b>", styles['Heading2']))
+    sentiment_text = "Positive" if analysis_results['sentiment_score'] > 0.6 else "Negative" if analysis_results['sentiment_score'] < 0.4 else "Neutral"
+    story.append(Paragraph(f"Sentiment Score: {analysis_results['sentiment_score']:.2f} ({sentiment_text})", styles['Normal']))
+    
+    doc.build(story)
+    buffer.seek(0)
+    return buffer
+
 def analyze_pdf(file_obj) -> List:
     """Main analysis function for Gradio interface"""
     try:
@@ -670,7 +757,7 @@ def analyze_pdf(file_obj) -> List:
             raise Exception(f"Risk score calculation failed: {str(e)}")
 
         try:
-            heatmap = generate_heatmap(risk_level)
+            heatmap_buffer = generate_heatmap(risk_level)
             risk_meter = generate_risk_meter(risk_score)
             sentiment_meter = generate_sentiment_meter(sentiment_score)
         except Exception as e:
@@ -776,6 +863,21 @@ def analyze_pdf(file_obj) -> List:
             "High": "This contract is high risk! Immediate legal review required."
         }
 
+        risk_summary_html = f"""
+        <div class='result-box'>
+            {sf_link_html}
+            <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>
+        """
+
         # Generate keyword matches section with highlighted context
         def format_keyword_matches(contexts_dict, title):
             sections = []
@@ -799,7 +901,7 @@ def analyze_pdf(file_obj) -> List:
             </div>
             """
 
-        keyword_matches = f"""
+        keyword_matches_html = f"""
         <div class="result-box">
             <div class="section-title">🔍 Keyword Matches in Document</div>
             {format_keyword_matches(penalty_contexts, "💰 Penalty Terms")}
@@ -808,29 +910,35 @@ def analyze_pdf(file_obj) -> List:
         </div>
         """
 
+        # Prepare data for PDF report
+        analysis_results = {
+            'risk_score': risk_score,
+            'risk_level': risk_level,
+            'penalty_count': total_penalties,
+            'obligation_count': total_obligations,
+            'delay_count': total_delays,
+            'penalty_values': penalty_values,
+            'penalty_details': "Penalty clauses found in the document",
+            'obligation_details': "Obligation clauses found in the document",
+            'delay_details': "Delay clauses found in the document",
+            'sentiment_score': sentiment_score
+        }
+
+        # Generate PDF report
+        pdf_buffer = generate_pdf_report(analysis_results, heatmap_buffer)
+        
+        # Return all outputs including the PDF data
         return [
-            f"""
-            <div class='result-box'>
-                {sf_link_html}
-                <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>
-            """,
-            "",  # Empty string for hidden risk visualization
+            risk_summary_html,
+            "",  # risk_visualization
             penalty_details,
             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'>📜 Text Extracted from PDF</div>{penalty_examples}</div>",
-            "",  # Empty string for hidden sentiment analysis
-            keyword_matches  # Add keyword matches as the last item
+            "",  # sentiment_analysis
+            keyword_matches_html,
+            pdf_buffer  # PDF data as the last output
         ]
     except Exception as e:
         logger.error(f"Analysis failed: {str(e)}")
@@ -846,7 +954,7 @@ def analyze_pdf(file_obj) -> List:
             </div>
         </div>
         """
-        return [error_message] * 9  # Update to match total outputs
+        return [error_message] * 10  # Update to match total outputs including PDF
 
 # 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:
@@ -895,8 +1003,11 @@ with gr.Blocks(css=css, title="PDF Contract Risk Analyzer", theme=gr.themes.Defa
     # Hidden sentiment analysis (kept in code but not displayed)
     sentiment_analysis = gr.HTML(label="Contract Sentiment Analysis", visible=False)
     
-    # New output for keyword matches
+    # Keyword matches output
     keyword_matches = gr.HTML(label="Keyword Matches in Document")
+    
+    # PDF download button
+    pdf_download = gr.File(label="Download PDF Report", visible=False)
    
     submit_btn.click(
         fn=analyze_pdf,
@@ -906,9 +1017,19 @@ with gr.Blocks(css=css, title="PDF Contract Risk Analyzer", theme=gr.themes.Defa
             penalty_count, penalty_amounts,
             obligation_count, delay_count,
             penalty_examples, sentiment_analysis,
-            keyword_matches  # Added new output
+            keyword_matches, pdf_download
         ]
     )
+    
+    # Show download button when PDF is ready
+    def toggle_download_visibility(pdf_data):
+        return gr.File.update(value=pdf_data, visible=pdf_data is not None)
+    
+    keyword_matches.change(
+        fn=toggle_download_visibility,
+        inputs=pdf_download,
+        outputs=pdf_download
+    )
 
 if __name__ == "__main__":
     demo.launch()