| import os | |
| import streamlit as st | |
| import | |
| from langchain_openai import OpenAI | |
| from langchain_core.prompts import ChatPromptTemplate | |
| st.title("Chat with your data") | |
| inp = st.text_input("Enter your prompt here") | |
| llm = OpenAI( | |
| api_key=os.environ.get("OPENAI_API_KEY"), | |
| temperature=0.2, | |
| model='gpt-3.5-turbo-0125') | |
| # prompt = ChatPromptTemplate.from_messages( | |
| # [ | |
| # ( | |
| # "system", | |
| # "You are a helpful assistant that answer questions from user.", | |
| # ), | |
| # ("human", "{input}"), | |
| # ] | |
| # ) | |
| chain = llm | |
| if inp: | |
| response = chain.invoke(input=inp) | |
| st.write(response) | |