roll back app

app.py

@@ -27,11 +27,10 @@ async def initialize_runner():
         config = RunnerConfig.from_dict({
             "environment": {
                 "name": "webbrowser",
-                # Set homepage to Salesforce's generic login URL to avoid premature waits for target page elements.
-                "homepage": "https://login.salesforce.com/", 
-                "headless": True, # Keep this False for local testing
+                "homepage": "https://dwd000006jia1mae.lightning.force.com/lightning/setup/AccountForecastSettings/home",
+                "headless": True,
                 "launch_args": ["--no-sandbox", "--disable-setuid-sandbox"],
-                "screenshot_delay": 0.5, # Reduced for faster debugging cycles
+                "screenshot_delay": 3.0,
                 "include_html": True,
                 "include_poi_text": True,
             },
@@ -61,9 +60,20 @@ async def initialize_runner():
         logger.info("Proxy-lite Runner initialized successfully.")
     return _runner
 
+# --- MODIFIED run_async_task FUNCTION (SIMPLIFIED) ---
+# This function is no longer needed in most cases with gevent.monkey.patch_all(asyncio=True)
+# but if you must call async functions from sync context, you simply await them.
+# However, you are already in an async function context within Flask routes when using Gunicorn/gevent.
+# The Gunicorn worker itself implicitly runs an event loop.
+# Let's remove the run_until_complete part.
+
+# DELETED: def run_async_task(coro): ...
+
+# --- END MODIFIED run_async_task FUNCTION ---
+
 
 @app.route('/run_proxy_task', methods=['POST'])
-async def run_proxy_task_endpoint():
+async def run_proxy_task_endpoint(): # <--- MAKE THIS FUNCTION ASYNC
     data = request.json
     request_task_instruction = data.get('task')
 
@@ -80,40 +90,27 @@ async def run_proxy_task_endpoint():
         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
 
-    # Define the specific Account Forecast Settings URL
-    account_forecast_url = "https://dwd000006jia1mae.lightning.force.com/lightning/setup/AccountForecastSettings/home"
-
-    # Define the tool code block to open a new tab and navigate after login
-    # Using a raw f-string for multiline tool code block
-    tool_code_block_new_tab = fr"""
-<tool_code>
-await browser.open_new_tab_and_go_to(url='{account_forecast_url}')
-</tool_code>
-"""
-
-    # Refined agent_task instruction to be sequential and robust to Salesforce redirects
-    agent_task = f"""
-    **Task Instructions for Proxy Lite Agent:**
-    1.  **Start on Login Page:** Navigate to the Salesforce login page.
-    2.  **Perform Login:** Log in to Salesforce using the provided username '{salesforce_username}' and password '{salesforce_password}'. Ensure all login fields are filled and the 'Log In' button is clicked.
-    3.  **Handle Post-Login Redirect:** After clicking the 'Log In' button:
-        * Observe the current URL. If the URL has changed from the initial login domain (e.g., from `login.salesforce.com` or `my.salesforce.com`) **immediately execute the following tool code block to open a new tab and navigate directly to the Account Forecast Settings page (`{account_forecast_url}`) to bypass any persistent loading issues or internal redirects:**
-        {tool_code_block_new_tab.strip()}
-    4.  **Confirm Target Page Load:** After successfully navigating to '{account_forecast_url}' (either directly after login or via the new tab strategy), ensure the page is fully loaded and stable. This means no loading spinners should be visible, and the main content for 'Account Forecast Settings' (like a clear heading, relevant toggles, or data tables) should be present and interactive.
-    5.  **Execute Main Task:** Once the Account Forecast Settings page is confirmed loaded and stable, proceed with the original user request: {request_task_instruction}.
-    6.  **Report Final Status:** Report the final status of the requested action, confirming both successful login and complete page load of the Account Forecast Settings.
-    """
+    agent_task = (
+        f"Log in to Salesforce. The username is '{salesforce_username}' and the password is '{salesforce_password}'. "
+        f"After attempting to log in, observe the page carefully. "
+        f"If the login was successful, the URL should change from the login page, and you should see elements indicating a logged-in state (e.g., a Salesforce navigation menu, a home screen, or a profile icon), rather than a login form or an error message. "
+        f"If the login is successful, {request_task_instruction}. "
+        f"Report the final status of the requested action and confirmation of successful login."
+    )
     
-    logger.info(f"Executing agent task (truncated for log): '{agent_task[:500]}...'")
+    logger.info(f"Executing agent task: '{agent_task[:200]}...'")
 
     try:
+        # Since run_proxy_task_endpoint is now async, you can directly await
         runner = await initialize_runner()
-        result = await runner.run(agent_task)
+        result = await runner.run(agent_task) # <--- AWAIT DIRECTLY
 
-        logger.info(f"Proxy-lite task completed. Output (truncated for log): {result[:500]}...")
+        logger.info(f"Proxy-lite task completed. Output: {result[:200]}...")
         return jsonify({"output": result})
     except Exception as e:
         logger.exception(f"Error processing Salesforce task: {e}")
+        # The RuntimeWarning: coroutine 'initialize_runner' was never awaited will disappear
+        # because initialize_runner is now awaited.
         return jsonify({"error": f"An error occurred: {str(e)}. Check logs for details."}), 500
 
 @app.route('/')
@@ -122,12 +119,8 @@ def root():
     return "Proxy-lite API is running. Send POST requests to /run_proxy_task with a 'task' in JSON body."
 
 if __name__ == '__main__':
-    # It is crucial to set HF_API_TOKEN as an environment variable (e.g., in a .env file or directly)
-    # for local testing as well, otherwise initialize_runner will fail.
     if not os.environ.get("HF_API_TOKEN"):
         logger.error("HF_API_TOKEN environment variable is not set. Please set it for local testing.")
-        # Removed exit(1) to allow the Flask app to start for basic connectivity checks,
-        # but runner initialization will still fail if token is missing.
-        # For full functionality, the token is essential.
+        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)