Update app.py
Browse files
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 |
-
|
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):
|