Update app.py

app.py

@@ -131,19 +131,19 @@ graph = Neo4jGraph(
 # text_splitter = CharacterTextSplitter(chunk_size=100, chunk_overlap=50)
 # documents = text_splitter.split_documents(data)
 
-# llm_transformer = LLMGraphTransformer(llm=llm)
+# llm_transformer = LLMGraphTransformer(llm=chat_model)
 # graph_documents = llm_transformer.convert_to_graph_documents(documents)
 # graph.add_graph_documents(graph_documents, baseEntityLabel=True, include_source=True)
 
 class Entities(BaseModel):
     names: list[str] = Field(..., description="All the person, organization, or business entities that appear in the text")
 
-prompt = ChatPromptTemplate.from_messages([
+entity_prompt = ChatPromptTemplate.from_messages([
     ("system", "You are extracting organization and person entities from the text."),
     ("human", "Use the given format to extract information from the following input: {question}"),
 ])
 
-entity_chain = prompt | llm.with_structured_output(Entities)
+entity_chain = entity_prompt | chat_model.with_structured_output(Entities)
 
 def remove_lucene_chars(input: str) -> str:
     return input.translate(str.maketrans({"\\": r"\\", "+": r"\+", "-": r"\-", "&": r"\&", "|": r"\|", "!": r"\!", 
@@ -224,7 +224,7 @@ Question: {question}
 Use natural language and be concise.
 Answer:"""
 
-prompt = ChatPromptTemplate.from_template(template)
+qa_prompt = ChatPromptTemplate.from_template(template)
 
 chain_neo4j = (
     RunnableParallel(
@@ -233,8 +233,8 @@ chain_neo4j = (
             "question": RunnablePassthrough(),
         }
     )
-    | prompt
-    | llm
+    | qa_prompt
+    | chat_model
     | StrOutputParser()
 )
 
@@ -338,7 +338,7 @@ def fetch_local_news():
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_news&q=birmingham headline&api_key={api_key}'
     response = requests.get(url)
-    if response.status_code == 200:
+    if response.status_code == 200):
         results = response.json().get("news_results", [])
         news_html = """
         <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Birmingham Today</h2>
@@ -632,7 +632,7 @@ def fetch_local_events():
     api_key = os.environ['SERP_API']
     url = f'https://serpapi.com/search.json?engine=google_events&q=Events+in+Birmingham&hl=en&gl=us&api_key={api_key}'
     response = requests.get(url)
-    if response.status_code == 200:
+    if response.status_code == 200):
         events_results = response.json().get("events_results", [])
         events_html = """
         <h2 style="font-family: 'Georgia', serif; color: #ff0000; background-color: #f8f8f8; padding: 10px; border-radius: 10px;">Local Events</h2>