Update app.py

app.py

@@ -40,16 +40,17 @@ def img_size(image):
     width = int(width)
     height = int(height)
     resized_img = image.resize((width, height))
+    # resized_img.save(image_path)
     return resized_img
 
 
+
 def encode_image(image):
     buffered = io.BytesIO()
     image.save(buffered, format="PNG")
     return base64.b64encode(buffered.getvalue()).decode("utf-8")
 
 
-
 def image_summarize(img_base64, prompt):
     chat = ChatOpenAI(model="gpt-4o", max_tokens=256)
     msg = chat.invoke(
@@ -156,7 +157,7 @@ Pre Thought: 简要理解要验证的内容，是否包含多个要验证的细
 Thought: 为了验证第一个细节部分, 你应该怎么做.
 Action: 工具列表({tool_names})之一的工具名，不能有其他内容
 Action Input: Action的输入参数.
- Observation: Action的执行结果.
+Observation: Action的执行结果.
 ... (这个 'Thought/Action/Action Input/Observation' 逻辑可以最多重复3次)
 
 Thought: 我现在知道最后的答案了.
@@ -212,7 +213,7 @@ flag = 8
 
 def update_image_summaries():
     global image_summaries
-    return image_summaries, gr.update(lines=5)
+    return image_summaries
 
 
 def update_flag():
@@ -280,9 +281,10 @@ def DailyNews():
     container_html = '<div class="new-container2">'
     for news in result:
         card_html = f"""
-            <div class="title2">
-                <a href="{news['url']}" target="_blank">{str(news['index'])+"、 "+news['title']}</a>
-            </div>
+        <div class="title2">
+            <span>{str(news['index']) + "、 " + news['title']}</span>
+            <a href="{news['url']}" target="_blank">跳转</a>
+        </div>
         """
         container_html += card_html
     container_html += '</div>'
@@ -330,7 +332,7 @@ css = """
 }
 
 .new-container2{
-    width: 300px;  /* 容器宽度 */
+    width: 400px;  /* 容器宽度 */
     height: 200px;  /* 容器高度 */
     overflow-y: auto;  /* 超出容器的内容显示滚动条 */
     border: 1px solid #ccc;  /* 给容器添加边框 */
@@ -446,21 +448,16 @@ def react(dict, space):
     agent_thought = []
     pattern = r"(.*?)\nThought: (.*?)\nAction: (.*?)\nAction Input: (.*)"
     for index in response['intermediate_steps']:
-        match = re.search(pattern, index[0].log, re.S) 
+        match = re.search(pattern, index[0].log, re.S)
         before_thought = match.group(1).strip() if match else ""
         thought = match.group(2).strip() if match else ""
         action = match.group(3).strip() if match else ""
         action_input = match.group(4).strip() if match else ""
-        agent_thought.append(
-            ChatMessage(
-                role="assistant",
-                content=before_thought
-            )
-        )
         agent_thought.append(
             ChatMessage(
                 role="assistant",
                 content=thought,
+                metadata={"title": "💥" + before_thought}
             )
         )
         agent_thought.append(
@@ -524,13 +521,12 @@ with gr.Blocks(css=css, theme='soft') as demo:
                                        {
                                            "text": "据最新天文研究，火星的轨道将逐渐接近地球，最终成为地球的“第二月亮”。天文学家预测这一变化将在2025年发生，届时火星将在夜空中与月亮一样明亮，影响全球潮汐和生态平衡。"}],
                              chatbot=gr.Chatbot(label='聊天机器人',
-                                                avatar_images=("./image/human.png", "./image/bot.png"),
+                                                avatar_images=("../image/human.png", "../image/bot.png"),
                                                 type="messages"
                                                 ),
                              multimodal=True,
                              show_progress='full',
-                             type="messages",
-                             cache_examples = False
+                             type="messages"
                              )
             with gr.Row():
                 language_select = gr.Dropdown(["中文", "英文"], label="请选择要使用的语言", scale=1, value="中文")
@@ -542,14 +538,18 @@ with gr.Blocks(css=css, theme='soft') as demo:
         with gr.Column(scale=1):
             flag = gr.Image(label="新闻标签", type="numpy", visible=False)
             contrast_model = gr.Textbox(label="对比模型", lines=5)
-            img_info = gr.Textbox(label="提取到的信息", lines=1)
+            with gr.Accordion("展示提取到的图片信息", open=False):
+                img_info = gr.Textbox(label="提取到的信息", lines=5)
             # daily_news = gr.Textbox(label="每日热搜", lines=1)
-            daily_news = gr.HTML(label="每日热搜", show_label=True,container=True)
+            with gr.Accordion("展示每日热搜", open=False):
+                daily_news = gr.HTML(label="每日热搜", show_label=True, container=True)
             img_bn = gr.Button("显示图片信息")
             # flag_bn = gr.Button("展示标签信息")
             daily_bn = gr.Button("查看每日热搜")
             constract_bn = gr.Button("展示对比模型")
-        img_bn.click(update_image_summaries, [], [img_info, img_info])
+            # with gr.Accordion("See Details"):
+            #     gr.Markdown("lorem ipsum")
+        img_bn.click(update_image_summaries, [], img_info)
         # flag_bn.click(update_flag, [], flag)
         daily_bn.click(DailyNews, [], [daily_news])
         language_select.change(SelectLanguage, language_select, [])