Update app.py

app.py

@@ -71,12 +71,7 @@ def initialize(arxivcode):
 
     return RAG(llm, docs, embeddings)
 
-def handle_prompt(message, history, arxivcode): 
-    if rag_chain is None:
-        # initialize RAG chain 
-        rag_chain = initialize(arxivcode)
-
-        
+def handle_prompt(message, history, arxivcode, rag_chain): 
     try:
         # Stream output
         out=""
@@ -90,15 +85,13 @@ def handle_prompt(message, history, arxivcode):
 greetingsmessage = "Hi, I'm your personal arXiv reader. Ask me questions about the arXiv paper above"
 
 with gr.Blocks() as demo:     
-
   arxiv_code = gr.Textbox("", label="arxiv.number")
-  
+  rag_chain = initialize(arxiv_code)
+    
   gr.ChatInterface(handle_prompt, type="messages", theme=gr.themes.Soft(), 
                           description=greetingsmessage, 
-                   additional_inputs=[arxiv_code]
+                   additional_inputs=[arxiv_code, rag_chain]
                   )
                           
 if __name__=='__main__': 
-    global rag_chain
-    rag_chain = None 
     demo.launch()