#!/bin/bash

filter_question="$1"
summarization_question="$2"
list="$3"
filename="$4"

if [ -z "$filter_question" ] || [ -z "$summarization_question" ] || [ -z "$list" ]; then
    echo "Usage: $0 <filter_question> <summarization_question> <list_name> <output_file_name>"
    echo "Example: $0 'Is this tweet talking about an important whitehouse action?' 'Summarize what the White house is up to this week.' 'whitehouse' '03-whitehouse.md'"
    exit 1
fi

# Create JSON payload using jq to properly escape values
JSON_PAYLOAD=$(jq -n \
  --arg filter_question "$filter_question" \
  --arg summarization_question "$summarization_question" \
  --arg list "$list" \
  '{
    "filter_question": $filter_question,
    "summarization_question": $summarization_question,
    "list": $list
  }')

JOB_RESPONSE=$(curl -X POST https://tweets.nunosempere.com/api/filter-job \
  -H "Content-Type: application/json" \
  -d "$JSON_PAYLOAD")

# Extract job ID
JOB_ID=$(echo $JOB_RESPONSE | jq -r '.data.job_id')
echo "Created job: $JOB_ID"

# 2. Poll for status until complete
while true; do
  STATUS=$(curl -s "https://tweets.nunosempere.com/api/filter-job/$JOB_ID/status")
  CURRENT_STATUS=$(echo $STATUS | jq -r '.data.status')
  
  echo "Status: $CURRENT_STATUS"
  
  if [ "$CURRENT_STATUS" = "completed" ]; then
    echo "Job completed!"
    break
  elif [ "$CURRENT_STATUS" = "failed" ]; then
    echo "Job failed: $(echo $STATUS | jq -r '.data.error_message')"
    exit 1
  fi
  
  # Show progress
  echo $STATUS | jq '.data.progress'
  
  sleep 2
done

# 3. Get results
WEEK=$(date +%V)
mkdir -p $WEEK

curl "https://tweets.nunosempere.com/api/filter-job/$JOB_ID/results" | jq | tee "$WEEK/$filename.full"

cat "$WEEK/$filename.full" | jq -r .data.results.summary > "$WEEK/$filename.md"