#!/bin/bash

#qwen
# export QWEN_API_KEY='your_qwen_api_key' # openai
# export QWEN_API_BASE_URL='https://dashscope.aliyuncs.com/compatible-mode/v1' # openai

# deepseek
# export OPENAI_API_KEY='your_deepseek_api_key'
# export OPENAI_API_BASE_URL='https://api.deepseek.com'

# export OPENAI_API_KEY='your_openai_api_key' # openai
# export OPENAI_API_BASE_URL='https://api.openai.com' # openai

# gemini
# export GEMINI_API_KEY="your_gemini_api_key"

# ZhiPu glm
# export ZHIPUAI_API_KEY='your_glm_api_key'
# export OPENAI_API_BASE_URL='https://open.bigmodel.cn/api/paas/v4/'


export OPENROUTER_API_KEY="your_openrouter_api_key"
export OPENROUTER_API_BASE_URL="https://openrouter.ai/api/v1"

dataset_dir="pdfs"
paper_name="AutoPage"
                    
python -m ProjectPageAgent.main_pipline\
    --paper_path="${dataset_dir}/${paper_name}.pdf" \
    --model_name_t="your_text_model" \
    --model_name_v="your_vlm_model" \
    --template_root="templates" \
    --template_dir="your_template_dir" \
    --template_file="your_template_file" \
    --output_dir="generated_project_pages" \
    --full_content_check_times=2 \
    --html_check_times=2 \
    --resume='parse_pdf' \
    --human_input='1' \
    --background_color='dark' \
    --has_navigation="yes" \
    --has_hero_section="no" \
    --title_color="colorful" \
    --page_density="compact" \
    --image_layout="rotation"