fix path in app.py

.gitignore

@@ -1,5 +1,7 @@
 __pycache__
 
+._*
+
 # Byte-compiled / optimized / DLL files
 logs/
 

app.py

@@ -1,6 +1,9 @@
 import gradio as gr
 import pandas as pd
 import os
+import re
+import csv
+import time
 # # https://discuss.huggingface.co/t/issues-with-sadtalker-zerogpu-spaces-inquiry-about-community-grant/110625/10
 # if os.environ.get("SPACES_ZERO_GPU") is not None:
 #     import spaces
@@ -16,20 +19,18 @@ from qatch.connectors.sqlite_connector import SqliteConnector
 from qatch.generate_dataset.orchestrator_generator import OrchestratorGenerator
 from qatch.evaluate_dataset.orchestrator_evaluator import OrchestratorEvaluator
 from prediction import ModelPrediction
-import utils_get_db_tables_info
+from utils_get_db_tables_info import utils_extract_db_schema_as_string
 import utilities as us
-import time
 import plotly.express as px
 import plotly.graph_objects as go
 import plotly.colors as pc
-import re
-import csv
 
-# @spaces.GPU
-# def model_prediction():
-#   pass
 pnp_path = os.path.join("data", "evaluation_p_metrics.csv")
 
+us.check_and_create_dir('data/data_interface/')
+us.check_and_create_dir('data/data_results/')
+us.check_and_create_dir('data/databases/')
+
 with open('style.css', 'r') as file:
     css = file.read()
 

models.csv

@@ -1,3 +1,7 @@
 name,code,price,image_path
-Meta-LLAMA,meta-llama/Llama-3.3-70B-Instruct-Turbo-Free,0.0,./models_logo/LLAMA.jpg
-DEEPSEEK,deepseek-ai/DeepSeek-R1-Distill-Llama-70B-free,0.0,./models_logo/DEEPSEEK.jpg
+Meta LLAMA-8,llama-8,0.0,./models_logo/LLAMA.jpg
+DeepSeek-R1-Distill-Llama-70B,DeepSeek-R1-Distill-Llama-70B,0.0,./models_logo/DEEPSEEK.jpg
+CHAT GPT-3.5,gpt-3.5,0.0,models_logo/CHATGPT3_5.png
+CHAT GPT-4 mini,gpt-4o-mini,0.0,./models_logo/CHATGPT4mini.png
+CHAT GPT-o1-mini,o1-mini,0.0,./models_logo/CHATGPTo1mini.png
+QwQ,QwQ,0.0,./models_logo/QWQ.png

test_prediction.py

@@ -1,4 +1,5 @@
 from prediction import ModelPrediction
+import utilities as us
 
 
 def main():
@@ -8,5 +9,7 @@ def main():
 
 
 if __name__ == "__main__":
-    main()
-    # do something with prompt
+    # main()
+    us.check_and_create_dir('data/data_interface/')
+    us.check_and_create_dir('data/data_results/')
+    us.check_and_create_dir('data/databases/')

utilities.py

@@ -129,4 +129,12 @@ def extract_tables_dict(pnp_path):
             if tbl_name not in tables_dict:
                 tables_dict[tbl_name] = []
             #tables_dict[tbl_name].append(row)
-    return tables_dict
+    return tables_dict
+
+def check_and_create_dir(db_path):
+    # Check if the folder exists, and create it if it doesn't
+    if not os.path.exists(db_path):
+        os.makedirs(db_path)
+        print(f"Folder created: {db_path}")
+    else:
+        print(f"Folder already exists: {db_path}")

utils_get_db_tables_info.py

@@ -2,7 +2,6 @@ import os
 import sqlite3
 import re
 
-
 def utils_extract_db_schema_as_string(
     db_id, base_path, normalize=False, sql: str | None = None
 ):