Update app.py

app.py

@@ -17,15 +17,20 @@ cohere_api_key = os.getenv('COHERE')
 weaviate_api_key = os.getenv('WEAVIATE')
 weaviate_url = os.getenv('WEAVIATE_URL')
 
+
 # Weaviate connection
 auth_config = weaviate.auth.AuthApiKey(api_key=weaviate_api_key)
-client = weaviate.Client(url=weaviate_url, auth_client_secret=auth_config, additional_headers={"X-Cohere-Api-Key": cohere_api_key})
-
+client = weaviate.client(url=weaviate_url, auth_client_secret=auth_config, 
+                         additional_headers={"X-Cohere-Api-Key": cohere_api_key})
 # Initialize vectorstore
-vectorstore = weaviate(client, index_name="Articles", text_key="text")
+vectorstore = Weaviate(client, index_name="Articles", text_key="text")
 vectorstore._query_attrs = ["text", "title", "url", "views", "lang", "_additional {distance}"]
 vectorstore.embedding = CohereEmbeddings(model="embed-multilingual-v2.0", cohere_api_key=cohere_api_key)
 
+# Initialize OpenAI and RetrievalQA
+llm = OpenAI(temperature=0, openai_api_key=openai_api_key)
+qa = RetrievalQA.from_chain_type(llm, retriever=vectorstore.as_retriever())
+
 def embed_pdf(file, collection_name):
     # Save the uploaded file
     filename = file.name