fix unchoose file error

Canner · Jul 8, 2024 · 97cf028 · 97cf028
1 parent b707f4b
commit 97cf028
Showing 1 changed file with 23 additions and 18 deletions.
diff --git a/wren-ai-service/eval/data_curation/app.py b/wren-ai-service/eval/data_curation/app.py
@@ -61,27 +61,32 @@ def on_click_generate_question_sql_pairs(llm_client: AsyncClient):
 
 def on_click_setup_uploaded_file():
     uploaded_file = st.session_state.get("uploaded_mdl_file")
-    match = re.match(
-        r".+_(" + "|".join(DATA_SOURCES) + r")_mdl\.json$",
-        uploaded_file.name,
-    )
-    if not match:
-        st.error(
-            f"the file name must be [xxx]_[datasource]_mdl.json, now we support these datasources: {DATA_SOURCES}"
+    if uploaded_file:
+        match = re.match(
+            r".+_(" + "|".join(DATA_SOURCES) + r")_mdl\.json$",
+            uploaded_file.name,
         )
-        st.stop()
+        if not match:
+            st.error(
+                f"the file name must be [xxx]_[datasource]_mdl.json, now we support these datasources: {DATA_SOURCES}"
+            )
+            st.stop()
 
-    data_source = match.group(1)
-    st.session_state["data_source"] = data_source
-    st.session_state["mdl_json"] = orjson.loads(
-        uploaded_file.getvalue().decode("utf-8")
-    )
+        data_source = match.group(1)
+        st.session_state["data_source"] = data_source
+        st.session_state["mdl_json"] = orjson.loads(
+            uploaded_file.getvalue().decode("utf-8")
+        )
 
-    st.session_state["connection_info"] = {
-        "project_id": os.getenv("bigquery.project-id"),
-        "dataset_id": os.getenv("bigquery.dataset-id"),
-        "credentials": os.getenv("bigquery.credentials-key"),
-    }
+        st.session_state["connection_info"] = {
+            "project_id": os.getenv("bigquery.project-id"),
+            "dataset_id": os.getenv("bigquery.dataset-id"),
+            "credentials": os.getenv("bigquery.credentials-key"),
+        }
+    else:
+        st.session_state["data_source"] = None
+        st.session_state["mdl_json"] = None
+        st.session_state["connection_info"] = None
 
 
 def on_change_sql(i: int, key: str):