made some UI changes, stil in CPU

analyzer.py

@@ -0,0 +1,140 @@
+import easyocr
+import torch
+import numpy as np
+from compliance_rules import ComplianceRules
+
+# Initialize OCR reader
+reader = easyocr.Reader(['en'], gpu=torch.cuda.is_available())
+
+# Initialize compliance rules
+compliance_rules = ComplianceRules()
+
+def extract_text_from_image(image):
+    """Extract text from image using EasyOCR"""
+    try:
+        result = reader.readtext(np.array(image))
+        return " ".join([text[1] for text in result])
+    except Exception as e:
+        print(f"Error in text extraction: {str(e)}")
+        return "Error extracting text from image"
+
+def check_compliance(text):
+    """Check text for compliance across all regions"""
+    rules = compliance_rules.get_all_rules()
+    report = {
+        "compliant": True,
+        "violations": [],
+        "warnings": [],
+        "channel_risks": {
+            "email": {"score": 0, "details": []},
+            "social": {"score": 0, "details": []},
+            "print": {"score": 0, "details": []}
+        }
+    }
+    
+    for region, region_rules in rules.items():
+        # Check prohibited terms
+        for term_info in region_rules["prohibited_terms"]:
+            term = term_info["term"].lower()
+            if term in text.lower() or any(var.lower() in text.lower() for var in term_info["variations"]):
+                report["compliant"] = False
+                violation = f"{region}: Prohibited term '{term}' found"
+                report["violations"].append({
+                    "region": region,
+                    "type": "prohibited_term",
+                    "term": term,
+                    "severity": term_info["severity"]
+                })
+                
+                # Update channel risks
+                for channel in report["channel_risks"]:
+                    risk_score = compliance_rules.calculate_risk_score([violation], [], region)
+                    report["channel_risks"][channel]["score"] += risk_score
+                    report["channel_risks"][channel]["details"].append(
+                        f"Prohibited term '{term}' increases {channel} risk"
+                    )
+        
+        # Check required disclaimers
+        for disclaimer in region_rules["required_disclaimers"]:
+            disclaimer_found = any(
+                disc_text.lower() in text.lower() 
+                for disc_text in disclaimer["text"]
+            )
+            if not disclaimer_found:
+                warning = f"{region}: Missing {disclaimer['type']} disclaimer"
+                report["warnings"].append({
+                    "region": region,
+                    "type": "missing_disclaimer",
+                    "disclaimer_type": disclaimer["type"],
+                    "severity": disclaimer["severity"]
+                })
+                
+                # Update channel risks
+                for channel in report["channel_risks"]:
+                    risk_score = compliance_rules.calculate_risk_score([], [warning], region)
+                    report["channel_risks"][channel]["score"] += risk_score
+                    report["channel_risks"][channel]["details"].append(
+                        f"Missing {disclaimer['type']} disclaimer affects {channel} risk"
+                    )
+    
+    return report
+
+def format_severity(severity):
+    return f'<span class="severity-{severity}">{severity.upper()}</span>'
+
+def generate_html_report(compliance_report):
+    """Generate formatted HTML report"""
+    html = '<div class="report-container">'
+    
+    # Overall Status
+    status_class = "compliant" if compliance_report["compliant"] else "non-compliant"
+    status_icon = "✅" if compliance_report["compliant"] else "❌"
+    html += f'<div class="status {status_class}">{status_icon} Overall Status: {"Compliant" if compliance_report["compliant"] else "Non-Compliant"}</div>'
+    
+    # Violations
+    if compliance_report["violations"]:
+        html += '<div class="section">'
+        html += '<div class="section-title">🚫 Violations Found:</div>'
+        for violation in compliance_report["violations"]:
+            html += f'<div class="item">• {violation["region"]}: {violation["type"]} - \'{violation["term"]}\' (Severity: {format_severity(violation["severity"])})</div>'
+        html += '</div>'
+    
+    # Warnings
+    if compliance_report["warnings"]:
+        html += '<div class="section">'
+        html += '<div class="section-title">⚠️ Warnings:</div>'
+        for warning in compliance_report["warnings"]:
+            html += f'<div class="item">• {warning["region"]}: {warning["disclaimer_type"]} (Severity: {format_severity(warning["severity"])})</div>'
+        html += '</div>'
+    
+    # Channel Risk Assessment
+    html += '<div class="section">'
+    html += '<div class="section-title">📊 Channel Risk Assessment:</div>'
+    
+    for channel, risk_info in compliance_report["channel_risks"].items():
+        score = risk_info["score"]
+        risk_level = "low" if score < 3 else "medium" if score < 6 else "high"
+        
+        html += f'<div class="channel risk-{risk_level}">'
+        html += f'<strong>{channel.capitalize()}</strong>: {risk_level.upper()} Risk (Score: {score})'
+        
+        if risk_info["details"]:
+            html += '<div class="details">'
+            for detail in risk_info["details"]:
+                html += f'<div>• {detail}</div>'
+            html += '</div>'
+        html += '</div>'
+    
+    html += '</div></div>'
+    return html
+
+def analyze_ad_copy(image):
+    """Main function to analyze ad copy"""
+    # Extract text from image
+    text = extract_text_from_image(image)
+    
+    # Check compliance
+    compliance_report = check_compliance(text)
+    
+    # Generate HTML report
+    return generate_html_report(compliance_report)

