Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
| 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().
|
| 79 |
"Upload_Status__c": "Success",
|
| 80 |
-
"Comments__c": f"AI Prediction: {milestone} with {confidence_score*100
|
| 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 |
|