Update app.py

app.py

@@ -132,154 +132,6 @@ def calculate_iou(box1, box2):
     
     return intersection / union if union > 0 else 0
 
-# ==========================
-# Salesforce Integration (unchanged)
-# ==========================
-@retry(stop_max_attempt_number=3, wait_fixed=2000)
-def connect_to_salesforce():
-    try:
-        sf = Salesforce(**CONFIG["SF_CREDENTIALS"])
-        logger.info("Connected to Salesforce")
-        sf.describe()
-        return sf
-    except Exception as e:
-        logger.error(f"Salesforce connection failed: {e}")
-        raise
-
-def generate_violation_pdf(violations, score):
-    try:
-        pdf_filename = f"violations_{int(time.time())}.pdf"
-        pdf_path = os.path.join(CONFIG["OUTPUT_DIR"], pdf_filename)
-        pdf_file = BytesIO()
-        c = canvas.Canvas(pdf_file, pagesize=letter)
-        c.setFont("Helvetica", 12)
-        c.drawString(1 * inch, 10 * inch, "Worksite Safety Violation Report")
-        c.setFont("Helvetica", 10)
-
-        y_position = 9.5 * inch
-        report_data = {
-            "Compliance Score": f"{score}%",
-            "Violations Found": len(violations),
-            "Timestamp": time.strftime("%Y-%m-%d %H:%M:%S")
-        }
-        for key, value in report_data.items():
-            c.drawString(1 * inch, y_position, f"{key}: {value}")
-            y_position -= 0.3 * inch
-
-        y_position -= 0.3 * inch
-        c.drawString(1 * inch, y_position, "Violation Details:")
-        y_position -= 0.3 * inch
-        if not violations:
-            c.drawString(1 * inch, y_position, "No violations detected.")
-        else:
-            for v in violations:
-                display_name = CONFIG["DISPLAY_NAMES"].get(v["violation"], v["violation"])
-                text = f"{display_name} at {v['timestamp']:.2f}s (Confidence: {v['confidence']})"
-                c.drawString(1 * inch, y_position, text)
-                y_position -= 0.3 * inch
-                if y_position < 1 * inch:
-                    c.showPage()
-                    c.setFont("Helvetica", 10)
-                    y_position = 10 * inch
-
-        c.showPage()
-        c.save()
-        pdf_file.seek(0)
-
-        with open(pdf_path, "wb") as f:
-            f.write(pdf_file.getvalue())
-        public_url = f"{CONFIG['PUBLIC_URL_BASE']}{pdf_filename}"
-        logger.info(f"PDF generated: {public_url}")
-        return pdf_path, public_url, pdf_file
-    except Exception as e:
-        logger.error(f"Error generating PDF: {e}")
-        return "", "", None
-
-def upload_pdf_to_salesforce(sf, pdf_file, report_id):
-    try:
-        if not pdf_file:
-            logger.error("No PDF file provided for upload")
-            return ""
-        encoded_pdf = base64.b64encode(pdf_file.getvalue()).decode('utf-8')
-        content_version_data = {
-            "Title": f"Safety_Violation_Report_{int(time.time())}",
-            "PathOnClient": f"safety_violation_{int(time.time())}.pdf",
-            "VersionData": encoded_pdf,
-            "FirstPublishLocationId": report_id
-        }
-        content_version = sf.ContentVersion.create(content_version_data)
-        result = sf.query(f"SELECT Id, ContentDocumentId FROM ContentVersion WHERE Id = '{content_version['id']}'")
-        if not result['records']:
-            logger.error("Failed to retrieve ContentVersion")
-            return ""
-        file_url = f"https://{sf.sf_instance}/sfc/servlet.shepherd/version/download/{content_version['id']}"
-        logger.info(f"PDF uploaded to Salesforce: {file_url}")
-        return file_url
-    except Exception as e:
-        logger.error(f"Error uploading PDF to Salesforce: {e}")
-        return ""
-
-def push_report_to_salesforce(violations, score, pdf_path, pdf_file):
-    try:
-        sf = connect_to_salesforce()
-        violations_text = "\n".join(
-            f"{CONFIG['DISPLAY_NAMES'].get(v['violation'], v['violation'])} at {v['timestamp']:.2f}s (Confidence: {v['confidence']})"
-            for v in violations
-        ) or "No violations detected."
-        pdf_url = f"{CONFIG['PUBLIC_URL_BASE']}{os.path.basename(pdf_path)}" if pdf_path else ""
-
-        record_data = {
-            "Compliance_Score__c": score,
-            "Violations_Found__c": len(violations),
-            "Violations_Details__c": violations_text,
-            "Status__c": "Pending",
-            "PDF_Report_URL__c": pdf_url
-        }
-        logger.info(f"Creating Salesforce record with data: {record_data}")
-        try:
-            record = sf.Safety_Video_Report__c.create(record_data)
-            logger.info(f"Created Safety_Video_Report__c record: {record['id']}")
-        except Exception as e:
-            logger.error(f"Failed to create Safety_Video_Report__c: {e}")
-            record = sf.Account.create({"Name": f"Safety_Report_{int(time.time())}"})
-            logger.warning(f"Fell back to Account record: {record['id']}")
-        record_id = record["id"]
-
-        if pdf_file:
-            uploaded_url = upload_pdf_to_salesforce(sf, pdf_file, record_id)
-            if uploaded_url:
-                try:
-                    sf.Safety_Video_Report__c.update(record_id, {"PDF_Report_URL__c": uploaded_url})
-                    logger.info(f"Updated record {record_id} with PDF URL: {uploaded_url}")
-                except Exception as e:
-                    logger.error(f"Failed to update Safety_Video_Report__c: {e}")
-                    sf.Account.update(record_id, {"Description": uploaded_url})
-                    logger.info(f"Updated Account record {record_id} with PDF URL")
-                pdf_url = uploaded_url
-
-        return record_id, pdf_url
-    except Exception as e:
-        logger.error(f"Salesforce record creation failed: {e}", exc_info=True)
-        return None, ""
-
-def calculate_safety_score(violations):
-    penalties = {
-        "no_helmet": 25,
-        "no_harness": 30,
-        "unsafe_posture": 20,
-        "unsafe_zone": 35,
-        "improper_tool_use": 25
-    }
-    # Count unique violations per worker
-    unique_violations = set()
-    for v in violations:
-        key = (v["worker_id"], v["violation"])
-        unique_violations.add(key)
-    
-    total_penalty = sum(penalties.get(violation, 0) for _, violation in unique_violations)
-    score = 100 - total_penalty
-    return max(score, 0)
-
 # ==========================
 # Enhanced Video Processing
 # ==========================
@@ -531,4 +383,4 @@ interface = gr.Interface(
 
 if __name__ == "__main__":
     logger.info("Launching Enhanced Safety Analyzer App...")
-    interface.launch()
+    interface.launch()