update introduction

app.py

@@ -5,12 +5,15 @@ import os
 import json
 import requests
 import time
+from PIL import Image
+import datetime
+
 from concurrent.futures import ThreadPoolExecutor
 from utils import is_chinese, process_image_without_resize, parse_response, templates_agent_cogagent, template_grounding_cogvlm, postprocess_text
 
-DESCRIPTION = '''<h2 style='text-align: center'> <a href="https://github.com/THUDM/CogVLM"> CogVLM & CogAgent Chat Demo</a> </h2>'''
+DESCRIPTION = '''<h2 style='text-align: center'> <a href="https://github.com/THUDM/CogVLM2"> CogVLM2 </a>& <a href="https://github.com/THUDM/CogVLM">CogAgent Chat Demo</a> </h2>'''
 
-NOTES = 'This app is adapted from <a href="https://github.com/THUDM/CogVLM">https://github.com/THUDM/CogVLM</a>. It would be recommended to check out the repo if you want to see the detail of our model.\n\n该demo仅作为测试使用，不支持批量请求。如有大批量需求，欢迎联系[智谱AI](mailto:[email protected])。\n\n请注意该Demo目前仅支持英文，<a href="http://36.103.203.44:7861/">备用网页</a>支持中文。'
+NOTES = 'This app is adapted from <a href="https://github.com/THUDM/CogVLM">https://github.com/THUDM/CogVLM2</a> and <a href="https://github.com/THUDM/CogVLM">https://github.com/THUDM/CogVLM</a>. It would be recommended to check out the repo if you want to see the detail of our model.\n\n该demo仅作为测试使用，不支持批量请求。如有大批量需求，欢迎联系[智谱AI](mailto:[email protected])。\n\n请注意该Demo目前仅支持英文，<a href="http://36.103.203.44:7861/">备用网页</a>支持中文。'
 
 MAINTENANCE_NOTICE1 = 'Hint 1: If the app report "Something went wrong, connection error out", please turn off your proxy and retry.<br>Hint 2: If you upload a large size of image like 10MB, it may take some time to upload and process. Please be patient and wait.'
 
@@ -28,6 +31,41 @@ def make_request(URL, headers, data):
     response = requests.request("POST", URL, headers=headers, data=data, timeout=(60, 100))
     return response.json()
 
+def report_fn(image_prompt, result_previous, report=False, base_dir='/home/wangyan/data/badcases'):
+    if not report:
+        return "Report failed! Checking Report when you want to report a bad case!"    
+    
+    os.makedirs(base_dir, exist_ok=True)
+    date_dir = datetime.datetime.now().strftime('%Y%m%d')
+    new_dir = os.path.join(base_dir, date_dir)
+    os.makedirs(new_dir, exist_ok=True)
+    conv_path = os.path.join(base_dir, f'cases_report_{date_dir}.jsonl')
+
+    result_text = [(ele[0], ele[1]) for ele in result_previous]
+    for i in range(len(result_text)-1, -1, -1):
+        if result_text[i][0] == "" or result_text[i][0] == None:
+            del result_text[i]
+   
+    if image_prompt and len(result_text) > 0:
+        image = Image.open(image_prompt)
+        timestamp = time.time()
+        file_ext = os.path.splitext(image_prompt)[1]
+        filename = os.path.join(new_dir, f"{timestamp}{file_ext}")
+        image.save(filename)
+        
+        conv_data = {
+            'id': str(int(time.time())),
+            'history': [filename, *[i for c in result_text for i in c]]
+        }
+
+        mode = 'a' if os.path.isfile(conv_path) else 'w'
+        with open(conv_path, mode) as f:
+            f.write(json.dumps(conv_data, ensure_ascii=False)+"\n")
+        return "Submit success!"
+    return "Submit failed!"
+
+
+
 def post(
         input_text,
         temperature,
@@ -180,6 +218,10 @@ def main():
             with gr.Column(scale=5.5):
                 result_text = gr.components.Chatbot(label='Multi-round conversation History', value=[("", "Hi, What do you want to know about this image?")], height=550)
                 hidden_image_hash = gr.Textbox(visible=False)
+                with gr.Row():
+                    bad_case_button = gr.Button('Report bad case')
+                    report = gr.Checkbox(label='Report')
+                    report_text = gr.Textbox(value='', interactive=False, show_label=False)
 
         gr_examples = gr.Examples(examples=[[example["text"], example["image"], example["grounding"], example["cogagent"]] for example in examples], 
                                   inputs=[input_text, image_prompt, grounding, cogagent],
@@ -196,6 +238,7 @@ def main():
         clear_button.click(fn=clear_fn, inputs=clear_button, outputs=[input_text, result_text, image_prompt])
         image_prompt.upload(fn=clear_fn2, inputs=clear_button, outputs=[result_text])
         image_prompt.clear(fn=clear_fn2, inputs=clear_button, outputs=[result_text])
+        bad_case_button.click(fn=report_fn, inputs=[image_prompt, result_text, report], outputs=[report_text])
 
         print(gr.__version__)