lehen_llm / app.py
AItool's picture
Update app.py
82dc897 verified
raw
history blame contribute delete
885 Bytes
import streamlit as st
#from langchain.llms import OpenAI
from langchain_community.llms import OpenAI
st.title("πŸš€πŸ”— Quickstart App")
# Get the user's OpenAI API key
openai_api_key = st.sidebar.text_input("OpenAI API Key", type="password")
# Define a function to generate LLM responses
def generate_response(input_text):
llm = OpenAI(temperature=0.7, openai_api_key=openai_api_key)
st.info(llm(input_text))
# Create a text input area for user prompts
with st.form("my_form"):
text = st.text_area("Enter text:", "What are the three character traits that a good developer should have?")
submitted = st.form_submit_button("Submit")
# Validate the API key and generate a response
if not openai_api_key.startswith("sk-"):
st.warning("Please enter your OpenAI API key!", icon="⚠")
if submitted and openai_api_key.startswith("sk-"):
generate_response(text)