### Q&A Chatbot ###

import os
import streamlit as st
from langchain_openai import ChatOpenAI
from langchain.prompts.chat import ChatPromptTemplate


## Load Anthropic model and get response

def get_openai_response(question):
    system_template = "You are a helpful coder assistant. When the user asks a question, your task is to write simple Python codes."
    user_template = "{question}"
    chat_prompt = ChatPromptTemplate.from_messages([
        ("system", system_template),
        ("human", user_template)
    ])
    
    chat_llm = ChatOpenAI(temperature=0.5)
    
    chain = chat_prompt | chat_llm
    
    return chain.invoke(question).content
    
    
## Initialize Streamlit app

st.set_page_config(page_title="Coder Q&A Demo")
st.header("Python Helper")

OPENAI_API_KEY = st.text_input("OpenAI API Key: ", type="password")
os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY

input = st.text_input("Input: ", key=input)

submit = st.button("Ask your question...")
if submit:
    response = get_openai_response(input)
    st.subheader("The response is:")
    st.write(response)