Update app.py

app.py

@@ -900,10 +900,8 @@ logging.basicConfig(level=logging.DEBUG)
 
 embeddings = OpenAIEmbeddings(api_key=os.environ['OPENAI_API_KEY'])
 
-from pinecone import Pinecone
-pc = Pinecone(api_key=os.environ['PINECONE_API_KEY'])
-
-
+import pinecone
+pinecone.init(api_key=os.environ['PINECONE_API_KEY'])
 
 index_name = "radardata07242024"
 vectorstore = PineconeVectorStore(index_name=index_name, embedding=embeddings)
@@ -969,20 +967,6 @@ graph = Neo4jGraph(
     password="B_sZbapCTZoQDWj1JrhwqElsNa-jm5Zq1m_mAnyPYog"
 )
 
-# Avoid pushing the graph documents to Neo4j every time
-# Only push the documents once and comment the code below after the initial push
-# dataset_name = "Pijush2023/birmindata07312024"
-# page_content_column = 'events_description'
-# loader = HuggingFaceDatasetLoader(dataset_name, page_content_column)
-# data = loader.load()
-
-# text_splitter = CharacterTextSplitter(chunk_size=100, chunk_overlap=50)
-# documents = text_splitter.split_documents(data)
-
-# 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")
 
@@ -1099,9 +1083,11 @@ def generate_answer(message, choice, retrieval_mode):
             chain_type_kwargs={"prompt": prompt_template}
         )
         response = qa_chain({"query": message})
-        return response['output'], extract_addresses(response['output'])
+        logging.debug(f"Vector response: {response}")
+        return response['result'], extract_addresses(response['result'])
     elif retrieval_mode == "Knowledge-Graph":
         response = chain_neo4j.invoke({"question": message})
+        logging.debug(f"Knowledge-Graph response: {response}")
         return response, extract_addresses(response)
     else:
         return "Invalid retrieval mode selected.", []
@@ -1687,3 +1673,4 @@ demo.launch(share=True)
 
 
 
+