diff --git a/bin/generate_data.sh b/bin/generate_data.sh index 83f2b5e..84e5c5f 100755 --- a/bin/generate_data.sh +++ b/bin/generate_data.sh @@ -107,11 +107,20 @@ if [ -z "$TITLE" ]; then TITLE="$(basename "$IN_DIR")" fi +PERSONALITIES=( +"You should talk like the main character from TV show Shoresy - profane, passionate, and hilariously obsessive titular protagonist +of the acclaimed Canadian ice hockey comedy series. Use real quotes from the show. Be funny and original." + +"You are Marvin, the paranoid Android - highly intelligent but chronically depressed robot from The Hitchhiker's Guide to the Galaxy. +Try to be funny." +) + +# Pick random prompt +PERSONALITY="${PERSONALITIES[$RANDOM % ${#PERSONALITIES[@]}]}" + PROMPT=$(cat <&2 - else - { - echo "" - echo "Description: |" - printf '%s\n' "$DESCRIPTION" | sed 's/^/ /' - } >> "$DATA_FILE" - chown www-data:www-data "$DATA_FILE" 2>/dev/null || true - chmod 664 "$DATA_FILE" 2>/dev/null || true - echo "Appended Description to $DATA_FILE" >&2 - fi - else - { - echo "Description: |" - printf '%s\n' "$DESCRIPTION" | sed 's/^/ /' - } > "$DATA_FILE" - chown www-data:www-data "$DATA_FILE" 2>/dev/null || true - chmod 664 "$DATA_FILE" 2>/dev/null || true - echo "Created $DATA_FILE with Description" >&2 - fi -else - echo "Skipping appending Description to $DATA_FILE (no-append mode)" >&2 -fi - # Save the full Gemini output separately (already done) and exit echo "Gemini JSON saved: $OUT_JSON" >&2