changed layout

app.py

@@ -70,8 +70,6 @@ def generate_poem(country, era, meter, theme, lang_type, poet, num_lines, num_po
 
     return "\n\n".join(poems)   
     
-
-
 meters = ['البسيط', 'التفعيله', 'الحداء', 'الخفيف', 'الدوبيت', 'الرجز', 'الرمل', 'السريع', 'السلسلة', 'الصخري', 'الطويل', 'الكامل', 'الكان كان', 'اللويحاني', 'المتدارك', 'المتقارب', 'المجتث', 'المديد', 'المسحوب', 'المضارع', 'المقتضب', 'المنسرح', 'المواليا', 'الموشح', 'الهجيني', 'الهزج', 'الوافر', 'بحر أحذ الكامل', 'بحر أحذ المديد', 'بحر أحذ الوافر', 'بحر البسيط', 'بحر التفعيله', 'بحر الخبب', 'بحر الخفيف', 'بحر الدوبيت', 'بحر الرجز', 'بحر الرمل', 'بحر السريع', 'بحر السلسلة', 'بحر الطويل', 'بحر القوما', 'بحر الكامل', 'بحر الكامل المقطوع', 'بحر المتدارك', 'بحر المتدارك المنهوك', 'بحر المتقارب', 'بحر المجتث', 'بحر المديد', 'بحر المضارع', 'بحر المقتضب', 'بحر المنسرح', 'بحر المواليا', 'بحر الهزج', 'بحر الوافر', 'بحر تفعيلة الرجز', 'بحر تفعيلة الرمل', 'بحر تفعيلة الكامل', 'بحر تفعيلة المتقارب', 'بحر مجزوء البسيط', 'بحر مجزوء الخفيف', 'بحر مجزوء الدوبيت', 'بحر مجزوء الرجز', 'بحر مجزوء الرمل', 'بحر مجزوء الرمل ', 'بحر مجزوء السريع', 'بحر مجزوء الطويل', 'بحر مجزوء الكامل', 'بحر مجزوء المتدارك', 'بحر مجزوء المتقارب', 'بحر مجزوء المجتث', 'بحر مجزوء المديد', 'بحر مجزوء المنسرح', 'بحر مجزوء المواليا', 'بحر مجزوء الهزج', 'بحر مجزوء الوافر', 'بحر مجزوء موشح', 'بحر مخلع البسيط', 'بحر مخلع الرجز', 'بحر مخلع الرمل', 'بحر مخلع السريع', 'بحر مخلع الكامل', 'بحر مخلع موشح', 'بحر مربع البسيط', 'بحر مربع الرجز', 'بحر مشطور الرجز', 'بحر مشطور السريع', 'بحر مشطور الطويل', 'بحر منهوك البسيط', 'بحر منهوك الرجز', 'بحر منهوك الكامل', 'بحر منهوك المنسرح', 'بحر موشح', 'بسيط', 'زجل', 'شعر التفعيلة', 'شعر حر', 'عامي', 'عدة أبحر', 'عموديه', 'مجزوء الخفيف', 'نثريه', 'None']
 themes = ['قصيدة اعتذار', 'قصيدة الاناشيد', 'قصيدة المعلقات', 'قصيدة حزينه', 'قصيدة دينية', 'قصيدة ذم', 'قصيدة رثاء', 'قصيدة رومنسيه', 'قصيدة سياسية', 'قصيدة شوق', 'قصيدة عامه', 'قصيدة عتاب', 'قصيدة غزل', 'قصيدة فراق', 'قصيدة قصيره', 'قصيدة مدح', 'قصيدة هجاء', 'قصيدة وطنيه', 'None']
 language_types = ['شعبي', 'عامي', 'فصحى', 'فصيح', '-', 'None']
@@ -82,18 +80,25 @@ tokenizer: AutoTokenizer = AutoTokenizer.from_pretrained("bkhmsi/arapoet-mt5", u
 model: AutoModelForSeq2SeqLM = AutoModelForSeq2SeqLM.from_pretrained("bkhmsi/arapoet-mt5", use_auth_token="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")
 model.eval()
 
-iface = gr.Interface(generate_poem, 
-    [
-        gr.Dropdown(countries),
-        gr.Dropdown(poet_era),
-        gr.Dropdown(meters),
-        gr.Dropdown(themes),
-        gr.Dropdown(language_types),
-        gr.Dropdown(poet_names),
-        gr.Slider(2, 20, value=10),
-        gr.Slider(1, 4, value=1),
-        "text"
-    ], "text"
-)
-
-iface.launch()
+title = "AraPoet: Controlled Arabic Poetry Generation"
+with gr.Blocks(title=title) as demo:
+    inputs = []
+    with gr.Row():
+        inputs += [gr.Dropdown(countries, label="Country", value="None")]
+        inputs += [gr.Dropdown(poet_era, label="Era", value="None")]
+    with gr.Row():
+        inputs += [gr.Dropdown(meters, label="Meter", value="None")]
+        inputs += [gr.Dropdown(themes, label="Theme", value="None")]
+    with gr.Row():
+        inputs += [gr.Dropdown(language_types, label="Language Type", value="None")]
+        inputs += [gr.Dropdown(poet_names, label="Poet", value="None")]
+    with gr.Row():
+        inputs += [gr.Slider(2, 20, value=4, step=1, label="Number of Lines")]
+        inputs += [gr.Slider(1, 4, value=1, step=1, label="Number of Samples")]
+    with gr.Row():
+        inputs += [gr.Textbox(label="Title")]
+
+    btn = gr.Button("Generate")
+    btn.click(generate_poem, inputs, gr.TextArea(label="Generation"))
+
+demo.launch()