Add style column to examples

app.py

@@ -20,8 +20,11 @@ NOTES = '''
 FOOTER = '<img id="visitor-badge" alt="visitor badge" src="https://visitor-badge.glitch.me/badge?page_id=THUDM.CogView2" />'
 
 
-def set_example_text(example: list) -> dict:
-    return gr.Textbox.update(value=example[0])
+def set_example_text(example: list) -> list[dict]:
+    return [
+        gr.Textbox.update(value=example[0]),
+        gr.Dropdown.update(value=example[1]),
+    ]
 
 
 def main():
@@ -67,8 +70,11 @@ def main():
                                            value=8,
                                            label='Number of Images')
                     with open('samples.txt') as f:
-                        samples = [[line.strip()] for line in f.readlines()]
-                    examples = gr.Dataset(components=[text], samples=samples)
+                        samples = [
+                            line.strip().split('\t') for line in f.readlines()
+                        ]
+                    examples = gr.Dataset(components=[text, style],
+                                          samples=samples)
                     run_button = gr.Button('Run')
 
             with gr.Column():

samples.txt

@@ -1,9 +1,9 @@
-A lion teacher wearing a suit is in front of a blackboard.
-A robot is riding under the blue and cloudy sky.
-A lion man is typing in the office.
-A pirate captain with a skull.
-Earth in the eye.
-A magnificent church. Sketch.
-Mount Fuji, cherry blossom and Akita dog. Oil painting.
-A tiger with angel's wings.
-A fox is sitting on the books.
+A lion teacher wearing a suit is in front of a blackboard.	mainbody
+A robot is riding under the blue and cloudy sky.	mainbody
+A lion man is typing in the office.	mainbody
+A pirate captain with a skull.	mainbody
+Earth in the eye.	mainbody
+A magnificent church.	sketch
+Mount Fuji, cherry blossom and Akita dog.	oil
+A tiger with angel's wings.	mainbody
+A fox is sitting on the books.	mainbody