Update app.py

app.py

@@ -59,7 +59,7 @@ def mock_ai_model(image):
     return milestone, completion_percent, confidence_score
 
 # Image processing and Salesforce upload
-def process_image(image, project_name):
+def process_image(image, project_name, sobject_id):
     try:
         if image is None:
             return "Error: Please upload an image to proceed.", "Pending", "", "", 0
@@ -89,7 +89,7 @@ def process_image(image, project_name):
                 'Title': image_filename,
                 'PathOnClient': image_filename,
                 'VersionData': encoded_image,  # Using the base64-encoded image data
-                'FirstPublishLocationId': 'your_sobject_id',  # Replace with the appropriate Salesforce object ID (e.g., Project or Construction)
+                'FirstPublishLocationId': sobject_id,  # Use the actual Salesforce object ID here
             }
             content_version = sf.ContentVersion.create(content_version_data)
 
@@ -134,6 +134,7 @@ with gr.Blocks(css=".gradio-container {background-color: #f0f4f8; font-family: A
     with gr.Row():
         image_input = gr.Image(type="filepath", label="Upload Construction Site Photo (JPG/PNG, ≤ 20MB)")
         project_name_input = gr.Textbox(label="Project Name (Required)", placeholder="e.g. Project_12345")
+        sobject_id_input = gr.Textbox(label="Salesforce Record ID (Required)", placeholder="Enter Salesforce Record ID here")
 
     submit_button = gr.Button("Process Image")
     output_text = gr.Textbox(label="Result")
@@ -144,7 +145,7 @@ with gr.Blocks(css=".gradio-container {background-color: #f0f4f8; font-family: A
 
     submit_button.click(
         fn=process_image,
-        inputs=[image_input, project_name_input],
+        inputs=[image_input, project_name_input, sobject_id_input],
         outputs=[output_text, upload_status, milestone, confidence, progress]
     )