Update app.py

app.py

@@ -7,6 +7,10 @@ import base64
 from io import BytesIO
 import json
 from datetime import datetime
+import logging
+
+# Setup logging
+logging.basicConfig(level=logging.INFO)
 
 # Salesforce Credentials
 SALESFORCE_USERNAME = "drone@sathkrutha.com"
@@ -25,7 +29,9 @@ try:
         security_token=SALESFORCE_SECURITY_TOKEN,
         instance_url=SALESFORCE_INSTANCE_URL
     )
+    logging.info("Salesforce connection established.")
 except Exception as e:
+    logging.error(f"Failed to connect to Salesforce: {str(e)}")
     raise Exception(f"Failed to connect to Salesforce: {str(e)}")
 
 # Load Model
@@ -86,8 +92,10 @@ def upload_image_to_salesforce(image, filename="detected_image.jpg", record_id=N
             "VersionData": img_data,
             "FirstPublishLocationId": record_id if record_id else None
         })
+        logging.info(f"Image uploaded to Salesforce with ContentVersion ID: {content_version['id']}")
         return content_version["id"]
     except Exception as e:
+        logging.error(f"Failed to upload image to Salesforce: {str(e)}")
         raise Exception(f"Failed to upload image to Salesforce: {str(e)}")
 
 # Detect defects and integrate with Salesforce
@@ -162,6 +170,7 @@ def detect_defects(image):
         return result_image, output
 
     except Exception as e:
+        logging.error(f"Processing failed: {str(e)}")
         return None, {"error": f"Processing failed: {str(e)}"}
 
 # Gradio Interface
@@ -177,4 +186,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()