Update app.py

app.py

@@ -338,10 +338,13 @@ label_mapping_reverse = {value: key for key, value in label_mapping.items()}
 
 
 
-# fetch data
-def fetch_and_update_training_data(train_file):
+# create function for fetch data
+def fetch_and_update_training_data():
     model_name = f"{arg7}"
     export_type = "sentence"
+
+    print(arg2)
+    print(arg7)
     
     request_url = f"http://20.247.235.135/AiBot/BE/api/PromptEnhancer/ExportPEList?modelName={model_name}&exportType={export_type}"
 
@@ -350,13 +353,13 @@ def fetch_and_update_training_data(train_file):
     if response.status_code == 200:
         try:
             # Save response as CSV
-            with open(train_file + ".csv", "w", encoding="utf-8") as file:
+            with open(arg2 + ".csv", "w", encoding="utf-8") as file:
                 file.write(response.text)
 
             print("Training file updated successfully.")
 
             # Read the updated CSV into a DataFrame
-            return pd.read_csv(train_file + ".csv")
+            return pd.read_csv(arg2 + ".csv")
 
         except Exception as e:
             print(f"Error processing the response: {e}")
@@ -364,7 +367,9 @@ def fetch_and_update_training_data(train_file):
     else:
         print(f"Error fetching data: {response.status_code}")
         return None
-
+        
+# run the function     
+fetch_and_update_training_data()
 
 #Function to classify user input
 def classify_user_input():