Upload app.py

app.py

@@ -167,5 +167,20 @@ if user_query is not None and find_youtube_links(user_query) != "":
         st.write(response)
 
     st.session_state.chat_history.append(AIMessage(content=response))
-else:
-    st.session_state.chat_history.append(AIMessage(content="Please drop the YouTube link related to the financial market, and I will help you summarize and provide insights."))
+
+
+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)
+
+    # Remove any unwanted prefixes from the response
+    response = response.replace("AI response:", "").replace("chat response:", "").replace("bot response:", "").strip()
+
+    with st.chat_message("AI"):
+        st.write(response)
+
+    st.session_state.chat_history.append(AIMessage(content=response))