Update app.py

app.py

@@ -1,6 +1,5 @@
 from transformers import pipeline
-import gradio as gr 
-import torch
+import gradio as gr
 import time  # To simulate processing time for demonstration
 
 # Load the summarization pipelines
@@ -111,32 +110,28 @@ button:hover, .btn:hover {
 
 # Create the Gradio interface
 with gr.Blocks(css=custom_css) as interface:
-    gr.Markdown("<h1 style='text-align: center;' id='title'>Text Summarization Tool</h1>")
+    title = gr.Markdown("<h1 style='text-align: center;'>Text Summarization Tool</h1>")
     
-    text_input = gr.Textbox(label="Enter Text Here", lines=8, text_align="right", elem_id="text_input")
-    max_length = gr.Slider(minimum=80, maximum=200, value=100, step=1, label="Max Summary Length", elem_id="max_length")
-    min_length = gr.Slider(minimum=5, maximum=50, value=20, step=1, label="Min Summary Length", elem_id="min_length")
+    text_input = gr.Textbox(label="Enter Text Here", lines=8, text_align="right")
+    max_length = gr.Slider(minimum=80, maximum=200, value=100, step=1, label="Max Summary Length")
+    min_length = gr.Slider(minimum=5, maximum=50, value=20, step=1, label="Min Summary Length")
     
-    summarize_button = gr.Button("Summarize", elem_id="summarize_button")
-    language_button = gr.Button("Switch Language", elem_id="language_button")
-    summary_output = gr.Textbox(label="Summary", lines=6, text_align="right", elem_id="summary_output")
+    summarize_button = gr.Button("Summarize")
+    language_button = gr.Button("Switch Language")
+    summary_output = gr.Textbox(label="Summary", lines=6, text_align="right")
 
     summarize_button.click(summarize, inputs=[text_input, max_length, min_length], outputs=summary_output)
-    language_button.click(lambda: (toggle_language(), update_interface()), outputs=[summary_output, text_input, max_length, min_length, summarize_button, language_button])
+    language_button.click(lambda: (toggle_language(), update_labels()), outputs=[title, text_input, max_length, min_length, summarize_button, summary_output])
 
     # Update interface labels
     def update_labels():
         labels = update_interface()
-        return labels["title"], labels["textbox_label"], labels["max_length_label"], labels["min_length_label"], labels["summarize_button_label"], labels["summary_label"]
-
-    title, text_label, max_label, min_label, summarize_label, summary_label = update_labels()
-    
-    interface.set_title(title)
-    text_input.label = text_label
-    max_length.label = max_label
-    min_length.label = min_label
-    summarize_button.label = summarize_label
-    summary_output.label = summary_label
+        title.update(value=f"<h1 style='text-align: center;'>{labels['title']}</h1>")
+        text_input.label = labels["textbox_label"]
+        max_length.label = labels["max_length_label"]
+        min_length.label = labels["min_length_label"]
+        summarize_button.label = labels["summarize_button_label"]
+        summary_output.label = labels["summary_label"]
 
 if __name__ == "__main__":
     interface.launch()