run_virtual.sh

#!/bin/bash
# Mobile-Agent-v3 Virtual Runner
# Uses Gemini AI to generate virtual Android interfaces
#
# Usage:
#   ./run_virtual.sh                          # Run with default contact task

# Change to script directory
SCRIPT_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
cd "$SCRIPT_DIR"

# Load .env file
if [ -f .env ]; then
    echo "Loading configuration from .env..."
    set -a  # Auto-export all variables
    source .env
    set +a
else
    echo "Warning: .env file not found. Using environment variables or defaults."
fi

# Default task
TASKS="ContactsAddContact"
NUM_TASKS=1

current_time=$(date +"%Y-%m-%d_%H-%M-%S")
LOG="${LOG_PATH:-logs}/log_virtual_"$current_time".log"

# Agent LLM Configuration (use environment variables or defaults)
MODEL_NAME="${AGENT_NAME:-mobile_agent_v3}"
MODEL="${AGENT_MODEL:-GUI-Owl-7B}"
API_KEY="${AGENT_API_KEY:-EMPTY}"
BASE_URL="${AGENT_BASE_URL:-http://localhost:4243/v1}"

# Gemini Configuration (for virtual environment)
GEMINI_API_KEY="${GEMINI_API_KEY:-}"
GEMINI_BASE_URL="${GEMINI_BASE_URL:-https://api.dou.chat/v1}"
GEMINI_MODEL="${GEMINI_MODEL:-google/gemini-2.0-flash-exp:free}"

# Check required environment variables
if [ -z "$GEMINI_API_KEY" ]; then
    echo "Error: GEMINI_API_KEY environment variable is required"
    echo "Please set: export GEMINI_API_KEY='your-api-key'"
    exit 1
fi

# Output paths
TRAJ_OUTPUT_PATH="${TRAJ_PATH:-trajectories}/traj_virtual_"$current_time
INITIAL_IMAGE_PATH="${INITIAL_IMAGE:-start_Android.png}"

# Create output directories
mkdir -p logs trajectories

echo "=========================================="
echo "Mobile-Agent-v3 Virtual GUI Environment"
echo "=========================================="
echo "Time: $current_time"
echo "Task: $TASKS"
echo "Agent Model: $MODEL"
echo "Virtual Environment Model: $GEMINI_MODEL"
echo "Log File: $LOG"
echo "Trajectory Output: $TRAJ_OUTPUT_PATH"
echo "=========================================="

python -u run_virtual.py \
  --suite_family=agent_env \
  --agent_name=$MODEL_NAME \
  --model=$MODEL \
  --api_key=$API_KEY \
  --base_url=$BASE_URL \
  --gemini_api_key=$GEMINI_API_KEY \
  --gemini_base_url=$GEMINI_BASE_URL \
  --gemini_model=$GEMINI_MODEL \
  --initial_image_path=$INITIAL_IMAGE_PATH \
  --traj_output_path=$TRAJ_OUTPUT_PATH \
  --tasks=$TASKS \
  --n_task_combinations=$NUM_TASKS \
  --use_virtual_env=True 2>&1 | tee "$LOG"

echo ""
echo "=========================================="
echo "Execution Complete"
echo "Log saved to: $LOG"
echo "Trajectories saved to: $TRAJ_OUTPUT_PATH"
echo "=========================================="

echo ""
echo "=========================================="
echo "All Done!"
echo "=========================================="