Update app.py

app.py

@@ -1,37 +1,107 @@
-import gradio as gr
 from transformers import pipeline
+import gradio as gr 
+import torch
+import time  # To simulate processing time for demonstration
 
-# Load summarization models
-english_summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+# Load the summarization pipelines
 arabic_summarizer = pipeline("summarization", model="eslamxm/mt5-base-finetuned-persian-finetuned-persian-arabic")
+english_summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
-def summarize_text(text, language):
-    if language == 'English':
-        summary = english_summarizer(text, max_length=130, min_length=30, do_sample=False)
-        return str(summary[0]['summary_text'])
-    elif language == 'Arabic':
-        summary = arabic_summarizer(text, max_length=130, min_length=30, do_sample=False)
-        return str(summary[0]['summary_text'])
-    else:
-        return "Unsupported Language"
+# Initialize the current language
+current_language = 'Arabic'
 
-with gr.Blocks() as app:
-    gr.Markdown("<h1 style='text-align: center;'>Text Summarization App</h1>")
+def summarize(text, max_length, min_length, progress=gr.Progress()):
+    global current_language
+    # Simulate a delay to show the progress bar
+    for step in range(10):
+        time.sleep(0.1)  # Simulating processing time
+        progress(step / 10)  # Update the progress bar
+    
+    # Select the summarizer based on the current language
+    if current_language == 'Arabic':
+        summary = arabic_summarizer(text, max_length=max_length, min_length=min_length, do_sample=False)[0]['summary_text']
+    else:
+        summary = english_summarizer(text, max_length=max_length, min_length=min_length, do_sample=False)[0]['summary_text']
     
-    with gr.Row():
-        with gr.Column():
-            english_input = gr.Textbox(label="Enter English Text", placeholder="Type or paste your English text here...", lines=10)
-            english_button = gr.Button("Summarize English")
-            english_output = gr.Textbox(label="English Summary", interactive=False)
+    return summary
 
-        with gr.Column():
-            arabic_input = gr.Textbox(label="ادخل نص عربي", placeholder="اكتب أو الصق نصك العربي هنا...", lines=10)
-            arabic_button = gr.Button("تلخيص عربي")
-            arabic_output = gr.Textbox(label="ملخص عربي", interactive=False)
+def toggle_language():
+    global current_language
+    if current_language == 'Arabic':
+        current_language = 'English'
+        return "Switched to English summarization."
+    else:
+        current_language = 'Arabic'
+        return "Switched to Arabic summarization."
+
+# Custom CSS for the progress bar with animations
+custom_css = """
+body {
+    background: linear-gradient(135deg, #667eea, #764ba2);
+    font-family: 'Helvetica Neue', Helvetica, Arial, sans-serif;
+    color: #333;
+    margin: 0;
+    padding: 0;
+}
+.gradio-container {
+    background: rgba(255, 255, 255, 0.95);
+    border-radius: 15px;
+    padding: 30px 40px;
+    box-shadow: 0 8px 30px rgba(0, 0, 0, 0.3);
+    margin: 40px auto;
+    max-width: 1200px;
+}
+.gradio-container h1 {
+    color: #333;
+    text-shadow: 1px 1px 2px rgba(0, 0, 0, 0.2);
+}
+.fillable { 
+    width: 95% !important; 
+    max-width: unset !important;
+}
+#examples_container {
+    margin: auto;
+    width: 90%;
+}
+#examples_row {
+    justify-content: center;
+}
+.sidebar {
+    background: rgba(255, 255, 255, 0.98);
+    border-radius: 10px;
+    padding: 20px;
+    box-shadow: 0 4px 15px rgba(0, 0, 0, 0.2);
+}
+button, .btn {
+    background: linear-gradient(90deg, #ff8a00, #e52e71);
+    border: none;
+    color: #fff;
+    padding: 12px 24px;
+    text-transform: uppercase;
+    font-weight: bold;
+    letter-spacing: 1px;
+    border-radius: 5px;
+    cursor: pointer;
+    transition: transform 0.2s ease-in-out;
+}
+button:hover, .btn:hover {
+    transform: scale(1.05);
+}
+"""
+
+# Create the Gradio interface
+with gr.Blocks(css=custom_css) as interface:
+    gr.Markdown("<h1 style='text-align: center;'>Text Summarization Tool</h1>")
+    
+    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")
+    language_button = gr.Button("Switch Language")
+    summary_output = gr.Textbox(label="Summary", lines=6, text_align="right")
 
-    english_button.click(summarize_text, inputs=[*english_input, 'English'], outputs=english_output)
-    arabic_button.click(summarize_text, inputs=[*arabic_input, 'Arabic'], outputs=arabic_output)
+    summarize_button.click(summarize, inputs=[text_input, max_length, min_length], outputs=summary_output)
+    language_button.click(toggle_language, outputs=summary_output)
 
-# Launch the interface
 if __name__ == "__main__":
-    app.launch()
+    interface.launch()