Clémentine commited on
Commit
06c0fa4
·
1 Parent(s): 0b4b222
Files changed (2) hide show
  1. app.py +1 -1
  2. utils/io.py +8 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import time
3
  from apscheduler.schedulers.background import BackgroundScheduler
4
  import threading
5
  import globals
6
- from utils.io import initialize_models_providers_file, save_results, load_results, load_models_providers, get_results_table, load_models_providers_str, get_summary_stats
7
  from utils.jobs import run_single_job, run_multiple_jobs, launch_jobs, update_job_statuses, relaunch_failed_jobs
8
  from typing import List, Optional
9
 
 
3
  from apscheduler.schedulers.background import BackgroundScheduler
4
  import threading
5
  import globals
6
+ from utils.io import save_results, load_results, load_models_providers, get_results_table, load_models_providers_str, get_summary_stats
7
  from utils.jobs import run_single_job, run_multiple_jobs, launch_jobs, update_job_statuses, relaunch_failed_jobs
8
  from typing import List, Optional
9
 
utils/io.py CHANGED
@@ -56,9 +56,14 @@ def load_models_providers_str(file_path: str = globals.LOCAL_CONFIG_FILE) -> str
56
 
57
 
58
  def load_models_providers(file_path: str = globals.LOCAL_CONFIG_FILE) -> List[Tuple[str, str]]:
59
- """Load models and providers from text file."""
60
  models_providers = []
61
  try:
 
 
 
 
 
62
  with open(file_path, 'r') as f:
63
  for line in f:
64
  line = line.strip()
@@ -71,6 +76,8 @@ def load_models_providers(file_path: str = globals.LOCAL_CONFIG_FILE) -> List[Tu
71
  models_providers.append((model, provider))
72
  except Exception as e:
73
  print(f"Error loading model providers: {str(e)}")
 
 
74
  return models_providers
75
 
76
 
 
56
 
57
 
58
  def load_models_providers(file_path: str = globals.LOCAL_CONFIG_FILE) -> List[Tuple[str, str]]:
59
+ """Load models and providers from text file. Creates file if it doesn't exist."""
60
  models_providers = []
61
  try:
62
+ # Check if file exists, if not, create it
63
+ if not os.path.exists(file_path):
64
+ print(f"Config file {file_path} not found. Initializing...")
65
+ initialize_models_providers_file(file_path)
66
+
67
  with open(file_path, 'r') as f:
68
  for line in f:
69
  line = line.strip()
 
76
  models_providers.append((model, provider))
77
  except Exception as e:
78
  print(f"Error loading model providers: {str(e)}")
79
+ import traceback
80
+ traceback.print_exc()
81
  return models_providers
82
 
83