Update app.py

app.py

@@ -12,7 +12,6 @@ load_dotenv()
 SF_USERNAME = os.getenv("SF_USERNAME")
 SF_PASSWORD = os.getenv("SF_PASSWORD")
 SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
-PUBLIC_URL_BASE = os.getenv("PUBLIC_URL_BASE", "https://your-public-host.com/public_uploads")
 
 # Validate Salesforce credentials
 if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN]):
@@ -56,7 +55,7 @@ def process_image(image, project_name):
         if not str(image).lower().endswith(('.jpg', '.jpeg', '.png')):
             return "Error: Only JPG/PNG images are supported.", "Failure", "", "", 0
 
-        # Save image to public folder
+        # Save image to public folder for URL generation
         upload_dir = "public_uploads"
         os.makedirs(upload_dir, exist_ok=True)
         unique_id = datetime.now().strftime("%Y%m%d%H%M%S")
@@ -64,23 +63,21 @@ def process_image(image, project_name):
         saved_image_path = os.path.join(upload_dir, image_filename)
         shutil.copy(image, saved_image_path)
 
-        # Publicly accessible URL (replace base with deployed static site URL)
-        image_url = f"{PUBLIC_URL_BASE}/{image_filename}"
+        # Create public URL assuming you're serving /public_uploads/ via static web server (e.g., on localhost or external host)
+        public_url_base = os.getenv("PUBLIC_URL_BASE", "http://localhost:7860/public_uploads")
+        image_url = f"{public_url_base}/{image_filename}"
 
-        # AI Prediction
+        # Predict
         milestone, percent_complete, confidence_score = mock_ai_model(img)
 
-        # Format timestamp
-        formatted_timestamp = datetime.now().strftime("%d/%m/%Y, %I:%M %p")
-
-        # Salesforce record
+        # Construct Salesforce record
         record = {
             "Name__c": project_name,
             "Current_Milestone__c": milestone,
             "Completion_Percentage__c": percent_complete,
-            "Last_Updated_On__c": formatted_timestamp,
+            "Last_Updated_On__c": datetime.now().isoformat(),
             "Upload_Status__c": "Success",
-            "Comments__c": f"AI Prediction: {milestone} with {confidence_score * 100:.1f}% confidence",
+            "Comments__c": f"AI Prediction: {milestone} with {confidence_score*100}% confidence",
             "Last_Updated_Image__c": image_url
         }