added state in app.py

app.py

@@ -13,13 +13,19 @@ from PIL import Image
 
 import chromadb
 import re
+import uuid 
 
 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
+user_app = {}
+user_api = {}
+
+def set_apikey(state, api_key):
+
+        if state not in user_app:
+             user_api[state] = api_key
+        
         return disable_box
     
 def enable_api_box():
@@ -76,12 +82,21 @@ class my_app:
                                                   return_source_documents=True,)
         return self.chain
     
-def get_response(history, query, file):
-        global app
+
+def get_response(state, history, query, file):
+        
         if not file:
             raise gr.Error(message='Upload a PDF')
+        if state not in user_app:
+             app = my_app()
+             user_app[state] = app
+        else:
+            app = user_app[state]
+       
+        app.OPENAI_API_KEY = user_api[state]
         
         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']
@@ -90,18 +105,30 @@ def get_response(history, query, file):
            history[-1][-1] += char
            yield history,''
 
-def render_file(file):
-        global app       
+def render_file(state,file):
+        if state not in user_app:
+             app = my_app()
+             user_app[state] = app
+        else:
+            app = user_app[state]   
         doc = fitz.open(file.name)
         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:
 
+def render_first(file):
+        doc = fitz.open(file.name)
+        page = doc[0]
+        #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,[]
+
+
+with gr.Blocks() as demo:
+    state = gr.State(uuid.uuid4().hex)
     with gr.Column():
         with gr.Row():
             with gr.Column(scale=0.8):
@@ -123,12 +150,12 @@ with gr.Blocks() as demo:
             btn = gr.UploadButton("📁 upload a PDF", file_types=[".pdf"]).style()
         
   
-    api_key.submit(fn=set_apikey, inputs=[api_key], outputs=[api_key])
+    api_key.submit(fn=set_apikey, inputs=[state, 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],)
+    btn.upload(fn=render_first, inputs=[btn], outputs=[show_img,chatbot],)
     
-    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])
+    submit_btn.click(fn=add_text, inputs=[chatbot,txt], outputs=[chatbot, ], queue=False).success(fn=get_response,inputs = [state,chatbot, txt, btn],
+                                    outputs = [chatbot,txt]).success(fn=render_file,inputs = [state,btn], outputs=[show_img])
 
     
 demo.queue()