RainbowGPT_Agent_ui

ZhuJD-China · Dec 14, 2023 · 4786c9f · 4786c9f
1 parent d880694
commit 4786c9f
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 41 deletions.
diff --git a/SQL_Agent/RainbowSQL_Agent_ui.py b/SQL_Agent/RainbowSQL_Agent_ui.py
@@ -166,7 +166,7 @@ def update_tables_list(host, username, password):
     return gr.Dropdown.update(choices=get_database_tables(host, username, password))
 
 
-with gr.Blocks(theme=seafoam) as RainbowGPT:
+with gr.Blocks(theme=seafoam) as RainbowSQL_Agent:
     with gr.Row():
         with gr.Column(scale=3):
             # 左侧列: 所有控件
@@ -226,5 +226,4 @@ def update_tables_list(host, username, password):
                 # css=".gradio-container {background-color: #f0f0f0;}"  # Add your desired background color here
             )
 
-RainbowGPT.queue().launch()
-# RainbowGPT.queue().launch(share=True)
+RainbowSQL_Agent.queue().launch()
diff --git a/Stock_Agent/StockGPTAnalysis_ui.py b/Stock_Agent/StockGPTAnalysis_ui.py
@@ -403,41 +403,53 @@ def get_stock_data(llm_options_checkbox_group, llm_options_checkbox_group_qwen,
     return gpt_response, qwen_response
 
 
-# Define Gradio interface for the main application
-with gr.Row():
-    with gr.Column():
-        llm_options = ["gpt-3.5-turbo-1106", "gpt-4-1106-preview",
-                       "gpt-4", "gpt-3.5-turbo-16k"]
-        llm_options_checkbox_group = gr.Dropdown(llm_options, label="GPT Model Select Options",
-                                                 value=llm_options[0])
-        llm_options_qwen = ["qwen-72b-chat"]
-        llm_options_checkbox_group_qwen = gr.Dropdown(llm_options_qwen, label="Qwen Model Select Options",
-                                                      value=llm_options_qwen[0])
-        http_proxy = gr.Textbox(value="http://localhost:7890", label="System Http Proxy")
-        market = gr.Textbox(lines=1, placeholder="请输入股票市场（sz或sh，示例：sz）", value="sz")
-        symbol = gr.Textbox(lines=1, placeholder="请输入股票代码(示例股票代码:002665)", value="002665")
-        stock_name = gr.Textbox(lines=1, placeholder="请输入股票名称(示例股票名称:首航高科): ", value="首航高科")
-        start_date = gr.Textbox(lines=1, placeholder="请输入K线历史数据查询起始日期（YYYYMMDD，示例：20230805）: ",
-                                value="20230805")
-        end_date = gr.Textbox(lines=1, placeholder="请输入K线历史数据结束日期（YYYYMMDD，示例：20231206）: ",
-                              value="20231212")
-        concept = gr.Textbox(lines=1, placeholder="请输入当前股票所属概念板块名称(示例：光热发电): ", value="光热发电")
-    with gr.Column():
-        gpt_response = gr.Textbox(label="GPT Response")
-        qwen_response = gr.Textbox(label="QWEN Response")
-
-custom_title = "StockGPT Analysis"
-custom_description = """
-<p>How to reach me: <a href='mailto:zhujiadongvip@163.com'>zhujiadongvip@163.com</a></p>
-"""
-# Define Gradio interface for the main application
-iface = gr.Interface(
-    fn=get_stock_data,
-    inputs=[llm_options_checkbox_group, llm_options_checkbox_group_qwen, market, symbol, stock_name,
-            start_date,
-            end_date, concept, http_proxy],
-    outputs=[gpt_response, qwen_response],
-    title=custom_title,
-    description=custom_description
-)
-iface.queue().launch()
+# 定义 Gradio 界面
+with gr.Blocks() as stockgpt_interface:
+    gr.Markdown("## StockGPT Analysis")
+    with gr.Row():
+        with gr.Column():
+            # 左侧列: 输入控件，两两排列
+            with gr.Row():
+                llm_options = ["gpt-3.5-turbo-1106", "gpt-4-1106-preview", "gpt-4", "gpt-3.5-turbo-16k"]
+                llm_options_checkbox_group = gr.Dropdown(llm_options, label="GPT Model Select Options",
+                                                         value=llm_options[0])
+                llm_options_qwen = ["qwen-72b-chat"]
+                llm_options_checkbox_group_qwen = gr.Dropdown(llm_options_qwen, label="Qwen Model Select Options",
+                                                              value=llm_options_qwen[0])
+
+            with gr.Row():
+                http_proxy = gr.Textbox(value="http://localhost:7890", label="System Http Proxy")
+                market = gr.Textbox(lines=1, placeholder="请输入股票市场（sz或sh，示例：sz）", label="Market", value="sz")
+
+            with gr.Row():
+                symbol = gr.Textbox(lines=1, placeholder="请输入股票代码(示例股票代码:002665)", label="Symbol",
+                                    value="002665")
+                stock_name = gr.Textbox(lines=1, placeholder="请输入股票名称(示例股票名称:首航高科): ",
+                                        label="Stock Name", value="首航高科")
+
+            with gr.Row():
+                start_date = gr.Textbox(lines=1, placeholder="请输入K线历史数据查询起始日期（YYYYMMDD，示例：20230805）: ",
+                                        label="Start Date", value="20230805")
+                end_date = gr.Textbox(lines=1, placeholder="请输入K线历史数据结束日期（YYYYMMDD，示例：20231206）: ",
+                                      label="End Date", value="20231212")
+
+            with gr.Row():
+                concept = gr.Textbox(lines=1, placeholder="请输入当前股票所属概念板块名称(示例：光热发电): ",
+                                     label="Concept", value="光热发电")
+
+        with gr.Column():
+            # 右侧列: 输出控件
+            gpt_response = gr.Textbox(label="GPT Response")
+            qwen_response = gr.Textbox(label="QWEN Response")
+
+    # 提交按钮
+    submit_button = gr.Button("Submit")
+    submit_button.click(
+        fn=get_stock_data,
+        inputs=[llm_options_checkbox_group, llm_options_checkbox_group_qwen, market, symbol, stock_name, start_date,
+                end_date, concept, http_proxy],
+        outputs=[gpt_response, qwen_response]
+    )
+
+# 启动 Gradio 界面
+stockgpt_interface.queue().launch()