fix: add schema database for single query

app.py

@@ -620,13 +620,6 @@ with gr.Blocks(theme='shivi/calm_seafoam', css_paths='style.css', js=js_func) as
                 #TODO s
                 target_df = orchestrator_generator.generate_dataset(connector=input_data['data']['db'], tables_to_include=input_data['data']['selected_tables'])
                 #target_df = orchestrator_generator.generate_dataset(connector=input_data['data']['db'], tables_to_includes=None)
-    
-                schema_text = utils_get_db_tables_info.utils_extract_db_schema_as_string(
-                    db_id = input_data["db_name"], 
-                    base_path = input_data["data_path"], 
-                    normalize=False, 
-                    sql=None
-                )
 
                 predictor = ModelPrediction()
         
@@ -649,6 +642,14 @@ with gr.Blocks(theme='shivi/calm_seafoam', css_paths='style.css', js=js_func) as
                         yield gr.Image(), gr.Markdown(load_text), gr.Markdown(display_question), gr.Markdown(), metrics_conc, *[predictions_dict[model]for model in model_list]
                         start_time = time.time()
                         samples = us.generate_some_samples(input_data['data']['db'], row["tbl_name"])
+
+                        schema_text = utils_get_db_tables_info.utils_extract_db_schema_as_string(
+                        db_id = row["db_name"], 
+                        base_path = row["data_path"], 
+                        normalize=False, 
+                        sql=row["query"]
+                        )
+
                         prompt_to_send = us.prepare_prompt(input_data["prompt"], question, schema_text, samples)
                         #PREDICTION SQL
                         if prompt_to_send == prompt_default:

utils_get_db_tables_info.py

@@ -15,13 +15,6 @@ def utils_extract_db_schema_as_string(
     :param sql: Optional SQL query to filter specific tables.
     :return: Schema of the database as a single string.
     """
-    #db_path = os.path.join(base_path, db_id, f"{db_id}.sqlite")
-
-    # Connect to the SQLite database
-
-    #if not os.path.exists(db_path):
-    #    raise FileNotFoundError(f"Database file not found at: {db_path}")
-
     connection = sqlite3.connect(base_path)
     cursor = connection.cursor()