Changed few things

app.py

@@ -1,26 +1,29 @@
+from typing import Any
 import gradio as gr
 from langchain.embeddings.openai import OpenAIEmbeddings
-from langchain.text_splitter import CharacterTextSplitter
 from langchain.vectorstores import Chroma
 
 from langchain.chains import ConversationalRetrievalChain
 from langchain.chat_models import ChatOpenAI
 
 from langchain.document_loaders import PyPDFLoader
-import os
 
 import fitz
 from PIL import Image
 
-from functools import lru_cache
-import json
-import uuid
-import re
 import chromadb
-from chromadb.config import Settings
-
+import re
 
+enable_box = gr.Textbox.update(value=None,placeholder= 'Upload your OpenAI API key',interactive=True)
+disable_box = gr.Textbox.update(value = 'OpenAI API key is Set',interactive=False)
 
+def set_apikey(api_key):
+        global app
+        app.OPENAI_API_KEY = api_key
+        return disable_box
+    
+def enable_api_box():
+        return enable_box
 
 
 def add_text(history, text):
@@ -30,19 +33,20 @@ def add_text(history, text):
     return history
 
 class my_app:
-    def __init__(self) -> None:
+    def __init__(self, OPENAI_API_KEY= None ) -> None:
+        self.OPENAI_API_KEY = OPENAI_API_KEY
         self.chain = None
         self.chat_history = []
         self.N = 0
-        self.enable_box = gr.Textbox.update(value=None,placeholder= 'Upload your OpenAI API key',interactive=True)
-        self.disable_box = gr.Textbox.update(value = 'OpenAI API key is Set',interactive=False)
- 
-    def set_apikey(self,api_key):
-        self.OPENAI_API_KEY = api_key
-        return self.disable_box
-    
-    def enable_api_box(self):
-        return self.enable_box
+        self.count = 0
+
+    def __call__(self, file) -> Any:
+        if self.count==0:
+            print('This is here')
+            self.build_chain(file)
+            self.count+=1
+
+        return self.chain
     
     def chroma_client(self):
         #create a chroma client
@@ -70,34 +74,32 @@ class my_app:
         self.chain = ConversationalRetrievalChain.from_llm(ChatOpenAI(temperature=0.0, openai_api_key=self.OPENAI_API_KEY), 
                                                   retriever=pdfsearch.as_retriever(search_kwargs={"k": 1}),
                                                   return_source_documents=True,)
+        return self.chain
     
-    def get_response(self, history, query, file):
-
+def get_response(history, query, file):
+        global app
         if not file:
             raise gr.Error(message='Upload a PDF')
         
-        if not self.chain:
-            self.build_chain(file)
-
-        result = self.chain({"question": query, 'chat_history':self.chat_history},return_only_outputs=True)
-        self.chat_history += [(query, result["answer"])]
-        self.N = list(result['source_documents'][0])[1][1]['page']
+        chain = app(file)
+        result = chain({"question": query, 'chat_history':app.chat_history},return_only_outputs=True)
+        app.chat_history += [(query, result["answer"])]
+        app.N = list(result['source_documents'][0])[1][1]['page']
 
         for char in result['answer']:
            history[-1][-1] += char
            yield history,''
 
-    def render_file(self,file):
-                
+def render_file(file):
+        global app       
         doc = fitz.open(file.name)
-        page = doc[self.N]
+        page = doc[app.N]
         #Render the page as a PNG image with a resolution of 300 DPI
         pix = page.get_pixmap(matrix=fitz.Matrix(300/72, 300/72))
         image = Image.frombytes('RGB', [pix.width, pix.height], pix.samples)
         return image
-        
+  
 app = my_app()
-
 with gr.Blocks() as demo:
 
     with gr.Column():
@@ -120,14 +122,13 @@ with gr.Blocks() as demo:
         with gr.Column(scale=0.20):
             btn = gr.UploadButton("📁 upload a PDF", file_types=[".pdf"]).style()
         
+  
+    api_key.submit(fn=set_apikey, inputs=[api_key], outputs=[api_key])
+    change_api_key.click(fn= enable_api_box,outputs=[api_key])
+    btn.upload(fn=render_file, inputs=[btn], outputs=[show_img],)
     
-
-    api_key.submit(fn=app.set_apikey, inputs=[api_key], outputs=[api_key])
-    change_api_key.click(fn= app.enable_api_box,outputs=[api_key])
-    btn.upload(fn=app.render_file, inputs=[btn], outputs=[show_img],)
-    
-    submit_btn.click(fn=add_text, inputs=[chatbot,txt], outputs=[chatbot, ], queue=False).success(fn=app.get_response,inputs = [chatbot, txt, btn],
-                                    outputs = [chatbot,txt]).success(fn=app.render_file,inputs = [btn], outputs=[show_img])
+    submit_btn.click(fn=add_text, inputs=[chatbot,txt], outputs=[chatbot, ], queue=False).success(fn=get_response,inputs = [chatbot, txt, btn],
+                                    outputs = [chatbot,txt]).success(fn=render_file,inputs = [btn], outputs=[show_img])
 
     
 demo.queue()