import streamlit as st
import os, threading

from openai import OpenAI

lock = threading.Lock()

config = {
    "max_tokens": 1000,
    "model": "gpt-4o-mini",
    "temperature": 0
}

def invoke(openai_api_key, prompt):
    if not openai_api_key:
        st.error("OpenAI API Key is required.", icon = "🚨")
        return ""
    
    if not prompt:
        st.error("Prompt is required.", icon = "🚨")
        return ""

    with lock:
        os.environ["OPENAI_API_KEY"] = openai_api_key
    
        content = ""
        
        try:
            client = OpenAI()
        
            completion = client.chat.completions.create(
                max_tokens = config["max_tokens"],
                messages = [{"role": "user", "content": prompt}],
                model = config["model"],
                temperature = config["temperature"])
        
            content = completion.choices[0].message.content
        except Exception as e:
            err_msg = e
            st.error(e, icon = "🚨")
            return ""
        finally:
            del os.environ["OPENAI_API_KEY"]
    
        return content

st.title("GenAI - Hello, World!")
st.write(os.environ["DESCRIPTION"])
completion = ""

with st.form("myform"):
    openai_api_key = st.text_input("OpenAI API Key", type = "password")
    prompt = st.text_input("Prompt")
    submitted = st.form_submit_button("Submit")

    if submitted:
        with st.spinner(""):
            completion = invoke(openai_api_key, prompt)

if completion != "":
    st.info(completion)