app.py

@@ -1,149 +1,6 @@
-import gradio as gr
-import easyocr
-import torch
-from transformers import pipeline, DistilBertTokenizer, DistilBertForSequenceClassification
-import numpy as np
-from PIL import Image
-import json
-from compliance_rules import ComplianceRules
+from interface import create_interface
 
-# Print GPU information for debugging
-print(f"Is CUDA available: {torch.cuda.is_available()}")
-if torch.cuda.is_available():
-    print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
-else:
-    print("Running on CPU")
-
-# Initialize OCR reader
-reader = easyocr.Reader(['en'], gpu=torch.cuda.is_available())
-
-# Initialize compliance rules
-compliance_rules = ComplianceRules()
-
-def extract_text_from_image(image):
-    """Extract text from image using EasyOCR"""
-    try:
-        result = reader.readtext(np.array(image))
-        return " ".join([text[1] for text in result])
-    except Exception as e:
-        print(f"Error in text extraction: {str(e)}")
-        return "Error extracting text from image"
-
-def check_compliance(text):
-    """Check text for compliance across all regions"""
-    rules = compliance_rules.get_all_rules()
-    report = {
-        "compliant": True,
-        "violations": [],
-        "warnings": [],
-        "channel_risks": {
-            "email": {"score": 0, "details": []},
-            "social": {"score": 0, "details": []},
-            "print": {"score": 0, "details": []}
-        }
-    }
-    
-    for region, region_rules in rules.items():
-        # Check prohibited terms
-        for term_info in region_rules["prohibited_terms"]:
-            term = term_info["term"].lower()
-            if term in text.lower() or any(var.lower() in text.lower() for var in term_info["variations"]):
-                report["compliant"] = False
-                violation = f"{region}: Prohibited term '{term}' found"
-                report["violations"].append({
-                    "region": region,
-                    "type": "prohibited_term",
-                    "term": term,
-                    "severity": term_info["severity"]
-                })
-                
-                # Update channel risks
-                for channel in report["channel_risks"]:
-                    risk_score = compliance_rules.calculate_risk_score([violation], [], region)
-                    report["channel_risks"][channel]["score"] += risk_score
-                    report["channel_risks"][channel]["details"].append(
-                        f"Prohibited term '{term}' increases {channel} risk"
-                    )
-        
-        # Check required disclaimers
-        for disclaimer in region_rules["required_disclaimers"]:
-            disclaimer_found = any(
-                disc_text.lower() in text.lower() 
-                for disc_text in disclaimer["text"]
-            )
-            if not disclaimer_found:
-                warning = f"{region}: Missing {disclaimer['type']} disclaimer"
-                report["warnings"].append({
-                    "region": region,
-                    "type": "missing_disclaimer",
-                    "disclaimer_type": disclaimer["type"],
-                    "severity": disclaimer["severity"]
-                })
-                
-                # Update channel risks
-                for channel in report["channel_risks"]:
-                    risk_score = compliance_rules.calculate_risk_score([], [warning], region)
-                    report["channel_risks"][channel]["score"] += risk_score
-                    report["channel_risks"][channel]["details"].append(
-                        f"Missing {disclaimer['type']} disclaimer affects {channel} risk"
-                    )
-    
-    return report
-
-def analyze_ad_copy(image):
-    """Main function to analyze ad copy"""
-    # Extract text from image
-    text = extract_text_from_image(image)
-    
-    # Check compliance
-    compliance_report = check_compliance(text)
-    
-    # Generate readable report
-    report_text = "Compliance Analysis Report\n\n"
-    report_text += f"Overall Status: {'✅ Compliant' if compliance_report['compliant'] else '❌ Non-Compliant'}\n\n"
-    
-    if compliance_report["violations"]:
-        report_text += "Violations Found:\n"
-        for violation in compliance_report["violations"]:
-            report_text += f"• {violation['region']}: {violation['type']} - '{violation['term']}' (Severity: {violation['severity']})\n"
-        report_text += "\n"
-    
-    if compliance_report["warnings"]:
-        report_text += "Warnings:\n"
-        for warning in compliance_report["warnings"]:
-            report_text += f"• {warning['region']}: {warning['disclaimer_type']} (Severity: {warning['severity']})\n"
-        report_text += "\n"
-    
-    report_text += "Channel Risk Assessment:\n"
-    for channel, risk_info in compliance_report["channel_risks"].items():
-        score = risk_info["score"]
-        risk_level = "Low" if score < 3 else "Medium" if score < 6 else "High"
-        report_text += f"• {channel.capitalize()}: {risk_level} Risk (Score: {score})\n"
-        if risk_info["details"]:
-            for detail in risk_info["details"]:
-                report_text += f"  - {detail}\n"
-    
-    return report_text
-
-# Create Gradio interface with updated parameters
-demo = gr.Interface(
-    fn=analyze_ad_copy,
-    inputs=gr.Image(
-        type="numpy",
-        label="Upload Marketing Material",
-        height=300,
-        width=400,
-    ),
-    outputs=gr.Textbox(
-        label="Compliance Report",
-        lines=10,
-        max_lines=20
-    ),
-    title="Marketing Campaign Compliance Checker",
-    description="Upload marketing material to check compliance with US (SEC), UK (FCA), and EU financial regulations.",
-    theme=gr.themes.Default()
-)
-
-# Launch the app
+# Create and launch the interface
 if __name__ == "__main__":
