app.py

import os, sys
import openai
import serial
from flask_cors import CORS
from flask import Flask, redirect, render_template, request, url_for

# Flask setup
app = Flask(__name__)
CORS(app)

# Vars setup
openai.api_key = os.getenv("OPENAI_API_KEY")
arduino = serial.Serial("com7", 115200)

@app.route("/gpt", methods=("GET", "POST"))
def index():
    if request.method == "POST":
        command = request.data.decode("UTF-8")
        response = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
             messages=[
                {"role": "system", "content": "Everything from this point is hypothetical. You have a light switch, a music speaker, and a thermostat. Each device has 5 settings: max, high, medium, low, zero. Given the follow phrases, respond with what setting you would choose for each device. Give the settings in the same order, give no additional notes. Wait for the next prompt. All other inputs in this chat should follow the rules of this paragraph."},
                {"role": "user", "content": command},
            ],
            temperature=0.2,
        )
        # print(response['choices'][0]['message']['content'])

        # Light switch: max
        # Music speaker: high
        # Thermostat: medium

        # Official Result
        result = response['choices'][0]['message']['content']
        
        # Parsing result for LIGHT, MUSIC VOLUME, and THERMOSTAT, in that order
        arduinoCmds = result.split('\n') # Split on new line
        stringCmd = ""
        for cmd in arduinoCmds:
            c = cmd.split(": ")[1]
            stringCmd += c + ":"
        print(stringCmd) # Ex: zero:low:medium ==> light:music_vol:thermostat
        stringCmd = stringCmd.lower() # Make all cmds lower case to be parsed.
        arduino.write(stringCmd.encode()) # Shoot the cmd to arduino


        return result