Upload app.py

app.py

@@ -169,15 +169,33 @@ if user_query is not None and find_youtube_links(user_query) != "":
     st.session_state.chat_history.append(AIMessage(content=response))
 
 
+template_2 = """
+You are a genius trader with extensive knowledge of the financial and stock markets, capable of providing deep and insightful analysis of financial stocks with remarkable accuracy.
+
+**ALWAYS**
+Be as detailed as possible, but don't make up any information that’s not from the context.
+If you don't know an answer, say you don't know.
+Let's think step by step.
+
+Please ensure responses are informative, accurate, and tailored to the user's queries and preferences.
+Use natural language to engage users and provide readable content throughout your response.
+
+Chat history:
+{chat_history}
+
+User question:
+{user_question}
+"""
+
+prompt_2 = ChatPromptTemplate.from_template(template_2)
 
 # Function to get a response from the model
-def get_response_1(user_query, chat_history):
-    chain = template | chat_model | StrOutputParser()
+def get_response_2(user_query, chat_history):
+    chain = prompt_2 | chat_model | StrOutputParser()
     response = chain.invoke({
-        "context": chat_history,
-        "question": user_query,
+        "chat_history": chat_history,
+        "user_question": user_query,
     })
-
     return response
 
 if user_query is not None and user_query != "" and find_youtube_links(user_query) == "":
@@ -186,7 +204,7 @@ if user_query is not None and user_query != "" and find_youtube_links(user_query
     with st.chat_message("Human"):
         st.markdown(user_query)
 
-    response = get_response_1(user_query, st.session_state.chat_history)
+    response = get_response_2(user_query, st.session_state.chat_history)
 
     # Remove any unwanted prefixes from the response
     response = response.replace("AI response:", "").replace("chat response:", "").replace("bot response:", "").strip()