Update app.py

app.py

@@ -41,5 +41,58 @@ def get_results(input_text, target_rhyme, beam_width=20, num_candidates=4000):
 # # 长韵脚需要更广的搜索空间
 # print_results(rhymer, text="我还是觉得音乐的[M][M][M][M][M]", target_rhyme="背靠背钻研")
 # print_results(rhymer, text="我还是觉得音乐的[M][M][M][M][M]", target_rhyme="背靠背钻研", beam_width=300)
-demo = gr.Interface(fn=get, inputs="text", outputs="text")
+demo = gr.Interface(
+    fn=get_results,
+    inputs=[
+        gr.Textbox(
+            label="Input Text",
+            placeholder="Enter your text with [M] markers for rhyme positions",
+            lines=3,
+            value="[M][M][M][M]"
+        ),
+        gr.Textbox(
+            label="Target Rhyme",
+            placeholder="Enter target rhyme pattern (Chinese characters or pinyin)",
+            value="乱比高低"
+        ),
+        gr.Slider(
+            minimum=1,
+            maximum=50,
+            value=20,
+            step=1,
+            label="Beam Width"
+        ),
+        gr.Slider(
+            minimum=100,
+            maximum=10000,
+            value=4000,
+            step=100,
+            label="Number of Candidates"
+        ),
+        gr.Slider(
+            minimum=1,
+            maximum=20,
+            value=10,
+            step=1,
+            label="Top Results to Show"
+        )
+    ],
+    outputs=gr.Textbox(
+        label="Generated Rhymes",
+        lines=15,
+        max_lines=20
+    ),
+    title="CiJiang Chinese Poetry Rhymer",
+    description="Generate Chinese poetry rhymes using the CiJiang library. Use [M] markers to indicate positions where rhymes should be generated. Supports both Chinese characters and pinyin as target rhymes.",
+    examples=[
+        ["[M][M][M][M]", "乱比高低", 20, 4000, 10],
+        ["[M][M][M][M]", "luan4 bi3 gao1 di1", 20, 4000, 10],
+        ["[M][M][M]", "收视率", 20, 4000, 10],
+        ["我总是天亮了[M][M]", "排队", 20, 4000, 10],
+        ["我们两个人的verse价值一辆[M][M][M][M]", "韩国低级", 20, 4000, 10],
+        ["报名嘻[M][M]", "八个", 20, 4000, 10],
+        ["我还是觉得音乐的[M][M][M][M][M]", "背靠背钻研", 20, 4000, 10],
+        ["我还是觉得音乐的[M][M][M][M][M]", "背靠背钻研", 300, 4000, 10]
+    ]
+)
 demo.launch()