Update app.py

app.py

@@ -30,7 +30,7 @@ def run(DATASET_ID):
   lookml = None
   lookml_explore = None
   dataset = model_response(f'''You are an expert in BQ public datasets. Generate a dataset schema related to {DATASET_ID}. You need to come up with atleast 5 tables with each table
-containing atleast 10 columns along with their descriptions.''')
+containing atleast 10 columns along with their descriptions.Ensure that these tables have columns that talk about data quality issues.''')
   dataset_description = model_response(f'''Generate a succinct 3-4 line description of the dataset: {dataset}.''')
   yield dataset_description, None, None, None, None, None, None
 
@@ -77,6 +77,9 @@ containing atleast 10 columns along with their descriptions.''')
   queries = generate_dataset_queries(dataset, generated_glossary, schema_relations)
   yield dataset_description, py, schema_relations, generated_glossary, queries, None, None
   lookml = generate_lookml(dataset, generated_glossary, schema_relations)
+  yield dataset_description, py, schema_relations, generated_glossary, queries, lookml 
+  lookml_explore = model_response(f'''Given the dataset: {dataset}, schema relationships: {py} and graph:{schema_relations}, generate a data preparation pipeline that
+  fixes the possible data quality issues across the tables in the dataset.''')
   yield dataset_description, py, schema_relations, generated_glossary, queries, lookml, lookml_explore
   return dataset_description, py, schema_relations, generated_glossary, queries, lookml, lookml_explore
 
@@ -84,11 +87,13 @@ containing atleast 10 columns along with their descriptions.''')
 def wrapper(dataset_id):
     for outputs in run(dataset_id):
         yield (
+            outputs[0],
             outputs[1],
             outputs[2],  # Schema Relationships
             outputs[3],  # Generated Glossary
             outputs[4],  # Queries
-            outputs[5] # LookML Model
+            outputs[5], # LookML Model
+            outputs[6]
         )
 
 import gradio as gr
@@ -101,7 +106,8 @@ iface = gr.Interface(
             gr.Markdown(label="Schema Relationships"),
             gr.Markdown(label="Generated Glossary"),
             gr.Textbox(label="Queries"),
-            gr.Markdown(label="LookML Model")
+            gr.Markdown(label="LookML Model").
+            gr.Markdown(label="Data preparation pipeline")
         ],
         live=False,
         theme = gr.themes.Ocean(),