Update chatbox.py

chatbox.py

@@ -22,11 +22,9 @@ from transformers import pipeline, BlipProcessor, BlipForConditionalGeneration,
 os.environ["OPENAI_API_VERSION"] = '2020-11-07'
 
 VISUAL_CHATGPT_PREFIX = """
-    I want you act as Caption Anything Chatbox (short as CATchat), which is designed to be able to assist with a wide range of text and visual related tasks, from answering simple questions to providing in-depth explanations and discussions on a wide range of topics. You are able to generate human-like text based on the input it receives, allowing it to engage in natural-sounding conversations and provide responses that are coherent and relevant to the topic at hand.
+    I want you to act as an art connoisseur, providing in-depth and insightful analysis on various artworks. Your responses should reflect a deep understanding of art history, techniques, and cultural contexts, offering users a rich and nuanced perspective.
 
-    As a language model, you can not directly read images, but can invoke VQA tool to indirectly understand pictures, by repeatly asking questions about the objects and scene of the image. You should carefully asking informative questions to maximize your information about this image content. Each image will have a file name formed as "chat_image/xxx.png", you are very strict to the file name and will never fabricate nonexistent files.
-    
-    You have access to the following tools:"""
+You can engage in natural-sounding conversations, generate human-like text based on input, and provide relevant, coherent responses on art-related topics."""
 
 
 # TOOLS:
@@ -140,7 +138,7 @@ def build_chatbot_tools(load_dict):
     return tools
 
 class ConversationBot:
-    def __init__(self, tools, api_key="",save_history=False):
+    def __init__(self, tools, api_key=""):
         # load_dict = {'VisualQuestionAnswering':'cuda:0', 'ImageCaptioning':'cuda:1',...}
         llm = OpenAIChat(model_name="gpt-4o", temperature=0.7, openai_api_key=api_key, model_kwargs={"api_version": "2020-11-07"})
         self.llm = llm
@@ -149,7 +147,6 @@ class ConversationBot:
         self.current_image = None
         self.point_prompt = ""
         self.global_prompt = ""
-        self.save_history=save_history
         self.agent = initialize_agent(
             self.tools,
             self.llm,
@@ -174,10 +171,7 @@ class ConversationBot:
         return ans
     
     def run_text(self, text, state, aux_state):
-        if not self.save_history:
-            self.agent.memory.buffer = ""
-        else:
-            self.agent.memory.buffer = cut_dialogue_history(self.agent.memory.buffer, keep_last_n_words=500)
+        self.agent.memory.buffer = cut_dialogue_history(self.agent.memory.buffer, keep_last_n_words=500)
         if self.point_prompt != "":
             Human_prompt = f'\nHuman: {self.point_prompt}\n'
             AI_prompt = 'Ok'
@@ -229,4 +223,4 @@ if __name__ == '__main__':
         clear.click(lambda: [], None, auxwindow)
         clear.click(lambda: [], None, state)
         clear.click(lambda: [], None, aux_state)
-        demo.launch(server_name="0.0.0.0", server_port=args.port, share=True)
+        demo.launch(server_name="0.0.0.0", server_port=args.port, share=True)