langchain-chat / app.py
taimoor61's picture
Update app.py
18be5f3
import os
import time
import streamlit as st
from langchain.llms import openai
def get_openai_response(question):
try:
api_key = os.environ["OPENAI_API_KEY"]
llm = openai.OpenAI(
model_name="text-davinci-003", temperature=0.5, openai_api_key=api_key
)
with st.spinner("Generating response..."):
response = llm(question)
return response
except Exception as e:
st.error(f"Error: {e}")
return None
st.set_page_config(page_title="Enhanced Q&A Demo")
st.header("Xoxo-Langchain App")
input_text = st.text_input("Ask me anything:", key="input")
submit_button = st.button("Get Answer")
response_history = st.empty()
if submit_button:
with st.spinner("Processing..."):
response = get_openai_response(input_text)
if response:
st.subheader("The answer is:")
st.write(response)
# Add response to history
response_history.code(f"""
{input_text}
**Answer:** {response}
---
""")
else:
st.error("Something went wrong. Please try again later.")