Add stop buttons

app.py

@@ -184,6 +184,7 @@ iface_category = gr.Interface(
     inputs=gr.Textbox(lines=10, placeholder="Enter Email Content Here..."),
     outputs="text",
     title="Email Category Classifier",
+    stop_btn=gr.Button("Stop", variant="stop", visible=True),
     description="This model classifies email text into one of five categories: Urgent Requests, Project Updates, Client Communications, Meeting Coordination, Internal Announcements."
 )
 
@@ -192,6 +193,7 @@ iface_sentiment = gr.Interface(
     fn=classify_sentiment,
     inputs=gr.Textbox(lines=5, placeholder="Enter Email Text Here..."),
     outputs=gr.Textbox(label="Sentiment Analysis"),
+    stop_btn=gr.Button("Stop", variant="stop", visible=True),
     title="Sentiment Analysis"
 )
 
@@ -200,6 +202,7 @@ iface_summary = gr.Interface(
     fn=generate_summary,
     inputs=[gr.Textbox(lines=5, placeholder="Enter Email Text Here...")],
     outputs=gr.Textbox(label="Generated Summary"),
+    stop_btn=gr.Button("Stop", variant="stop", visible=True),
     title="Summary Generation"
 )
 
@@ -211,6 +214,7 @@ iface_ner = gr.Interface(
         gr.Dataframe(label="Transformer Entity Recognition"),
         gr.Textbox(label="Extracted Dates")
     ],
+    stop_btn=gr.Button("Stop", variant="stop", visible=True),
     title="NER Analysis",
     description="Performs Named Entity Recognition using spaCy and Transformer models."
 )
@@ -219,6 +223,7 @@ iface_response = gr.Interface(
     inputs=gr.Textbox(lines=10, placeholder="Enter the email prompt..."),
     outputs=gr.Textbox(label="Generated Email Response"),
     title="Email Response Generator",
+    stop_btn=gr.Button("Stop", variant="stop", visible=True),
     description="Generate email responses using Llama-2 model."
 )