Update app.py

app.py

@@ -8,7 +8,7 @@ import logging
 from datetime import datetime
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
-import gradio as gr  # Add this import here
+import gradio as gr
 
 # Setup logging
 logging.basicConfig(level=logging.INFO)
@@ -44,7 +44,6 @@ transform = transforms.Compose([
     transforms.ToTensor(),
 ])
 
-# Define valid picklist values for Fault_Type__c and Severity__c
 VALID_FAULT_TYPES = ["Crack", "Rust", "Spalling", "Deformation", "Corrosion"]
 VALID_SEVERITIES = ["Minor", "Moderate", "Critical"]
 
@@ -57,7 +56,6 @@ def get_severity(score):
         return "Minor"
 
 def map_defect_type():
-    # Always return first valid fault type (customize if needed)
     return VALID_FAULT_TYPES[0]
 
 def upload_image_to_salesforce(image, filename="detected_image.jpg", record_id=None):
@@ -78,7 +76,6 @@ def upload_image_to_salesforce(image, filename="detected_image.jpg", record_id=N
         logging.error(f"Failed to upload image to Salesforce: {str(e)}")
         raise Exception(f"Failed to upload image to Salesforce: {str(e)}")
 
-# New function: generate PDF report bytes from defect list
 def create_pdf_report(defect_list):
     buffer = BytesIO()
     c = canvas.Canvas(buffer, pagesize=letter)
@@ -103,7 +100,6 @@ def create_pdf_report(defect_list):
     buffer.close()
     return pdf
 
-# New function: upload PDF bytes to Salesforce ContentVersion
 def upload_pdf_to_salesforce(pdf_bytes, filename="report.pdf", record_id=None):
     try:
         pdf_data = base64.b64encode(pdf_bytes).decode("utf-8")
@@ -219,4 +215,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()   # without share=True
+    demo.launch()  # no share=True, so no built-in share button