# Chat-Bot
from langchain.llms import OpenAI
from dotenv import load_dotenv
import streamlit as st
import os

if os.getenv('SYSTEM') == 'spaces':
    subprocess.run(shlex.split('pip install langchain-community langchain-core'))
#    subprocess.run(
#        shlex.split(
#            'pip install git+https://github.com/modelscope/modelscope.git@refs/pull/207/head'
#        ))

load_dotenv()

def GetOpenaiResponse(question,openaikey):
    llm = OpenAI(
        openai_api_key=os.getenv("OPENAI_API_KEY"),
        temperature=0.6,
    )
    response = llm.invoke(question)
    return response


st.set_page_config(page_title="OpenAI Q&A Demo")
st.header("Open AI GPT Chat:")

input = st.text_input("Input : ", key="input")
apikey = st.text_input("API Key : ", key="input")
response = GetOpenaiResponse(input,apikey)


submit = st.button("Generate Answer")

if submit:
    st.subheader("OpenAI response :")
    st.write(response)