Update app.py

app.py

@@ -2,7 +2,7 @@
 # -*- coding: utf-8 -*-
 """
 🪶 Council Matters Classifier – PT
-Dark modern Gradio interface with working suggestions carousel.
+Dark modern Gradio interface with suggestions carousel.
 """
 
 import gradio as gr
@@ -20,8 +20,7 @@ try:
 except ImportError:
     TORCH_AVAILABLE = False
 
-
-# ---------------- Model Definition ----------------
+# ---------------- Classifier ----------------
 class PortugueseClassifier:
     def __init__(self):
         self.model_path = Path("models")
@@ -127,7 +126,6 @@ class PortugueseClassifier:
         predicted_labels.sort(key=lambda x: x["probability"], reverse=True)
         return predicted_labels
 
-
 # ---------------- Load Classifier ----------------
 classifier = PortugueseClassifier()
 
@@ -149,12 +147,15 @@ example_idx = 0
 def next_example():
     global example_idx
     example_idx = (example_idx + 1) % len(suggestions)
-    return suggestions[example_idx], suggestions[example_idx]
+    return suggestions[example_idx]
 
 def prev_example():
     global example_idx
     example_idx = (example_idx - 1 + len(suggestions)) % len(suggestions)
-    return suggestions[example_idx], suggestions[example_idx]
+    return suggestions[example_idx]
+
+def use_suggestion(suggestion):
+    return suggestion
 
 def classify_display(text):
     preds = classifier.predict(text)
@@ -181,6 +182,8 @@ button:hover { background-color: #00aaff !important; }
 .suggestion-box { background-color: #111; border-radius: 10px; border: 1px solid #222; padding: 10px; display: flex; align-items: center; justify-content: space-between; color: #aaa; margin-top: 25px; }
 .arrow-btn { background: none; border: none; color: #00c3ff; font-size: 22px; cursor: pointer; }
 .arrow-btn:hover { color: #00e0ff; transform: scale(1.15); }
+.use-btn { background-color:#00c3ff !important; color:#000 !important; font-weight:600 !important; border-radius:6px !important; padding:3px 10px !important; }
+.use-btn:hover { background-color:#00e0ff !important; }
 """
 
 # ---------------- Gradio UI ----------------
@@ -188,24 +191,23 @@ with gr.Blocks(css=custom_css, theme="gradio/soft") as demo:
     gr.Markdown("## 🏛️ **Council Matters Classifier – PT**")
     gr.Markdown("### Insira texto administrativo em português:")
 
-    input_text = gr.Textbox(
-        label="", placeholder="Escreva aqui o texto em português...", lines=6, elem_id="input-text"
-    )
+    input_text = gr.Textbox(label="", placeholder="Escreva aqui o texto em português...", lines=6)
 
     classify_btn = gr.Button("Classificar")
-    output = gr.HTML(label="Tópicos previstos")
+    output = gr.HTML()
     classify_btn.click(fn=classify_display, inputs=input_text, outputs=output)
 
     # Sugestões
     prev_btn = gr.Button("⟨")
     suggestion_display = gr.Textbox(value=suggestions[0], interactive=False)
     next_btn = gr.Button("⟩")
+    use_btn = gr.Button("Usar sugestão")
 
-    prev_btn.click(fn=prev_example, outputs=[input_text, suggestion_display])
-    next_btn.click(fn=next_example, outputs=[input_text, suggestion_display])
+    prev_btn.click(fn=prev_example, outputs=suggestion_display)
+    next_btn.click(fn=next_example, outputs=suggestion_display)
+    use_btn.click(fn=use_suggestion, inputs=suggestion_display, outputs=input_text)
 
-    gr.Row([prev_btn, suggestion_display, next_btn], elem_id="suggestion-box")
+    gr.Row([prev_btn, suggestion_display, next_btn, use_btn], elem_id="suggestion-box")
 
-# ---------------- Launch ----------------
 if __name__ == "__main__":
     demo.launch()