| |
| |
| |
| |
| |
| |
| |
|
|
|
|
| import streamlit as st |
| from langchain.chat_models import ChatOpenAI |
| from langchain.schema import HumanMessage |
|
|
| st.set_page_config(page_title="Chat", |
| page_icon=":chat:") |
| st.header("💬 Hugging Chat 💬") |
|
|
| col1, col2 = st.columns([1,1]) |
|
|
| with col1: |
| option_llm = st.selectbox( |
| "Model", |
| ('gpt-4', |
| 'gpt-3.5-turbo') |
| ) |
|
|
| def get_question(): |
| input_text = st.text_area(label="Your question ...", |
| placeholder="Ask me anything ...", |
| key="question_text", label_visibility="collapsed") |
| return input_text |
|
|
| question_text = get_question() |
| if question_text and len(question_text) > 1: |
| output = "" |
| agent = ChatOpenAI(model_name=option_llm, temperature=0.5) |
| response = agent([HumanMessage(content=question_text)]) |
| print(f"> {response}") |
|
|
| if response and response.content: |
| output = response.content |
| height = min(2*len(output), 280) |
| st.text_area(label="In response ...", |
| value=output, height=height) |
