Update app.py

app.py

@@ -357,6 +357,23 @@ def main():
 
         with st.expander("Download SDK Code", expanded=False):
             config_text = f"""
+        #!pip install -Uqq git+https://github.com/gretelai/navigator-helpers.git
+
+        import logging
+        import sys
+        import pandas as pd
+
+        from navigator_helpers import DataAugmentationConfig, DataAugmenter
+
+        # Configure the logger
+        logger = logging.getLogger()
+        logger.setLevel(logging.INFO)
+
+        DATASET = "YOUR_DATASET"
+        API_KEY = "YOUR_API_KEY"
+
+        df = pd.read_csv(DATASET)
+
         # Create the data augmentation configuration
         config = DataAugmentationConfig(
             input_fields={st.session_state.selected_fields},
@@ -367,7 +384,7 @@ def main():
             temperature={temperature},
             max_tokens_instruction={max_tokens_instruction},
             max_tokens_response={max_tokens_response},
-            api_key=YOUR_GRETEL_API_KEY,
+            api_key=API_KEY,
             navigator_tabular="{navigator_tabular}",
             navigator_llm="{navigator_llm}",
             co_teach_llms={co_teach_llms},