Update app.py

app.py

@@ -41,15 +41,43 @@ def generate_summary(text, style):
   )
   return output[0]["summary_text"]
 
-demo = gr.Interface(
-  fn=generate_summary,
-  inputs=[
-    gr.Textbox(label="Input Text", lines=10, placeholder="Paste text here..."),
-    gr.Radio(["harsh", "standard", "detailed"], label="Summary Style", value="standard")
-  ],
-  outputs=gr.Textbox(label="Summary"),
-  title="Style Summarizer",
-  description="A custom Flan-T5-Base model fine-tuned to generate summaries in different styles."
-)
+custom_css = """
+#header {text-align: center; margin-bottom: 25px;}
+.gradio-container {max-width: 1000px !important;}
+footer {display: none !important;}
+"""
+
+with gr.Blocks(css=custom_css, theme=gr.themes.Base()) as demo:
+  with gr.Column(elem_id="header"):
+    gr.Markdown("# Style Summarizer")
+    gr.Markdown("Fine-tuned Flan-T5 model for multi-style document summarization.")
+
+  with gr.Row():
+    with gr.Column(scale=1):
+      input_box = gr.Textbox(
+        label="Input Text", 
+        placeholder="Enter text to be summarized...", 
+        lines=12
+      )
+      style_radio = gr.Radio(
+        choices=["harsh", "standard", "detailed"], 
+        label="Summary Type", 
+        value="standard"
+      )
+      submit_btn = gr.Button("Process Summary", variant="primary")
+      
+    with gr.Column(scale=1):
+      output_box = gr.Textbox(
+        label="Output Summary", 
+        lines=15, 
+        interactive=False,
+        show_copy_button=True
+      )
+
+  submit_btn.click(
+    fn=generate_summary, 
+    inputs=[input_box, style_radio], 
+    outputs=output_box
+  )
 
 demo.launch()