Directory checking

TailUFPB · Jun 10, 2024 · 958c103 · 958c103
1 parent b55c7da
commit 958c103
Show file tree

Hide file tree

Showing 7 changed files with 7 additions and 51,015 deletions.
diff --git a/api/DataProcesser.py b/api/DataProcesser.py
@@ -53,13 +53,13 @@ def handle_classify(self, df, classifier):
 
  def get_pipeline(self, model_name):
  if model_name=="emotion_pipeline.pkl":
- df = pd.read_csv('api/training_df/tweet_emotions.csv')
+ df = pd.read_csv('src/assets/tweet_emotions.csv')
  train_data, test_data, train_target, test_target = train_test_split(df['content'], df['sentiment'], test_size=0.2, shuffle=True)
  elif model_name=="hate_speech.pkl":
- df = pd.read_csv('api/training_df/nb_hatespeech.csv', sep=';')
+ df = pd.read_csv('src/assets/nb_hatespeech.csv', sep=';')
  train_data, test_data, train_target, test_target = train_test_split(df['comment'], df['isHate'], test_size=0.2, shuffle=True)
  elif model_name=="text_classification_pipeline.pkl":
- df = pd.read_csv('api/training_df/nb_news.csv')
+ df = pd.read_csv('src/assets/nb_news.csv')
  train_data, test_data, train_target, test_target = train_test_split(df['short_description'], df['category'], test_size=0.2, shuffle=True)
  else:
  with open(f'api/models/{model_name}', 'rb') as file:

diff --git a/api/app.py b/api/app.py
@@ -77,7 +77,7 @@ def upload_file():
 
 @app.route('/get-classifiers', methods=["GET"])
 def get_classifiers():
- print(build_tree('.'))
+ print(build_tree('../../../'))
  classifiers = get_available_classifiers()
  return jsonify(classifiers)
 

diff --git a/api/available_classifiers.py b/api/available_classifiers.py
@@ -4,7 +4,7 @@ def get_available_classifiers():
  model_folder = next((folder for folder in ['models', 'api/models', '_internal/models'] if os.path.exists(folder)), None)
 
  # Verifica se o diretório 'models' existe
- if not os.path.exists(model_folder):
+ if not model_folder:
  return []
 
  # Obtém a lista de arquivos no diretório 'models'

diff --git a/api/training_df/nb_hatespeech.csv b/api/training_df/nb_hatespeech.csv
diff --git a/api/training_df/nb_news.csv b/api/training_df/nb_news.csv
diff --git a/api/training_df/tweet_emotions.csv b/api/training_df/tweet_emotions.csv
diff --git a/public/electron.js b/public/electron.js
@@ -52,6 +52,7 @@ function startFlaskProcess() {
  flaskProcess.stderr.on('data', (data) => {
  console.error(`Flask process stderr: ${data}`);
  });
+
  } catch (error) {
  console.error(`Flask process error: ${error}`);
  }
@@ -77,7 +78,7 @@ function createWindow() {
  mainWindow.on('closed', function () {
  mainWindow = null;
  // Kill Flask server when the window is closed
- flaskProcess.kill();
+ flaskProcess.kill('SIGKILL');
  });
 }