Update app-azure.py

app-azure.py

@@ -103,6 +103,39 @@ def enable_api_box():
     return enable_box
 """
 
+def set_DRlink(dr_link: str):
+    print("DR Link set")    
+    chat_history = []
+    history = []
+    #count = 0        
+    dr_link_url = 'https://repository.lib.cuhk.edu.hk/en/islandora/object/cuhk%3A'+ dr_link + '/datastream/PDF/'   
+    app.DR_LINK = dr_link        
+    hdr = { 'User-Agent' : user_agent }        
+
+    response = requests.get(dr_link_url, headers=hdr)
+    #print(response.status_code)
+    #print(response.text)
+    file_Path = dr_link + ".pdf"
+    file = None
+    image = None
+    if response.status_code == 200:
+        with open(file_Path, 'wb') as file:
+            file.write(response.content)
+        print('File downloaded successfully')
+    
+    
+    # Use PyMuPDF to render the first page of the uploaded document
+    if file:        
+        image, txt = purge_chat_and_render_first(file)
+    else:
+        raise gr.Error("Error reading DR Link!")
+        
+           
+    
+    #return disable_box
+    #return dr_link
+    return dr_link, image, [], file
+
 def add_text(history, text: str):
     if not text:
         raise gr.Error("enter text")
@@ -346,11 +379,12 @@ with gr.Blocks() as demo:
     vector_db = gr.State()
     qa_chain = gr.State()
     collection_name = gr.State()
+    DR_LINK = gr.State()
     #N = gr.Number()
     #count = gr.Number()
 
     N = 0
-    count = 0
+    #count = 0
     
     #chat_history = gr.State()
     #chat_history: list = []
@@ -402,6 +436,12 @@ with gr.Blocks() as demo:
         ],
     )
     """
+    dr_link.submit(
+        fn=set_DRlink,
+        inputs=[dr_link],
+        #outputs=[dr_link,],
+        outputs=[dr_link, show_img, chatbot, btn],
+    )
 
     btn.upload(
         fn=purge_chat_and_render_first,