Upload app.py

app.py

@@ -169,13 +169,24 @@ if user_query is not None and find_youtube_links(user_query) != "":
     st.session_state.chat_history.append(AIMessage(content=response))
 
 
+
+# Function to get a response from the model
+def get_response_1(user_query, chat_history):
+    chain = template | chat_model | StrOutputParser()
+    response = chain.invoke({
+        "context": chat_history,
+        "question": user_query,
+    })
+
+    return response
+
 if user_query is not None and user_query != "" and find_youtube_links(user_query) == "":
     st.session_state.chat_history.append(HumanMessage(content=user_query))
 
     with st.chat_message("Human"):
         st.markdown(user_query)
 
-    response = get_response(user_query)
+    response = get_response_1(user_query)
 
     # Remove any unwanted prefixes from the response
     response = response.replace("AI response:", "").replace("chat response:", "").replace("bot response:", "").strip()