Update app.py

app.py

@@ -26,8 +26,38 @@ available_model_versions = ["Demographic Base"]
 default_model_version = "Demographic Base"
 
 def generate_code_example(dataset, attributes, model_version):
+    attributes_with_types = ", ".join([f"fantix.type.{attr.upper().replace(' ', '_')}" for attr in attributes])
+    prediction_columns = ", ".join([f'"{attr}"' for attr in attributes])
+    
     code = f"""
 import fantix
+import pandas as pd
+
+client = fantix.Client(api_key="YOUR_API_KEY")
+
+df = pd.DataFrame(
+    {
+        {
+            "Name": ["John", "Doe", "Jane", "Smith"],
+            "Age": [25, 30, 35, 40],
+            "Income": [50000, 60000, 70000, 80000],
+            "Marital Status": ["Single", "Married", "Single", "Married"],
+            "Education": ["High School", "Bachelor", "Master", "PhD"],
+        }
+    }
+)
+
+response = client.predict(
+    df=df,
+    attributes=[
+        fantix.type.INCOME,
+        fantix.type.AGE,
+        fantix.type.MARITAL_SATUS,
+        fantix.type.EDUCATION,
+    ],
+    prediction_columns=[{attributes_with_types}],
+    model_version='{model_version}'',
+)
 """
     return code.strip()
 
@@ -112,7 +142,7 @@ with gr.Blocks(theme=interface_theme) as demo:
     dataset_preview = gr.Dataframe()
 
     with gr.Accordion("Code Example", open=False):
-        code_example = gr.Code()
+        code_example = gr.Code(language="python")
         
     predict_button = gr.Button("Predict Attributes")