Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 import PyPDF2
 import pptx
 import os
-from langchain.llms import HuggingFaceHub
+from langchain.llms import OpenAI
 from langchain.vectorstores.cassandra import Cassandra
 from langchain.indexes.vectorstore import VectorStoreIndexWrapper
 from langchain.embeddings import OpenAIEmbeddings
@@ -16,6 +16,11 @@ ASTRA_DB_ID = os.getenv("ASTRA_DB_ID")
 HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
+
+openai.api_key = OPENAI_API_KEY
+
+llm = OpenAI(openai_api_key=OPENAI_API_KEY)
+
 if not ASTRA_DB_APPLICATION_TOKEN or not ASTRA_DB_ID:
     st.error("Astra DB credentials are missing. Set the environment variables.")
     st.stop()
@@ -99,7 +104,7 @@ def main():
 
     if submit_query and query:
        
-        response = astra_vector_index.query(query, llm =hf_llm)
+        response = astra_vector_index.query(query, llm =llm)
         st.write(f"Response: {response}")