logging in

app.py

@@ -18,16 +18,21 @@ async def initialize_runner():
     global _runner
     if _runner is None:
         logger.info("Initializing Proxy-lite Runner...")
+        
+        # Retrieve Hugging Face API token from environment variables
         hf_api_token = os.environ.get("HF_API_TOKEN")
         if not hf_api_token:
             logger.error("HF_API_TOKEN environment variable not set. Cannot initialize Runner.")
             raise ValueError("HF_API_TOKEN environment variable not set. Please set it as a Space secret.")
 
+        # Define RunnerConfig
         config = RunnerConfig.from_dict({
             "environment": {
                 "name": "webbrowser",
+                # The homepage is already set to the Account Forecast Settings page.
+                # The agent will start here, but still needs to handle login if not already logged in.
                 "homepage": "https://dwd000006jia1mae.lightning.force.com/lightning/setup/AccountForecastSettings/home",
-                "headless": True,
+                "headless": True, # Keep headless for production environments
                 "launch_args": ["--no-sandbox", "--disable-setuid-sandbox"] 
             },
             "solver": {
@@ -69,20 +74,43 @@ def run_async_task(coro):
 @app.route('/run_proxy_task', methods=['POST'])
 def run_proxy_task_endpoint():
     data = request.json
-    task = data.get('task')
-    if not task:
+    # The 'task' from the user request will now be the dynamic instruction
+    request_task_instruction = data.get('task')
+
+    if not request_task_instruction:
         logger.warning("Received request without 'task' field. Returning 400.")
         return jsonify({"error": "No 'task' provided in request body"}), 400
 
-    logger.info(f"Received task for proxy-lite: '{task}'")
+    logger.info(f"Received user request task: '{request_task_instruction}'")
+
+    # Retrieve Salesforce credentials from environment variables (Hugging Face Space secrets)
+    salesforce_username = os.environ.get("SALESFORCE_USERNAME")
+    salesforce_password = os.environ.get("SALESFORCE_PASSWORD")
+
+    if not salesforce_username or not salesforce_password:
+        logger.error("Salesforce credentials (SALESFORCE_USERNAME, SALESFORCE_PASSWORD) environment variables not set.")
+        return jsonify({"error": "Salesforce credentials not configured. Please set SALESFORCE_USERNAME and SALESFORCE_PASSWORD as Space secrets."}), 500
+
+    # Construct the full task for the proxy-lite agent,
+    # combining login instructions with the dynamic task from the user.
+    # The agent is expected to handle the login flow if needed.
+    agent_task = (
+        f"Log in to Salesforce. The username is '{salesforce_username}' and the password is '{salesforce_password}'. "
+        f"Once logged in, {request_task_instruction}. "
+        f"Report the final status of the requested action."
+    )
+    
+    logger.info(f"Executing agent task: '{agent_task[:200]}...'")
+
     try:
         runner = run_async_task(initialize_runner())
-        result = run_async_task(runner.run(task))
+        # Pass the constructed agent_task to the runner
+        result = run_async_task(runner.run(agent_task))
 
         logger.info(f"Proxy-lite task completed. Output: {result[:200]}...")
         return jsonify({"output": result})
     except Exception as e:
-        logger.exception(f"Error processing task '{task}':")
+        logger.exception(f"Error processing Salesforce task: {e}")
         return jsonify({"error": f"An error occurred: {str(e)}. Check logs for details."}), 500
 
 @app.route('/')
@@ -95,4 +123,4 @@ if __name__ == '__main__':
         logger.error("HF_API_TOKEN environment variable is not set. Please set it for local testing.")
         exit(1)
     logger.info("Starting Flask development server on 0.0.0.0:7860...")
-    app.run(host='0.0.0.0', port=7860, debug=True)
+    app.run(host='0.0.0.0', port=7860, debug=True)