yq 5 months ago
parent
commit
47d35c8157
2 changed files with 12 additions and 10 deletions
  1. 1 1
      app.py
  2. 11 9
      webui/utils.py

+ 1 - 1
app.py

@@ -83,7 +83,7 @@ with gr.Blocks() as demo:
                     download_code.click(fn=f_download_code, inputs=input_elems, outputs=download_code)
 
             demo.queue(default_concurrency_limit=5)
-            demo.launch(share=False, show_error=True, server_name="0.0.0.0", server_port=18067)
+            demo.launch(share=False, show_error=True, server_name="0.0.0.0", server_port=18075)
 
 if __name__ == "__main__":
     pass

+ 11 - 9
webui/utils.py

@@ -15,6 +15,7 @@ import pandas as pd
 
 from commom import f_read_file
 from config import BaseConfig
+from strategy_parse import StrategyParse
 from .manager import engine
 
 DATA_SUB_DIR = "data"
@@ -61,12 +62,14 @@ def _get_upload_data(data) -> pd.ExcelFile:
     excel = pd.ExcelFile(file_path)
     return excel
 
-def _get_upload_data_path(data) ->str:
+
+def _get_upload_data_path(data) -> str:
     base_dir = _get_base_dir(data)
     save_path = os.path.join(base_dir, DATA_SUB_DIR)
     file_path = _get_prefix_file(save_path, UPLOAD_DATA_PREFIX)
     return file_path
 
+
 def _get_node_func_dict(data) -> dict:
     node_func_dict_path = _get_save_path(data, BaseConfig.node_map_name)
     if not os.path.exists(node_func_dict_path):
@@ -178,11 +181,10 @@ def f_project_load(data):
             }
 
     return {
-                engine.get_elem_by_id("data_upload"): gr.update(visible=False),
-                engine.get_elem_by_id("code_view"): gr.update(visible=False),
-                engine.get_elem_by_id("download_code"): gr.update(visible=False),
-            }
-
+        engine.get_elem_by_id("data_upload"): gr.update(visible=False),
+        engine.get_elem_by_id("code_view"): gr.update(visible=False),
+        engine.get_elem_by_id("download_code"): gr.update(visible=False),
+    }
 
 
 def f_code_generate(data, progress=gr.Progress(track_tqdm=True)):
@@ -200,7 +202,7 @@ def f_code_generate(data, progress=gr.Progress(track_tqdm=True)):
         yield _reset_component_state()
 
     # 清空储存目录
-    # _clean_base_dir(data)
+    _clean_base_dir(data)
 
     yield _reset_component_state()
 
@@ -211,8 +213,8 @@ def f_code_generate(data, progress=gr.Progress(track_tqdm=True)):
     try:
         excel = _get_upload_data(data)
 
-        # strategy_parse = StrategyParse(**all_param)
-        # strategy_parse.f_parse_strategy(excel, progress)
+        strategy_parse = StrategyParse(**all_param)
+        strategy_parse.f_parse_strategy(excel, progress)
 
         excel.close()
     except Exception as msg: