test chatbot

app.py

@@ -3,6 +3,8 @@ import os
 import datetime
 import re
 import pandas as pd
+from sqlalchemy import true
+from chatbot.chat_main import create_financial_chatbot
 import globals as g
 from service.mysql_service import get_companys, insert_company, get_company_by_name
 from service.chat_service import get_analysis_report, get_stock_price_from_bailian, search_company, search_news, get_invest_suggest, chat_bot
@@ -14,6 +16,28 @@ from service.three_year_table_tool import build_table_format
 from service.three_year_tool import  process_financial_data_with_metadata
 from service.tool_processor import get_stock_price
 
+get_companys_state = True
+# JavaScript代码用于读取和存储数据
+js_code = """
+function handleStorage(operation, key, value) {
+    if (operation === 'set') {
+        localStorage.setItem(key, value);
+        return `已存储: ${key} = ${value}`;
+    } else if (operation === 'get') {
+        let storedValue = localStorage.getItem(key);
+        if (storedValue === null) {
+            return `未找到键: ${key}`;
+        }
+        return `读取到: ${key} = ${storedValue}`;
+    } else if (operation === 'clear') {
+        localStorage.removeItem(key);
+        return `已清除: ${key}`;
+    } else if (operation === 'clearAll') {
+        localStorage.clear();
+        return '已清除所有数据';
+    }
+}
+"""
 custom_css = """
 /* 匹配所有以 gradio-container- 开头的类 */
 div[class^="gradio-container-"],
@@ -109,14 +133,20 @@ def get_stock_code_by_company_name(company_name):
 
 # 创建一个简单的函数来获取公司列表
 def get_company_list_choices():
+    choices = []
+    print(f"Getting init add company list choices...{get_companys_state}")
+    if not get_companys_state:
+        return gr.update(choices=choices)
     try:
         companies_data = get_companys()
+        print(f"Getting init add company list choices...companies_data: {companies_data}")
         if isinstance(companies_data, pd.DataFrame) and not companies_data.empty:
             choices = [str(row.get('company_name', 'Unknown')) for _, row in companies_data.iterrows()]
         else:
             choices = []
     except:
         choices = []
+    
     return gr.update(choices=choices)
 
 # Sidebar service functions
@@ -257,6 +287,9 @@ def initialize_companies_map():
         
         # 从数据库获取公司数据
         companies_data = get_companys()
+        # companies_data = window.cachedCompanies or []
+
+
         print(f"Companies data from DB: {companies_data}")
         
         # 如果数据库中有公司数据，则添加到映射中（去重）
@@ -557,33 +590,45 @@ def create_header():
         with gr.Column(scale=2):
             gr.Markdown(current_time, elem_classes=["text-sm-top-time"])
 
-def create_company_list():
-    """创建公司列表组件"""
-    # 获取公司列表数据
-    try:
-        companies_data = get_companys()
-        if isinstance(companies_data, pd.DataFrame) and not companies_data.empty:
-            choices = [str(row.get('company_name', 'Unknown')) for _, row in companies_data.iterrows()]
-        else:
-            choices = []
-    except:
-        choices = []
-    
-    # 添加默认公司选项
-    if not choices:
-        choices = []
-    
-    # 使用Radio组件显示公司列表，不显示标签
+def create_company_list(get_companys_state):
     company_list = gr.Radio(
-        choices=choices,
+        choices=[],
         label="",
         interactive=True,
         elem_classes=["company-list-container"],
         container=False,  # 不显示外部容器边框
         visible=True
-    )
+      ) 
+    if (get_companys_state == False):
+        return company_list
+    else:
+        """创建公司列表组件"""
+        # 获取公司列表数据
+        try:
+            companies_data = get_companys()
+            print(f"创建公司列表组件 - Companies data: {companies_data}")
+            if isinstance(companies_data, pd.DataFrame) and not companies_data.empty:
+                choices = [str(row.get('company_name', 'Unknown')) for _, row in companies_data.iterrows()]
+            else:
+                choices = []
+        except:
+            choices = []
+        
+        # 添加默认公司选项
+        if not choices:
+            choices = []
+        
+        # 使用Radio组件显示公司列表，不显示标签
+        company_list = gr.Radio(
+            choices=choices,
+            label="",
+            interactive=True,
+            elem_classes=["company-list-container"],
+            container=False,  # 不显示外部容器边框
+            visible=True
+        )
     
-    return company_list
+        return company_list
 
 def create_company_selector():
     """创建公司选择器组件"""
@@ -722,6 +767,7 @@ def format_financial_metrics(data: dict, prev_data: dict = None) -> list:  # pyr
 
     return result
 
+
 def create_sidebar():
     """创建侧边栏组件"""
     # 初始化 companies_map
@@ -732,8 +778,16 @@ def create_sidebar():
         with gr.Group(elem_classes=["card"]):
             gr.Markdown("### Select Company", elem_classes=["card-title", "left-card-title"])
             with gr.Column():
+                company_list = create_company_list(get_companys_state)
+                
                 # 创建公司列表
-                company_list = create_company_list()
+                # if not get_companys_state:
+                #     getCompanyFromStorage = gr.Button("读取")
+                #     getCompanyFromStorage.click(
+                #         fn=create_company_list(True),
+                #         inputs=[],
+                #         outputs=[company_list, status_message]
+                #     )
                 
                 # 创建公司选择器
                 company_input, status_message, company_modal = create_company_selector()
@@ -1617,19 +1671,18 @@ def create_tab_content(tab_name, company_name):
                     gr.Markdown("Report Preview", elem_classes=["font-medium", "mb-3"])
                     # 这里将显示报告预览
 
-from chatbot.chat_main import create_financial_chatbot
-
 def create_chat_panel():
     """创建聊天面板组件"""
     with gr.Column(elem_classes=["chat-panel"]):
-        chat_component = create_financial_chatbot()
-        chat_component.render()
         # 聊天头部
         # with gr.Row(elem_classes=["p-4", "border-b", "border-gray-200", "items-center", "gap-2"]):
         #     gr.Markdown("🤖", elem_classes=["text-xl", "text-blue-600"])
         #     gr.Markdown("Financial Assistant", elem_classes=["font-medium"])
         
         # 聊天区域
+         # 一行代码嵌入！
+            chat_component = create_financial_chatbot()
+            chat_component.render()
         # chatbot = gr.Chatbot(
         #     value=[
         #         {"role": "assistant", "content": "I'm your financial assistant, how can I help you today?"},
@@ -1650,7 +1703,7 @@ def create_chat_panel():
         #     container=False, 
         # )
         
-        # 输入区域
+        # # 输入区域
         # with gr.Row(elem_classes=["border-t", "border-gray-200", "gap-2"]):
         #     msg = gr.Textbox(
         #         placeholder="Ask a financial question...",
@@ -1737,7 +1790,7 @@ def main():
                     create_metrics_dashboard()
                     
                     with gr.Row(elem_classes=["main-content-box"]):
-                        with gr.Column(scale=9):
+                        with gr.Column(scale=8):
                             # Tab内容
                             with gr.Tabs():
                                 with gr.TabItem("Invest Suggest", elem_classes=["tab-item"]):
@@ -1850,7 +1903,7 @@ def main():
                                     )
                                 # with gr.TabItem("Comparison", elem_classes=["tab-item"]):
                                 #     create_tab_content("comparison")
-                        with gr.Column(scale=1):
+                        with gr.Column(scale=2):
                             # 聊天面板
                             create_chat_panel()
             
@@ -1859,7 +1912,7 @@ def main():
                 fn=get_company_list_choices,
                 inputs=[],
                 outputs=[company_list_component],
-                concurrency_limit=None
+                concurrency_limit=None,
             )
             
             # 绑定公司选择事件到状态更新