Update app.py

app.py

@@ -12,17 +12,21 @@ def classify_sentiment(text):
     inputs = tokenizer(text, return_tensors="tf", truncation=True, padding=True)
     outputs = model(**inputs)
     prediction = tf.argmax(outputs.logits, axis=1).numpy()[0]
-    return labels[prediction].lower()
+    sentiment = labels[prediction].lower()
+
+    # Return both processed message and classification
+    return text, sentiment
 
 with gr.Blocks() as demo:
     gr.Markdown("# Fas7ni Feedbacks\nClassify your feedback as positive or negative!")
 
     with gr.Row():
         msg = gr.Textbox(placeholder="Write your comment or feedback here...", label="Enter Feedback")
-        output = gr.Textbox(label="Sentiment")
+        processed = gr.Textbox(label="Processed Message")
+        sentiment = gr.Textbox(label="Sentiment")
 
     submit = gr.Button("Classify")
 
-    submit.click(fn=classify_sentiment, inputs=msg, outputs=output)
+    submit.click(fn=classify_sentiment, inputs=msg, outputs=[processed, sentiment])
 
 demo.launch()