Update response function

app.py

@@ -48,7 +48,6 @@ def format_text(text):
 
 
 def get_chat_response(chatbot, task_history, model, tokenizer):
-    """Generate a response using the model."""
     chat_query = chatbot[-1][0]
     query = task_history[-1][0]
     history_cp = copy.deepcopy(task_history)
@@ -77,17 +76,18 @@ def get_chat_response(chatbot, task_history, model, tokenizer):
         name = f"tmp{secrets.token_hex(5)}.jpg"
         filename = temp_dir / name
         image.save(str(filename))
-        chatbot[-1] = (chat_query, (str(filename),))
+        chatbot[-1] = (format_text(chat_query), (str(filename),))  # Hier verwenden wir format_text statt _parse_text
         chat_response = response.replace("<ref>", "")
         chat_response = chat_response.replace(r"</ref>", "")
+        chat_response = re.sub(BOX_TAG_PATTERN, "", chat_response)
         if chat_response != "":
             chatbot.append((None, chat_response))
     else:
-        chatbot[-1] = (chat_query, response)
-    full_response = response  # modified this line to ensure full_response is not empty
+        chatbot[-1] = (format_text(chat_query), response)  # Hier verwenden wir format_text statt _parse_text
+    full_response = format_text(response)  # Hier verwenden wir format_text statt _parse_text
 
     task_history.append((message, full_response))
-    chatbot.append((format_text(message), format_text(full_response)))
+    chatbot.append((format_text(message), format_text(full_response)))  # Hier verwenden wir format_text statt _parse_text
 
     return chatbot, task_history
 
@@ -160,8 +160,8 @@ with gr.Blocks(css=css) as demo:
     
     gr.Markdown("### Key Features:\n- **Strong Performance**: Surpasses existing LVLMs on multiple English benchmarks including Zero-shot Captioning and VQA.\n- **Multi-lingual Support**: Supports English, Chinese, and multi-lingual conversation.\n- **High Resolution**: Utilizes 448*448 resolution for fine-grained recognition and understanding.")
     submit_btn.click(handle_text_input, [chatbot, task_history, query], [chatbot, task_history]).then(
-        get_chat_response, [chatbot, task_history, model, tokenizer], [chatbot], show_progress=True
-    )
+    get_chat_response, [chatbot, task_history, model, tokenizer], [chatbot], show_progress=True
+)