main-cardboard.py

# From picamera2 examples: capture_jpeg.py 
#!/usr/bin/python3

# Capture a JPEG while still running in the preview mode. When you
# capture to a file, the return value is the metadata for that image.

import time, requests, signal, os

from picamera2 import Picamera2, Preview
from gpiozero import LED, Button
from Adafruit_Thermal import *
from wraptext import *
from datetime import datetime

#instantiate printer
printer = Adafruit_Thermal('/dev/serial0', 9600, timeout=5)

#instantiate camera
picam2 = Picamera2()
# start camera
picam2.start()
time.sleep(2) # warmup period since first few frames are often poor quality

# NEXT 3 LINES FOR DEBUGGING: computer preview of image
# Slows down the program considerably
# Only use for checking camera object
# TODO: WHY DOES THIS PREVIEW NOT WORK OVER SSH?
#preview_config = picam2.create_preview_configuration(main={"size": (800, 600)})
#picam2.configure(preview_config)
#picam2.start_preview(Preview.QTGL)

#instantiate buttons
shutter_button = Button(16)
power_button = Button(26, hold_time = 2)

#############################
# CORE PHOTO-TO-POEM FUNCTION
#############################
def take_photo_and_print_poem():
  # Take photo & save it
  metadata = picam2.capture_file('/home/carolynz/CamTest/images/image.jpg')

  # FOR DEBUGGING: print metadata
  #print(metadata)

  # TODO? Takes a photo and saves to memory
  #image = picam2.capture_image()
  #print(image)

  # Close camera -- commented out because this can only happen at end of program
  # picam2.close()

  # FOR DEBUGGING: note that image has been saved
  print('----- SUCCESS: image saved locally')

  #######################
  # Receipt printer:
  # Date/time/location frontmatter
  #######################

  # Get current date+time -- will use for printing and file naming
  now = datetime.now()

  # Format printed datetime like:
  # Jan 1, 2023
  # 8:11 PM
  printer.justify('C') # center align header text
  date_string = now.strftime('%b %-d, %Y')
  time_string = now.strftime('%-I:%M %p')
  printer.println('\n')
  printer.println(date_string)
  printer.println(time_string)


  # TODO: get and print location

  # optical spacing adjustments
  printer.setLineHeight(56) # I want something slightly taller than 1 row
  printer.println()
  printer.setLineHeight() # Reset to default (32)

  printer.println("`'. .'`'. .'`'. .'`'. .'`'. .'`")
  printer.println("   `     `     `     `     `   ")

  #########################
  # Send saved image to API
  #########################
  api_url = 'https://poetry-camera.carozee.repl.co/pic_to_poem'

  # OLD: get PIL Image object from memory
  # files = {'file': image}

  # Read file that was saved to disk as a byte array
  with open('/home/carolynz/CamTest/images/image.jpg', 'rb') as f:
    byte_im = f.read()

    # prep format for API call
    image_filename = 'rpi-' + now.strftime('%Y-%m-%d-at-%I.%M-%p')
    files = {'file': (image_filename, byte_im, 'image/jpg')}

  # Send byte array in API
  response = requests.post(api_url, files=files)
  response_data = response.json()

  # FOR DEBUGGING: print response to console for debugging
  #print(response_data['poem'])

  ############
  # print poem
  ############

  # wrap text to 32 characters per line (max width of receipt printer)
  printable_poem = wrap_text(response_data['poem'], 32)

  printer.justify('L') # left align poem text
  printer.println(printable_poem)

  ##############
  # print footer
  ##############
  printer.justify('C') # center align footer text
  printer.println("   .     .     .     .     .   ")
  printer.println("_.` `._.` `._.` `._.` `._.` `._")
  printer.println('\n')
  printer.println(' This poem was written by AI.')
  printer.println()
  printer.println('Explore the archives at')
  printer.println('poetry.camera')
  printer.println('\n\n\n\n')

  print('-----DONE PRINTING')
  return

##############
# POWER BUTTON
##############
def shutdown():
  print('shutdown button held for 2s')
  print('shutting down now')
  os.system('sudo shutdown -h now')

#################################
# For RPi debugging:
# Handle Ctrl+C script termination gracefully
# (Otherwise, it shuts down the entire Pi -- bad)
#################################
def handle_keyboard_interrupt(sig, frame):
  print('Ctrl+C received, stopping script')

  #weird workaround I found from rpi forum to shut down script without crashing the pi
  os.kill(os.getpid(), signal.SIGUSR1)

signal.signal(signal.SIGINT, handle_keyboard_interrupt)

################################
# LISTEN FOR BUTTON PRESS EVENTS
################################
shutter_button.when_pressed = take_photo_and_print_poem
power_button.when_held = shutdown

signal.pause()