some UI changes, region selection, using CPU

analyzer.py

@@ -18,9 +18,16 @@ def extract_text_from_image(image):
         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()
+def check_compliance(text, regions=None):
+    """Check text for compliance across selected regions"""
+    all_rules = compliance_rules.get_all_rules()
+    
+    # Filter rules based on selected regions
+    if regions is not None:
+        rules = {region: rules for region, rules in all_rules.items() if regions.get(region, True)}
+    else:
+        rules = all_rules
+
     report = {
         "compliant": True,
         "violations": [],
@@ -128,13 +135,13 @@ def generate_html_report(compliance_report):
     html += '</div></div>'
     return html
 
-def analyze_ad_copy(image):
+def analyze_ad_copy(image, regions=None):
     """Main function to analyze ad copy"""
     # Extract text from image
     text = extract_text_from_image(image)
     
     # Check compliance
-    compliance_report = check_compliance(text)
+    compliance_report = check_compliance(text, regions)
     
     # Generate HTML report
     return generate_html_report(compliance_report)

interface.py

@@ -26,6 +26,12 @@ def create_interface():
             color: #6c757d;
             text-align: center;
         }
+        .checkbox-group {
+            margin: 10px 0;
+            padding: 10px;
+            background-color: #f8f9fa;
+            border-radius: 5px;
+        }
     """
 
     # Create a placeholder HTML
@@ -45,25 +51,53 @@ def create_interface():
     </div>
     """
 
-    demo = gr.Interface(
-        fn=analyze_ad_copy,
-        inputs=[
-            gr.Image(
+    def analyze_with_regions(image, us_sec, uk_fca, eu):
+        if not any([us_sec, uk_fca, eu]):
+            return """
+            <div class="report-container">
+                <div class="status non-compliant">⚠️ Please select at least one region for compliance checking.</div>
+            </div>
+            """
+        return analyze_ad_copy(image, {"US_SEC": us_sec, "UK_FCA": uk_fca, "EU": eu})
+
+    def reset_interface():
+        return None, True, True, True, placeholder_html
+
+    with gr.Blocks(theme=gr.themes.Default(), css=css) as demo:
+        gr.Markdown("# Marketing Campaign Financial Compliance Checker")
+        gr.Markdown("Upload marketing material to check compliance with financial regulations.")
+        
+        with gr.Row():
+            image_input = gr.Image(
                 type="numpy",
                 label="Upload Marketing Material",
                 height=300,
                 width=400
             )
-        ],
-        outputs=[
-            gr.HTML(
-                label="Compliance Report",
-                value=placeholder_html  # Set the initial placeholder
-            )
-        ],
-        title="Marketing Campaign Compliance Checker",
-        description="Upload marketing material to check compliance with US (SEC), UK (FCA), and EU financial regulations.",
-        css=css,
-        theme=gr.themes.Default()
-    )
+
+        with gr.Row(elem_classes="checkbox-group"):
+            us_sec = gr.Checkbox(value=True, label="US - SEC", interactive=True)
+            uk_fca = gr.Checkbox(value=True, label="UK - FCA", interactive=True)
+            eu = gr.Checkbox(value=True, label="EU", interactive=True)
+
+        output = gr.HTML(value=placeholder_html, label="Compliance Report")
+
+        with gr.Row():
+            clear_btn = gr.ClearButton([image_input, us_sec, uk_fca, eu, output])
+            submit_btn = gr.Button("Submit")
+
+        # Set up the submit action
+        submit_btn.click(
+            fn=analyze_with_regions,
+            inputs=[image_input, us_sec, uk_fca, eu],
+            outputs=output
+        )
+
+        # Set up the clear action to reset to defaults
+        clear_btn.click(
+            fn=reset_interface,
+            inputs=[],
+            outputs=[image_input, us_sec, uk_fca, eu, output]
+        )
+
     return demo

requirements.txt

@@ -1,3 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cu113
 torch>=2.0.0
 torchvision
 gradio>=4.8.0