Update app.py

app.py

@@ -10,41 +10,41 @@ import fitz
 from PIL import Image
 import streamlit as st
 
-# # Global variables
-# COUNT, N = 0, 0
-# chat_history = []
-# chain = None  # Initialize chain as None
-
-# # Function to set the OpenAI API key
-# def set_apikey(api_key):
-#     os.environ['OPENAI_API_KEY'] = api_key
-#     return disable_box  # Update the disable_box
-
-# # Function to enable the API key input box
-# def enable_api_box():
-#     return enable_box  # Update the enable_box
-
-# # Function to add text to the chat history
-# def add_text(history, text):
-#     if not text:
-#         raise gr.Error('Enter text')
-#     history = history + [(text, '')]
-#     return history
-
-# # Function to process the PDF file and create a conversation chain
-# def process_file(file):
-#     global chain  # Access the global 'chain' variable
-#     if 'OPENAI_API_KEY' not in os.environ:
-#         raise gr.Error('Upload your OpenAI API key')
-
-#     loader = PyPDFLoader(file.name)
-#     documents = loader.load()
-#     embeddings = OpenAIEmbeddings()
-#     pdfsearch = Chroma.from_documents(documents, embeddings)
-#     chain = ConversationalRetrievalChain.from_llm(ChatOpenAI(temperature=0.3),
-#                                     retriever=pdfsearch.as_retriever(search_kwargs={"k": 1}),
-#                                     return_source_documents=True)
-#     return chain
+# Global variables
+COUNT, N = 0, 0
+chat_history = []
+chain = None  # Initialize chain as None
+
+# Function to set the OpenAI API key
+def set_apikey(api_key):
+    os.environ['OPENAI_API_KEY'] = api_key
+    return disable_box  # Update the disable_box
+
+# Function to enable the API key input box
+def enable_api_box():
+    return enable_box  # Update the enable_box
+
+# Function to add text to the chat history
+def add_text(history, text):
+    if not text:
+        raise gr.Error('Enter text')
+    history = history + [(text, '')]
+    return history
+
+# Function to process the PDF file and create a conversation chain
+def process_file(file):
+    global chain  # Access the global 'chain' variable
+    if 'OPENAI_API_KEY' not in os.environ:
+        raise gr.Error('Upload your OpenAI API key')
+
+    loader = PyPDFLoader(file.name)
+    documents = loader.load()
+    embeddings = OpenAIEmbeddings()
+    pdfsearch = Chroma.from_documents(documents, embeddings)
+    chain = ConversationalRetrievalChain.from_llm(ChatOpenAI(temperature=0.3),
+                                    retriever=pdfsearch.as_retriever(search_kwargs={"k": 1}),
+                                    return_source_documents=True)
+    return chain
 
 # # Function to generate a response based on the chat history and query
 def generate_response(history, query, btn):