Update app.py

app.py

@@ -2,35 +2,33 @@
 #Once you have Streamlit installed, you can import it into your Python script using the import statement,
 
 import streamlit as st
-
-
 from langchain.chat_models import ChatOpenAI
+from langchain.schema import HumanMessage  # Used to pass a message from the user
 
-#Function to return the response
+# Function to return the response
 def load_answer(question):
-    llm =ChatOpenAI(model_name="gpt-4-turbo-preview",temperature=0)
-    answer=llm(question)
-    return answer
+    llm = ChatOpenAI(model_name="gpt-4", temperature=0)
+    
+    # The input to the model must be a list of messages, specifically a HumanMessage.
+    answer = llm([HumanMessage(content=question)])
+    return answer.content  # Extract the content of the response
 
-
-#App UI starts here
+# App UI starts here
 st.set_page_config(page_title="LangChain Demo", page_icon=":robot:")
 st.header("LangChain Demo")
 
-#Gets the user input
+# Gets the user input
 def get_text():
     input_text = st.text_input("You: ", key="input")
     return input_text
 
+user_input = get_text()
+submit = st.button('Generate')
 
-user_input=get_text()
-response = load_answer(user_input)
-
-submit = st.button('Generate')  
-
-#If generate button is clicked
-if submit:
-
+# If generate button is clicked
+if submit and user_input:
     st.subheader("Answer:")
-
-    st.write(response)
+    response = load_answer(user_input)  # Generate the response
+    st.write(response)
+elif submit and not user_input:
+    st.warning("Please enter a question.")