Update app.py

app.py

@@ -12,21 +12,56 @@ with open('sinhala_comment_classifier_v1.pkl', "rb") as f:
 
 def predict_sinhala(text):
     prediction = model_sinhala.predict([text])
-    return "POSITIVE" if prediction[0] == 1 else "NEGATIVE"
+    return "No Harmful" if prediction[0] == 1 else "Harmful"
 
 def predict_singlish(text):
     prediction = model_singlish.predict([text])
-    return "POSITIVE" if prediction[0] == 1 else "NEGATIVE"
+    return "No Harmful" if prediction[0] == 1 else "Harmful"
 
 interface = gr.Interface(
     fn=lambda text, model_choice: predict_singlish(text) if model_choice == "Singlish" else predict_sinhala(text),
     inputs=[
-        gr.Textbox(label="Enter Text"),
-        gr.Radio(["Singlish", "Sinhala"], label="Choose Model")
+        gr.Textbox(
+            label="Enter Text", 
+            placeholder="Type a comment to analyze here...", 
+            lines=3, 
+            elem_id="input_text"  
+        ),
+        gr.Radio(
+            choices=["Singlish", "Sinhala"], 
+            label="Choose Language", 
+            info="Select which language model to use for the analysis.", 
+            elem_id="radio_choice"
+        ),
     ],
     outputs="text",
-    title="Sinhala and Singlish harm comment detector",
-    description="Choose a model and enter text to get predictions."
+    title="Sinhala and Singlish Harmful Comment Detector",
+    description="Select the language model and input a comment to detect whether it is harmful or not.",
+    theme="huggingface",  
+    css="""
+        .gradio-container { 
+            background-color: #f4f4f9; 
+            padding: 20px; 
+            border-radius: 8px;
+            box-shadow: 0px 4px 10px rgba(0, 0, 0, 0.1);
+        }
+        .gradio-input { 
+            border: 1px solid #ddd; 
+            padding: 10px; 
+            border-radius: 6px;
+        }
+        .gradio-button { 
+            background-color: #4CAF50; 
+            color: white; 
+            font-weight: bold; 
+            border-radius: 6px; 
+            padding: 12px 24px;
+        }
+        .gradio-button:hover {
+            background-color: #45a049;
+        }
+    """, 
+    allow_flagging="never" 
 )
 
 interface.launch(share=True)