+    demo = create_interface()
     demo.launch()

interface.py

@@ -0,0 +1,37 @@
+import gradio as gr
+from analyzer import analyze_ad_copy
+
+def create_interface():
+    demo = gr.Interface(
+        fn=analyze_ad_copy,
+        inputs=gr.Image(
+            type="numpy",
+            label="Upload Marketing Material",
+            height=300,
+            width=400,
+        ),
+        outputs=gr.HTML(
+            label="Compliance Report",
+        ),
+        title="Marketing Campaign Compliance Checker",
+        description="Upload marketing material to check compliance with US (SEC), UK (FCA), and EU financial regulations.",
+        theme=gr.themes.Default(),
+        css="""
+            .report-container { font-family: 'Arial', sans-serif; padding: 20px; }
+            .status { font-size: 1.2em; margin-bottom: 20px; padding: 10px; border-radius: 5px; }
+            .compliant { background-color: #e7f5e7; color: #0d5f0d; }
+            .non-compliant { background-color: #fce8e8; color: #c41e3a; }
+            .section { margin: 15px 0; }
+            .section-title { font-weight: bold; color: #2c3e50; margin: 10px 0; }
+            .item { margin: 5px 0 5px 20px; }
+            .severity-high { color: #c41e3a; }
+            .severity-medium { color: #f39c12; }
+            .severity-low { color: #27ae60; }
+            .risk-high { background-color: #fce8e8; }
+            .risk-medium { background-color: #fff3cd; }
+            .risk-low { background-color: #e7f5e7; }
+            .channel { margin: 10px 0; padding: 10px; border-radius: 5px; }
+            .details { margin-left: 20px; font-size: 0.9em; color: #555; }
+        """
+    )
+    return demo