fix multi-line bug, add SDK code to download zip

app.py

@@ -11,8 +11,7 @@ import requests
 import streamlit as st
 from datasets import load_dataset
 from gretel_client import Gretel
-from navigator_helpers import (DataAugmentationConfig, DataAugmenter,
-                               StreamlitLogHandler)
+from navigator_helpers import DataAugmentationConfig, DataAugmenter, StreamlitLogHandler
 
 # Create a StringIO buffer to capture the logging output
 log_buffer = StringIO()
@@ -363,7 +362,9 @@ def main():
 
         with st.expander("Download SDK Code", expanded=False):
             st.markdown("### Ready to generate data at scale?")
-            st.write("Get started with your current configuration using the SDK code below:")
+            st.write(
+                "Get started with your current configuration using the SDK code below:"
+            )
 
             config_text = f"""
         #!pip install -Uqq git+https://github.com/gretelai/navigator-helpers.git
@@ -397,8 +398,8 @@ def main():
             navigator_tabular="{navigator_tabular}",
             navigator_llm="{navigator_llm}",
             co_teach_llms={co_teach_llms},
-            instruction_format_prompt="{instruction_format_prompt}",
-            response_format_prompt="{response_format_prompt}"
+            instruction_format_prompt='''{instruction_format_prompt}''',
+            response_format_prompt='''{response_format_prompt}'''
         )
 
         # Create the data augmenter and perform augmentation
@@ -563,17 +564,23 @@ def main():
                     with open(augmented_data_file_path, "w") as augmented_data_file:
                         augmented_data_file.write(augmented_data_jsonl)
 
-                # Create a ZIP file containing the logs and augmented data
+                # Write the SDK code to a file
+                sdk_file_path = os.path.join(temp_dir, "data_augmentation_code.py")
+                with open(sdk_file_path, "w") as sdk_file:
+                    sdk_file.write(config_text)
+
+                # Create a ZIP file containing the logs, augmented data, and SDK code
                 zip_file_path = os.path.join(temp_dir, "augmentation_results.zip")
                 with zipfile.ZipFile(zip_file_path, "w") as zip_file:
                     zip_file.write(log_file_path, "complete_logs.jsonl")
                     if augmented_data_jsonl:
                         zip_file.write(augmented_data_file_path, "augmented_data.jsonl")
+                    zip_file.write(sdk_file_path, "data_augmentation_code.py")
 
                 # Download the ZIP file
                 with open(zip_file_path, "rb") as zip_file:
                     st.download_button(
-                        label="💾 Download Synthetic Data and Logs",
+                        label="💾 Download Synthetic Data, Logs, and SDK Code",
                         data=zip_file.read(),
                         file_name="gretel_synthetic_data.zip",
                         mime="application/zip",