chore: add new parameters for the prediction in app.py

app.py

@@ -728,13 +728,22 @@ with gr.Blocks(theme='shivi/calm_seafoam', css_paths='style.css', js=js_func) as
                             db_id = input_data["db_name"], 
                             base_path = input_data["data_path"], 
                             normalize=False, 
-                            sql=row["query"]
+                            sql=row["query"],
+                            get_insert_into=True
                             )
                             
-                            prompt_to_send = us.prepare_prompt(input_data["prompt"], question, schema_text, samples)
+                            #prompt_to_send = us.prepare_prompt(input_data["prompt"], question, schema_text, samples)
+                            prompt_to_send = input_data["prompt"]
                             #PREDICTION SQL
 
-                            response = predictor.make_prediction(question=question, db_schema=schema_text, model_name=model, prompt=f"{prompt_to_send}")
+                            # TODO add button for QA or SP and pass to .make_prediction parameter TASK
+                            response = predictor.make_prediction(
+                                question=question,
+                                db_schema=schema_text,
+                                model_name=model,
+                                prompt=f"{prompt_to_send}",
+                                task="SP"  # TODO change accordingly
+                            )
                             prediction = response['response_parsed']
                             price = response['cost']
                             answer = response['response']