Update app.py

app.py

@@ -1,32 +1,53 @@
-# Q&A Chatbot
-from langchain.llms import openai
-#from dotenv import load_dotenv
-
-#load_dotenv() # Take environment variables from .env
+import os
+import time
+from dotenv import load_dotenv
 
 import streamlit as st
-import os
+from langchain.llms import openai
 
-#Function to load OpenAI model and get response
+# Load environment variables
+load_dotenv()
 
+# Function to get OpenAI response
 def get_openai_response(question):
-    llm = openai.OpenAI(model_name= 'text-davinci-003', 
-                 temperature = 0.5)
-    response = llm(question)
-    return response
-
-## Initialize our Streamlit Apps
-
-st.set_page_config(page_title="Q&A Demo")
-
-st.header("Langchain Application")
-
-input = st.text_input("Input: ",key='input')
-#response = get_openai_response(input)
-
-submit = st.button("Ask the 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
+
+# Initialize Streamlit app
+st.set_page_config(page_title="Enhanced Q&A Demo")
+st.header("Langchain App")
+
+# User input and submit button
+input_text = st.text_input("Ask me anything:", key="input")
+submit_button = st.button("Get Answer")
+
+# Response history
+response_history = st.empty()
+
+# Handle button click and display response
+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.")
 
-if submit:
-    st.subheader("The response is")
-    response = get_openai_response(input)
-    st.write(response)