Update app.py

app.py

@@ -45,71 +45,72 @@ def process_text(text, task):
     }
     return task_funcs.get(task, lambda x: "Invalid Task")(text)
 
-# Task-specific examples with clear labels for each task
-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."]
-    ]]
+# Task-specific examples
+examples_translate_en_bn = [
+    ["The sky is blue, and the weather is nice."],
+    ["Artificial intelligence is shaping the future."],
+    ["Bangladesh is known for its rich culture and heritage."]
+]
+
+examples_translate_bn_en = [
+    ["বাংলাদেশ দক্ষিণ এশিয়ার একটি সার্বভৌম রাষ্ট্র।"],
+    ["ঢাকা বাংলাদেশের রাজধানী।"],
+    ["রবীন্দ্রনাথ ঠাকুরের গান বাংলা সংস্কৃতির একটি অবিচ্ছেদ্য অংশ।"]
+]
+
+examples_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."]
+]
+
+examples_paraphrase = [
+    ["The cat is sitting on the mat."],
+    ["He was very happy to receive the award."],
+    ["The weather today is sunny and warm."]
 ]
 
 # Enhanced visual layout and interface design
-iface = gr.Interface(
-    fn=process_text,
-    inputs=[
-        gr.Textbox(lines=5, label="Enter your text"),
-        gr.Dropdown(
-            ["Translate English to Bengali", "Translate Bengali to English", "Summarize", "Paraphrase"],
-            label="Select Task",
-            elem_id="dropdown-task",
-            interactive=True
-        )
-    ],
-    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,  # Show examples as per tasks with 3 for each
-    allow_flagging="auto",
-    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;
-        }
-        .gr-textbox {
-            border-radius: 10px;
-        }
-    """
-)
-
-# Launch the Gradio app
-iface.launch(inline=False)
+with gr.Blocks() as demo:
+    gr.Markdown("# BanglaT5 Model Hub - Translation, Summarization & Paraphrasing")
+    gr.Markdown("Explore the power of BanglaT5 with this easy-to-use interface for multiple tasks like translation, summarization, and paraphrasing.")
+
+    with gr.Tabs():
+        with gr.Tab("Translate English to Bengali"):
+            gr.Examples(
+                examples_translate_en_bn,
+                inputs=gr.Textbox(label="Enter English Text", lines=5),
+                outputs=gr.Textbox(label="Bengali Translation")
+            )
+        
+        with gr.Tab("Translate Bengali to English"):
+            gr.Examples(
+                examples_translate_bn_en,
+                inputs=gr.Textbox(label="Enter Bengali Text", lines=5),
+                outputs=gr.Textbox(label="English Translation")
+            )
+        
+        with gr.Tab("Summarize"):
+            gr.Examples(
+                examples_summarize,
+                inputs=gr.Textbox(label="Enter Text to Summarize", lines=5),
+                outputs=gr.Textbox(label="Summary")
+            )
+        
+        with gr.Tab("Paraphrase"):
+            gr.Examples(
+                examples_paraphrase,
+                inputs=gr.Textbox(label="Enter Text to Paraphrase", lines=5),
+                outputs=gr.Textbox(label="Paraphrased Text")
+            )
+    
+    input_text = gr.Textbox(label="Enter your text here", lines=5)
+    task = gr.Dropdown(["Translate English to Bengali", "Translate Bengali to English", "Summarize", "Paraphrase"], label="Select Task")
+    output_text = gr.Textbox(label="Output", lines=5)
+    
+    submit_btn = gr.Button("Run Task")
+    submit_btn.click(process_text, inputs=[input_text, task], outputs=output_text)
+
+demo.launch()