Update app.py

app.py

@@ -45,55 +45,91 @@ def process_text(text, task):
     }
     return task_funcs.get(task, lambda x: "Invalid Task")(text)
 
-# Task-specific examples
-examples_en_bn_translation = [
-    ["The sky is blue, and the weather is nice."],
-    ["Artificial intelligence is shaping the future."],
-    ["Bangladesh is known for its rich culture and heritage."]
+# Task-specific examples with clear labels
+examples = [
+    ("Translate English to Bengali", [
+        ["The sky is blue, and the weather is nice."],
+        ["Artificial intelligence is shaping the future."],
+        ["Bangladesh is known for its rich culture and heritage."]
+    ]),
+    ("Translate Bengali to English", [
+        ["বাংলাদেশ দক্ষিণ এশিয়ার একটি সার্বভৌম রাষ্ট্র।"],
+        ["ঢাকা বাংলাদেশের রাজধানী।"],
+        ["রবীন্দ্রনাথ ঠাকুরের গান বাংলা সংস্কৃতির একটি অবিচ্ছেদ্য অংশ।"]
+    ]),
+    ("Summarize", [
+        ["The Department of Computer Science and Engineering, established in 1982, was the first of its kind in Bangladesh. "
+         "Attracting top students from all over the country, it offers both undergraduate and postgraduate degrees."],
+        ["Climate change is one of the biggest challenges we face today. With rising temperatures and unpredictable weather, "
+         "the world needs to come together to find sustainable solutions."],
+        ["Technology has advanced rapidly over the past decade, with innovations in fields like AI, robotics, and quantum computing."]
+    ]),
+    ("Paraphrase", [
+        ["The cat is sitting on the mat."],
+        ["He was very happy to receive the award."],
+        ["The weather today is sunny and warm."]
+    ])
 ]
 
-examples_bn_en_translation = [
-    ["বাংলাদেশ দক্ষিণ এশিয়ার একটি সার্বভৌম রাষ্ট্র।"],
-    ["ঢাকা বাংলাদেশের রাজধানী।"],
-    ["রবীন্দ্রনাথ ঠাকুরের গান বাংলা সংস্কৃতির একটি অবিচ্ছেদ্য অংশ।"]
-]
-
-examples_summarization = [
-    ["The Department of Computer Science and Engineering, established in 1982, was the first of its kind in Bangladesh. "
-     "Attracting top students from all over the country, it offers both undergraduate and postgraduate degrees."],
-    ["Climate change is one of the biggest challenges we face today. With rising temperatures and unpredictable weather, "
-     "the world needs to come together to find sustainable solutions."],
-    ["Technology has advanced rapidly over the past decade, with innovations in fields like AI, robotics, and quantum computing."]
-]
-
-examples_paraphrasing = [
-    ["The cat is sitting on the mat."],
-    ["He was very happy to receive the award."],
-    ["The weather today is sunny and warm."]
-]
-
-# Define the Gradio interface with enhanced visuals
+# Enhanced visual layout and interface design
 iface = gr.Interface(
     fn=process_text,
     inputs=[
-        "text", 
+        gr.Textbox(lines=5, label="Enter your text"),
         gr.Dropdown(
             ["Translate English to Bengali", "Translate Bengali to English", "Summarize", "Paraphrase"],
-            label="Choose Task",
-            elem_id="dropdown-task"
+            label="Select Task",
+            elem_id="dropdown-task",
+            interactive=True
         )
     ],
-    outputs="text",
-    title="BanglaT5 Model Hub",
-    description="A multi-functional tool for translation, summarization, and paraphrasing using BanglaT5 models.",
-    theme="finlaydog/seafoam",
+    outputs=gr.Textbox(label="Output"),
+    title="BanglaT5 Model Hub - Translation, Summarization & Paraphrasing",
+    description="Explore the power of BanglaT5 with this easy-to-use interface for multiple tasks like translation, summarization, and paraphrasing.",
+    theme="huggingface/space-shuttle",
     examples=[
-        *examples_en_bn_translation,  # Adding 3 examples for English to Bengali translation
-        *examples_bn_en_translation,  # Adding 3 examples for Bengali to English translation
-        *examples_summarization,      # Adding 3 examples for Summarization
-        *examples_paraphrasing        # Adding 3 examples for Paraphrasing
+        ("Translate English to Bengali", [
+            ["The sky is blue, and the weather is nice."],
+            ["Artificial intelligence is shaping the future."],
+            ["Bangladesh is known for its rich culture and heritage."]
+        ]),
+        ("Translate Bengali to English", [
+            ["বাংলাদেশ দক্ষিণ এশিয়ার একটি সার্বভৌম রাষ্ট্র।"],
+            ["ঢাকা বাংলাদেশের রাজধানী।"],
+            ["রবীন্দ্রনাথ ঠাক��রের গান বাংলা সংস্কৃতির একটি অবিচ্ছেদ্য অংশ।"]
+        ]),
+        ("Summarize", [
+            ["The Department of Computer Science and Engineering, established in 1982, was the first of its kind in Bangladesh. "
+             "Attracting top students from all over the country, it offers both undergraduate and postgraduate degrees."],
+            ["Climate change is one of the biggest challenges we face today. With rising temperatures and unpredictable weather, "
+             "the world needs to come together to find sustainable solutions."],
+            ["Technology has advanced rapidly over the past decade, with innovations in fields like AI, robotics, and quantum computing."]
+        ]),
+        ("Paraphrase", [
+            ["The cat is sitting on the mat."],
+            ["He was very happy to receive the award."],
+            ["The weather today is sunny and warm."]
+        ])
     ],
     allow_flagging="auto",
+    layout="horizontal",  # Adding a horizontal layout for a cleaner design
+    css="""
+        #dropdown-task {
+            color: #2C3E50;
+            font-size: 18px;
+        }
+        .output-area {
+            color: #2C3E50;
+        }
+        .examples {
+            display: grid;
+            grid-template-columns: 1fr 1fr;
+            gap: 20px;
+        }
+        .output-text {
+            background-color: #E8F6F3;
+        }
+    """
 )
 
 # Launch the Gradio app