dynamic references gradio

app.py

@@ -35,7 +35,7 @@ def image_to_bytes(image):
     return base64.b64encode(img_byte_arr.getvalue()).decode("utf-8")
 
 
-@spaces.GPU
+@spaces.GPU(duration=60*4)
 def get_image_descriptions(images):
     torch.cuda.empty_cache()
     gc.collect()
@@ -244,8 +244,9 @@ def conversation(vectordb_client, msg, num_context, img_context, history):
     """
     prompt = PromptTemplate(template=template, input_variables=["context", "question"])
     context = "\n\n".join(results)
+    # references = [gr.Textbox(i, visible=True, interactive=False) for i in results]
     response = llm(prompt.format(context=context, question=msg, images=img_desc))
-    return history + [(msg, response)], context, images_and_locs
+    return history + [(msg, response)], results, images_and_locs
 
 
 def check_validity_and_llm(session_states):
@@ -272,6 +273,8 @@ with gr.Blocks(css=CSS) as demo:
     vectordb = gr.State()
     doc_collection = gr.State(value=[])
     session_states = gr.State(value={})
+    references = gr.State(value=[])
+    
     gr.Markdown(
         """<h2><center>Multimodal PDF Chatbot</center></h2>
     <h3><center><b>Interact With Your PDF Documents</b></center></h3>"""
@@ -330,27 +333,7 @@ with gr.Blocks(css=CSS) as demo:
                                 label="Sample Extracted Images", columns=1, rows=2
                             )
 
-                    # with gr.Row():
-                    #   image_desc = gr.Textbox(label="Image Descriptions", interactive=False)
-                    # with gr.Row(variant="panel"):
-                    #     ext_tables = gr.HTML("<h3>Sample Tables</h3>", label="Extracted Tables")
-
-        # with gr.TabItem("Embeddings", id=3) as embed_tab:
-        #     with gr.Row():
-        #         with gr.Column():
-        #             back_p2 = gr.Button(value="Back")
-        #         with gr.Column():
-        #             view_stats = gr.Button(value="View Stats")
-        #         with gr.Column():
-        #             next_p2 = gr.Button(value="Next")
-
-        #     with gr.Row():
-        #         with gr.Column():
-        #             text_stats = gr.Textbox(label="Text Stats", interactive=False)
-        #         with gr.Column():
-        #             table_stats = gr.Textbox(label="Table Stats", interactive=False)
-        #         with gr.Column():
-        #             image_stats = gr.Textbox(label="Image Stats", interactive=False)
+
 
         with gr.TabItem("Chat", id=2) as chat_tab:
             with gr.Column():
@@ -382,8 +365,14 @@ with gr.Blocks(css=CSS) as demo:
                 with gr.Column():
                     chatbot = gr.Chatbot(height=400)
             with gr.Accordion("Text References", open=False):
-                with gr.Row():
-                    text_context = gr.Textbox(interactive=False, lines=10)
+                # text_context = gr.Row()
+                
+                @gr.render(inputs=[references])
+                def gen_refs(refs):
+                    n = len(refs)
+                    for i in range(n):
+                        gr.Textbox(label=f"Ref-{i+1}", value=refs[i], lines=3)
+                
 
             with gr.Row():
                 msg = gr.Textbox(
@@ -408,27 +397,20 @@ with gr.Blocks(css=CSS) as demo:
             session_states,
             sample_data,
             ext_text,
-            # ext_tables,
             images,
             prog,
-            # image_desc
         ],
     )
 
     submit_btn.click(
         conversation,
         [vectordb, msg, num_context, img_context, chatbot],
-        [chatbot, text_context, ret_images],
+        [chatbot, references, ret_images],
     )
 
-    # view_stats.click(
-    #     get_stats, [vectordb], outputs=[text_stats, table_stats, image_stats]
-    # )
-
-    # Page Navigation
 
     back_p1.click(lambda: gr.Tabs(selected=0), None, tabs)
 
     next_p1.click(check_validity_and_llm, session_states, tabs)
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)