Pavan178 commited on
Commit
aff3a65
·
verified ·
1 Parent(s): f5e36af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -7,7 +7,8 @@ from langchain.vectorstores import FAISS
7
  from langchain.chains import ConversationalRetrievalChain
8
  from langchain.chat_models import ChatOpenAI
9
  from langchain.memory import ConversationBufferMemory
10
-
 
11
 
12
  class AdvancedPdfChatbot:
13
  def __init__(self, openai_api_key):
@@ -39,7 +40,7 @@ class AdvancedPdfChatbot:
39
  return result['answer']
40
 
41
  # Initialize the chatbot
42
- openai_api_key = "your-openai-api-key-here"
43
  pdf_chatbot = AdvancedPdfChatbot(openai_api_key)
44
 
45
  def upload_pdf(pdf_file):
 
7
  from langchain.chains import ConversationalRetrievalChain
8
  from langchain.chat_models import ChatOpenAI
9
  from langchain.memory import ConversationBufferMemory
10
+ import os
11
+ openai_api_key = os.environ.get("OPENAI_API_KEY")
12
 
13
  class AdvancedPdfChatbot:
14
  def __init__(self, openai_api_key):
 
40
  return result['answer']
41
 
42
  # Initialize the chatbot
43
+
44
  pdf_chatbot = AdvancedPdfChatbot(openai_api_key)
45
 
46
  def upload_pdf(pdf_file):