add infinity and chinese support

app.py

@@ -152,7 +152,7 @@ class ConversationBot:
               f"Current Memory: {self.agent.memory.buffer}")
         return state, state
 
-    def run_image(self, image, state, txt):
+    def run_image(self, image, state, txt, lang):
         image_filename = os.path.join('image', f"{str(uuid.uuid4())[:8]}.png")
         print("======>Auto Resize Image...")
         img = Image.open(image.name)
@@ -166,8 +166,12 @@ class ConversationBot:
         img.save(image_filename, "PNG")
         print(f"Resize image form {width}x{height} to {width_new}x{height_new}")
         description = self.models['ImageCaptioning'].inference(image_filename)
-        Human_prompt = f'\nHuman: provide a figure named {image_filename}. The description is: {description}. This information helps you to understand this image, but you should use tools to finish following tasks, rather than directly imagine from my description. If you understand, say \"Received\". \n'
-        AI_prompt = "Received.  "
+        if lang == 'Chinese':
+            Human_prompt = f'\nHuman: 提供一张名为 {image_filename}的图片。它的描述是: {description}。 这些信息帮助你理解这个图像，但是你应该使用工具来完成下面的任务，而不是直接从我的描述中想象。 如果你明白了, 说 \"收到\". \n'
+            AI_prompt = "收到。  "
+        else:
+            Human_prompt = f'\nHuman: provide a figure named {image_filename}. The description is: {description}. This information helps you to understand this image, but you should use tools to finish following tasks, rather than directly imagine from my description. If you understand, say \"Received\". \n'
+            AI_prompt = "Received.  "
         self.agent.memory.buffer = self.agent.memory.buffer + Human_prompt + 'AI: ' + AI_prompt
         state = state + [(f"![](/file={image_filename})*{image_filename}*", AI_prompt)]
         print(f"\nProcessed run_image, Input image: {image_filename}\nCurrent state: {state}\n"
@@ -261,7 +265,7 @@ with gr.Blocks(css="#chatbot {overflow:auto; height:500px;}") as demo:
     txt.submit(lambda: "", None, txt)
     run.click(bot.run_text, [txt, state], [chatbot, state])
     run.click(lambda: "", None, txt)
-    btn.upload(bot.run_image, [btn, state, txt], [chatbot, state, txt])
+    btn.upload(bot.run_image, [btn, state, txt, lang], [chatbot, state, txt])
     clear.click(bot.memory.clear)
     clear.click(lambda: [], None, chatbot)
     clear.click(lambda: [], None, state)