Update app.py

app.py

@@ -10,6 +10,7 @@ from langchain.document_loaders import PyPDFLoader, WebBaseLoader
 from langchain.document_loaders.blob_loaders.youtube_audio import YoutubeAudioLoader
 from langchain.document_loaders.generic import GenericLoader
 from langchain.document_loaders.parsers import OpenAIWhisperParser
+from langchain.schema import AIMessage, HumanMessage
 
 from langchain.embeddings.openai import OpenAIEmbeddings
 from langchain.prompts import PromptTemplate
@@ -204,7 +205,7 @@ def generate_prompt_with_history_langchain(prompt, history):
     for human, ai in history:
         history_langchain_format.append(HumanMessage(content=human))
         history_langchain_format.append(AIMessage(content=ai))
-    history_langchain_format.append(HumanMessage(content=message))
+    history_langchain_format.append(HumanMessage(content=prompt))
   
     return history_langchain_format