Update app.py

app.py

@@ -11,9 +11,9 @@ logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(
 logger = logging.getLogger(__name__)
 
 # Salesforce credentials (use environment variables for security)
-SF_USERNAME = os.getenv('[email protected]')
-SF_PASSWORD = os.getenv('Teja90325@')
-SF_SECURITY_TOKEN = os.getenv('clceSdBgQ30Rx9BSC66gAcRx')
+SF_USERNAME = os.getenv('SF_USERNAME')
+SF_PASSWORD = os.getenv('SF_PASSWORD')
+SF_SECURITY_TOKEN = os.getenv('SF_SECURITY_TOKEN')
 SF_DOMAIN = os.getenv('SF_DOMAIN', 'login')  # Default to 'login' if not set
 HUGGINGFACE_API_KEY = os.getenv('HUGGINGFACE_API_KEY')
 
@@ -184,7 +184,7 @@ def generate_outputs(role, project_id, milestones, reflection):
             quote = generated_text[quote_start:].strip()
     except Exception as e:
         logger.error(f"Error parsing output: {e}")
-       -return f"Error parsing output: {e}", "", ""
+        return f"Error parsing output: {e}", "", ""
 
     return checklist, suggestions, quote
 
@@ -227,6 +227,42 @@ def store_in_salesforce(project, role, project_id, milestones, reflection, check
         logger.error(f"Error storing data in Salesforce: {e}")
         return f"Error storing data in Salesforce: {e}"
 
+def fetch_and_process_salesforce_data():
+    """Fetch Project__c records from Salesforce and generate AI outputs."""
+    if sf is None:
+        logger.error("Salesforce connection not initialized")
+        return "Error: Salesforce connection not initialized"
+
+    try:
+        # Query Project__c records
+        query = """
+            SELECT Id, Name, Project_Name__c, Milestones__c, Weather_Log__c, 
+                   Supervisor__r.Role__c, Supervisor__r.Name, Supervisor__c
+            FROM Project__c
+            WHERE Milestones__c != null
+            LIMIT 10
+        """
+        projects = sf.query(query)['records']
+        logger.info(f"Fetched {len(projects)} Project__c records")
+
+        for project in projects:
+            role = project['Supervisor__r']['Role__c'] if project['Supervisor__r'] is not None else 'Supervisor'
+            project_id = project['Name']
+            milestones = project['Milestones__c'] or 'No milestones'
+            reflection = f"Weather: {project['Weather_Log__c'] or 'Sunny, 25°C'}"
+
+            # Generate AI outputs
+            checklist, suggestions, quote = generate_outputs(role, project_id, milestones, reflection)
+            logger.info(f"Generated outputs for Project {project_id}")
+
+            # Store in Salesforce
+            store_in_salesforce(project, role, project_id, milestones, reflection, checklist, suggestions, quote)
+
+        return f"Processed {len(projects)} projects successfully"
+    except Exception as e:
+        logger.error(f"Error processing Salesforce data: {e}")
+        return f"Error processing Salesforce data: {e}"
+
 def on_role_change(role):
     """Handle role selection change, fetch Salesforce data, generate outputs, and store in Salesforce."""
     # Fetch data from Salesforce based on the selected role
@@ -248,7 +284,7 @@ def create_interface():
         gr.Markdown("### Construction Supervisor AI Coach")
         with gr.Row():
             role = gr.Dropdown(choices=["Supervisor", "Foreman", "Project Manager"], label="Role", value="Supervisor")
-        project_id = gr.Textbox(label="Project ID", placeholder="e.g., PROJ-123")
+            project_id = gr.Textbox(label="Project ID", placeholder="e.g., PROJ-123")
         milestones = gr.Textbox(
             label="Milestones (comma-separated)",
             placeholder="e.g., Foundation complete, Framing started, Roof installed"
@@ -258,6 +294,10 @@ def create_interface():
             lines=3,
             placeholder="e.g., Facing delays due to weather and equipment issues."
         )
+        with gr.Row():
+            submit = gr.Button("Generate")
+            clear = gr.Button("Clear")
+            sf_button = gr.Button("Process Salesforce Data")
         checklist_output = gr.Textbox(label="Checklist", lines=4)
         suggestions_output = gr.Textbox(label="Suggestions", lines=4)
         quote_output = gr.Textbox(label="Quote", lines=2)
@@ -270,6 +310,22 @@ def create_interface():
             outputs=[checklist_output, suggestions_output, quote_output, sf_output, project_id, milestones, reflection]
         )
 
+        submit.click(
+            fn=generate_outputs,
+            inputs=[role, project_id, milestones, reflection],
+            outputs=[checklist_output, suggestions_output, quote_output]
+        )
+        clear.click(
+            fn=lambda: ("Supervisor", "", "", ""),
+            inputs=None,
+            outputs=[role, project_id, milestones, reflection]
+        )
+        sf_button.click(
+            fn=fetch_and_process_salesforce_data,
+            inputs=None,
+            outputs=sf_output
+        )
+    
     return demo
 
 if __name__ == "__main__":