Rekham1110 commited on
Commit
7b31474
·
verified ·
1 Parent(s): 44b0e58

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -12,7 +12,6 @@ load_dotenv()
12
  SF_USERNAME = os.getenv("SF_USERNAME")
13
  SF_PASSWORD = os.getenv("SF_PASSWORD")
14
  SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
15
- PUBLIC_URL_BASE = os.getenv("PUBLIC_URL_BASE", "https://your-username.github.io/your-repo/public_uploads")
16
 
17
  # Validate Salesforce credentials
18
  if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN]):
@@ -64,8 +63,9 @@ def process_image(image, project_name):
64
  saved_image_path = os.path.join(upload_dir, image_filename)
65
  shutil.copy(image, saved_image_path)
66
 
67
- # Create public URL
68
- image_url = f"{PUBLIC_URL_BASE}/{image_filename}"
 
69
 
70
  # Predict
71
  milestone, percent_complete, confidence_score = mock_ai_model(img)
@@ -75,9 +75,9 @@ def process_image(image, project_name):
75
  "Name__c": project_name,
76
  "Current_Milestone__c": milestone,
77
  "Completion_Percentage__c": percent_complete,
78
- "Last_Updated_On__c": datetime.now().strftime("%d/%m/%Y, %I:%M %p"),
79
  "Upload_Status__c": "Success",
80
- "Comments__c": f"AI Prediction: {milestone} with {confidence_score*100:.0f}% confidence",
81
  "Last_Updated_Image__c": image_url
82
  }
83
 
 
12
  SF_USERNAME = os.getenv("SF_USERNAME")
13
  SF_PASSWORD = os.getenv("SF_PASSWORD")
14
  SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
 
15
 
16
  # Validate Salesforce credentials
17
  if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN]):
 
63
  saved_image_path = os.path.join(upload_dir, image_filename)
64
  shutil.copy(image, saved_image_path)
65
 
66
+ # Create public URL assuming you're serving /public_uploads/ via static web server (e.g., on localhost or external host)
67
+ public_url_base = os.getenv("PUBLIC_URL_BASE", "http://localhost:7860/public_uploads")
68
+ image_url = f"{public_url_base}/{image_filename}"
69
 
70
  # Predict
71
  milestone, percent_complete, confidence_score = mock_ai_model(img)
 
75
  "Name__c": project_name,
76
  "Current_Milestone__c": milestone,
77
  "Completion_Percentage__c": percent_complete,
78
+ "Last_Updated_On__c": datetime.now().isoformat(),
79
  "Upload_Status__c": "Success",
80
+ "Comments__c": f"AI Prediction: {milestone} with {confidence_score*100}% confidence",
81
  "Last_Updated_Image__c": image_url
82
  }
83