Update app.py

app.py

@@ -25,11 +25,13 @@ available_model_versions = ["Demographic Base"]
 
 default_model_version = "Demographic Base"
 
+def generate_code_example(dataset, attributes, model_version):
+    code = f"""
+import fantix
+"""
+    return code.strip()
 
 def load_dataset(dataset, model_version):
-    """
-    Add 'model_version' parameter to utilize it for data loading if necessary.
-    """
     return pd.DataFrame(
         {
             "Name": ["John", "Doe", "Jane", "Smith"],
@@ -42,15 +44,11 @@ def load_dataset(dataset, model_version):
 
 
 def predict(dataset, attributes, access_token, model_version):
-    """
-    Updated to accept and utilize 'model_version'.
-    """
     if not access_token:
         raise gr.Error(
             "Access token missing or invalid. Please ensure you have a valid access token."
         )
 
-    # Use model_version here if needed
     prediction_message = "200 predictions made in 2.5 seconds."
     prediction_results = pd.DataFrame(
         {
@@ -62,10 +60,8 @@ def predict(dataset, attributes, access_token, model_version):
 
 
 def load_dataset_and_predict(dataset, attributes, access_token, model_version):
-    """
-    Updated to pass 'model_version' to 'load_dataset' and 'predict'.
-    """
     loaded_data = load_dataset(dataset, model_version)
+    code_example = generate_code_example(dataset, attributes, model_version)
 
     if access_token:
         prediction_message, prediction_results = predict(
@@ -75,7 +71,7 @@ def load_dataset_and_predict(dataset, attributes, access_token, model_version):
         prediction_message = "No access token provided, prediction skipped."
         prediction_results = None
 
-    return loaded_data, prediction_message, prediction_results
+    return loaded_data, prediction_message, prediction_results, code_example
 
 
 interface_theme = gr.themes.Default()
@@ -115,6 +111,9 @@ with gr.Blocks(theme=interface_theme) as demo:
     gr.Markdown("### Dataset Preview")
     dataset_preview = gr.Dataframe()
 
+    with gr.Accordion("Code Example", open=False):
+        code_example = gr.Code()
+        
     predict_button = gr.Button("Predict Attributes")
 
     gr.Markdown("### Prediction Results")
@@ -122,9 +121,9 @@ with gr.Blocks(theme=interface_theme) as demo:
     prediction_label = gr.Markdown("")
 
     selected_dataset.change(
-        fn=load_dataset,
-        inputs=[selected_dataset, model_version],
-        outputs=dataset_preview,
+        fn=generate_code_example,
+        inputs=[selected_dataset, selected_attributes, model_version],
+        outputs=code_example,
     )
 
     predict_button.click(
@@ -132,11 +131,16 @@ with gr.Blocks(theme=interface_theme) as demo:
         inputs=[selected_dataset, selected_attributes, access_token, model_version],
         outputs=[prediction_label, prediction_preview],
     )
+    predict_button.click(
+        fn=generate_code_example,
+        inputs=[selected_dataset, selected_attributes, model_version], 
+        outputs=code_example,
+    )
 
     demo.load(
         fn=load_dataset_and_predict,
         inputs=[selected_dataset, selected_attributes, access_token, model_version],
-        outputs=[dataset_preview, prediction_label, prediction_preview],
+        outputs=[dataset_preview, prediction_label, prediction_preview, code_example],
     )
 
-demo.launch()
+demo.launch()