Update app.py

app.py

@@ -17,10 +17,11 @@ SF_USERNAME = os.getenv("SF_USERNAME")
 SF_PASSWORD = os.getenv("SF_PASSWORD")
 SF_SECURITY_TOKEN = os.getenv("SF_SECURITY_TOKEN")
 
+API_KEY = os.getenv("API_KEY", "your-api-key-here")
 
 # Validate Salesforce credentials
-if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN )
-    raise ValueError("Missing Salesforce credentials. Set SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN in environment variables.")
+if not all([SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN ]):
+    raise ValueError("Missing Salesforce credentials. Set SF_USERNAME, SF_PASSWORD, SF_SECURITY_TOKEN  in environment variables.")
 
 # Initialize Salesforce connection
 try:
@@ -28,7 +29,7 @@ try:
         username=SF_USERNAME,
         password=SF_PASSWORD,
         security_token=SF_SECURITY_TOKEN,
-       
+        
         domain='login'  # Use 'test' for sandbox
     )
 except Exception as e:
@@ -152,4 +153,33 @@ def process_image(image, project_name):
             return f"Error: Failed to update Salesforce - {str(e)}", "Failure", "", "", 0
 
         return (
-            f
+            f"Success: Milestone: {milestone}, Completion: {percent_complete}%",
+            "Success",
+            milestone,
+            f"Confidence Score: {confidence_score}",
+            percent_complete
+        )
+
+    except Exception as e:
+        return f"Error: {str(e)}", "Failure", "", "", 0
+
+# Gradio interface for testing
+with gr.Blocks(css=".gradio-container {background-color: #f0f4f8; font-family: Arial;} .title {color: #2c3e50; font-size: 24px; text-align: center;}") as demo:
+    gr.Markdown("<h1 class='title'>Construction Milestone Detector</h1>")
+    project_name = gr.Textbox(label="Project Name", placeholder="Enter project name (e.g., My House or Site-123)")
+    image_input = gr.Image(type="filepath", label="Upload Construction Site Photo (JPG/PNG, ≤ 20MB)")
+    submit_button = gr.Button("Process Image")
+    output_text = gr.Textbox(label="Result")
+    upload_status = gr.Textbox(label="Upload Status")
+    milestone = gr.Textbox(label="Detected Milestone")
+    confidence = gr.Textbox(label="Confidence Score")
+    progress = gr.Slider(0, 100, label="Completion Percentage", interactive=False, value=0)
+
+    submit_button.click(
+        fn=process_image,
+        inputs=[image_input, project_name],
+        outputs=[output_text, upload_status, milestone, confidence, progress]
+    )
+
+# Launch the Gradio app
+demo.launch()