Update app.py

app.py

@@ -15,7 +15,7 @@ api_base = os.environ["OPENAI_API_BASE"]
 
 def chatbot_initialize():
     retriever = core.retriever.ChromaRetriever(pdf_dir="", 
-                                               collection_name="langchain", 
+                                               collection_name="pdfs_1000", 
                                                split_args={"size": 2048, "overlap": 10}, #embedding_model="text-embedding-ada-002"
                                                embed_model=models.BiomedModel()
                                                )
@@ -25,7 +25,11 @@ def chatbot_initialize():
 async def respond(query, chat_history, img_path_list, chat_history_string):
     time1 = time.time()
     global Chatbot
-    response, logs = await Chatbot.response(query, image_paths=img_path_list, return_logs=True)
+    result = await Chatbot.response(query, image_paths=img_path_list)
+    response = result["answer"]
+    logs = result["logs"]
+    titles_set = result["titles"]
+    titles = "\n".join(list(titles_set))
     chat_history.append((query, response))
     if img_path_list is None:
         chat_history_string += "Query: " + query + "\nImage: None" + "\nResponse: " + response + "\n\n\n"
@@ -33,7 +37,7 @@ async def respond(query, chat_history, img_path_list, chat_history_string):
         chat_history_string += "Query: " + query + "\nImages: " + "\n".join([path.name for path in img_path_list]) + "\nResponse: " + response + "\n\n\n"
     time2 = time.time()
     print(f"Total: {time2-time1}")
-    return "", chat_history, logs, chat_history_string
+    return "", chat_history, logs + "\n\n" + titles, chat_history_string
 
 if __name__ == "__main__":
     global Chatbot