ui

app.py

@@ -1,8 +1,9 @@
 # Import necessary libraries
+import gradio as gr
 import torch
 import spaces
 from transformers import AutoTokenizer, AutoConfig, AutoModelForCausalLM, pipeline
-import gradio as gr
+
 
 # Load Goldfish model for Dhivehi
 model_name = 'div_thaa_full'
@@ -29,31 +30,66 @@ def generate_text(input_text):
 
 # Create Gradio interface
 
-styles = """
-.thaana textarea {
+# Custom CSS with modern Gradio styling
+custom_css = """
+.thaana-textbox textarea {
     font-size: 18px !important;
-    font-family: 'MV_Faseyha', 'Faruma', 'A_Faruma', 'Noto Sans Thaana', 'MV Boli';
+    font-family: 'MV_Faseyha', 'Faruma', 'A_Faruma', 'Noto Sans Thaana', 'MV Boli' !important;
     line-height: 1.8 !important;
+    direction: rtl !important;
 }
 """
 
-demo = gr.Interface(
-    fn=generate_text,
-    css=styles,
-    inputs=gr.Textbox(lines=2, label="Enter Dhivehi Text", rtl=True, elem_classes="thaana"),
-    outputs=gr.Textbox(lines=2, rtl=True, elem_classes="thaana"),
-    title="Demo Dhivehi Text Generator",
-    description="Generate text in Dhivehi language. This model is trained to generate coherent text based on the input prompt.",
-    article="<p>Model: Goldfish is a suite of monolingual language models trained for 350 languages. This model is the Dhivehi (Thaana script). For more details, visit the <a href='https://github.com/tylerachang/goldfish' target='_blank'>Goldfish Models GitHub repository</a>.</p>",
-    examples=[
-        ["ދިވެހިރާއްޖެ"],
-        ["އެމެރިކާ އިންތިޚާބު"],
-        ["ސަލާމް"],
-        ["ދުނިޔޭގެ ސިއްޙަތު ޖަމްޢިއްޔާ"],
-        ["ޤަދީމީ ސަގާފަތް"],
-        ["ޑިމޮކްރަސީ"]
-    ]
-)
-
-# Launch the app
-demo.launch()
+# Example inputs
+examples = [
+    ["ދިވެހިރާއްޖެ"],
+    ["އެމެރިކާ އިންތިޚާބު"],
+    ["ސަލާމް"],
+    ["ދުނިޔޭގެ ސިއްޙަތު ޖަމްޢިއްޔާ"],
+    ["ޤަދީމީ ސަގާފަތް"],
+    ["ޑިމޮކްރަސީ"]
+]
+
+# Create Gradio app with modern components
+with gr.Blocks(css=custom_css) as demo:
+    gr.Markdown("# Demo Dhivehi Text Generator")
+    gr.Markdown("Generate text in Dhivehi language. This model is trained to generate coherent text based on the input prompt.")
+    
+    with gr.Row():
+        input_text = gr.Textbox(
+            label="Enter Dhivehi Text",
+            lines=2,
+            elem_classes=["thaana-textbox"],
+            rtl=True
+        )
+        output_text = gr.Textbox(
+            label="Generated Text",
+            lines=2,
+            elem_classes=["thaana-textbox"],
+            rtl=True
+        )
+    
+    generate_btn = gr.Button("Generate")
+    generate_btn.click(
+        fn=generate_text,
+        inputs=input_text,
+        outputs=output_text
+    )
+    
+    gr.Examples(
+        examples=examples,
+        inputs=input_text,
+        outputs=output_text,
+        fn=generate_text,
+        cache_examples=True
+    )
+    
+    gr.Markdown("""
+    ### Model Information
+    Model: Goldfish is a suite of monolingual language models trained for 350 languages. 
+    This model is the Dhivehi (Thaana script). For more details, visit the 
+    [Goldfish Models GitHub repository](https://github.com/tylerachang/goldfish).
+    """)
+
+if __name__ == "__main__":
+    demo.launch()