bug fix

app.py

@@ -1,3 +1,4 @@
+%%writefile app.py
 import gradio as gr
 from unstructured.partition.pdf import partition_pdf
 import pymupdf
@@ -217,7 +218,8 @@ def conversation(vectordb_client, msg, num_context, img_context, history):
     results = text_collection.query(
         query_texts=[msg], include=["documents"], n_results=num_context
     )["documents"][0]
-    
+    # print(results)
+    # print("R"*100)
     similar_images = image_collection.query(
         query_texts=[msg],
         include=["metadatas", "distances", "documents"],
@@ -249,7 +251,7 @@ def conversation(vectordb_client, msg, num_context, img_context, history):
     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))
-    yield history + [(msg, response)], results, images_and_locs
+    return history + [(msg, response)], results, images_and_locs
 
 
 def check_validity_and_llm(session_states):
@@ -370,11 +372,12 @@ with gr.Blocks(css=CSS) as demo:
             with gr.Accordion("Text References", open=False):
                 # text_context = gr.Row()
                 
-                @gr.render(inputs=[references])
-                def gen_refs(refs):
-                    n = len(refs)
+                @gr.render(inputs=references)
+                def gen_refs(references):
+                    # print(references)
+                    n = len(references)
                     for i in range(n):
-                        gr.Textbox(label=f"Ref-{i+1}", value=refs[i], lines=3)
+                        gr.Textbox(label=f"Reference-{i+1}", value=references[i], lines=3)
                 
 
             with gr.Row():
@@ -408,7 +411,7 @@ with gr.Blocks(css=CSS) as demo:
     submit_btn.click(
         conversation,
         [vectordb, msg, num_context, img_context, chatbot],
-        [chatbot, references, ret_images],
+        [chatbot,references ,ret_images],
     )
 
 
@@ -416,4 +419,4 @@ with gr.Blocks(css=CSS) as demo:
 
     next_p1.click(check_validity_and_llm, session_states, tabs)
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch()