Update app.py

app.py

@@ -6,6 +6,9 @@ from pymongo import MongoClient
 from langchain_community.vectorstores import MongoDBAtlasVectorSearch
 from langchain_openai import OpenAIEmbeddings
 from langchain_community.llms import OpenAI
+from langchain_openai import ChatOpenAI
+from langchain_core.prompts import ChatPromptTemplate
+
 # from langchain_community.prompts import PromptTemplate
 # from langchain.chains import LLMChain
 import json
@@ -20,6 +23,13 @@ collection = client[db_name][collection_name]
 
 try:
     vector_store = MongoDBAtlasVectorSearch(embedding=OpenAIEmbeddings(), collection=collection, index_name='vector_index', text_key='plot', embedding_key='plot_embedding')
+    llm = ChatOpenAI()
+    prompt = ChatPromptTemplate.from_messages([
+    ("system", "You are a movie recommendation engine please elaborate on movies."),
+    ("user", "List of movies: {input}")
+     ])
+    chain = prompt | llm 
+
 
 except:
     # If open ai key is wrong
@@ -34,6 +44,7 @@ def get_movies(message, history):
         for movie in movies:
             retrun_text = retrun_text + 'Title : ' +  movie.metadata['title'] + '\n------------\n' + 'Plot: ' + movie.page_content + '\n\n'
 
+        print_llm_text = chain.invoke({"input": retrun_text})
         for i in range(len(retrun_text)):
             time.sleep(0.05)
             yield "Found: " + "\n\n" + retrun_text[: i+1]