UI - theme

app.py

@@ -32,13 +32,13 @@ def classify_text(text):
         outputs = model(**inputs)
         probabilities = torch.softmax(outputs.logits, dim=1)[0]
         predicted_class = torch.argmax(probabilities).item()
-        confidence = probabilities[predicted_class].item() * 100
+        confidence = probabilities[predicted_class].item() * 100  # Confidence as %
 
         if predicted_class == 24:
-            prediction_label = f"✅ The text is **{confidence:.2f}%** likely **Human written**."
-            model_info = ""
+            prediction_label = f"✅ - The text is **{confidence:.2f}%** likely **Human written**."
+            model_info = ""  
         else:
-            prediction_label = f"🤖 The text is **{confidence:.2f}%** likely **AI generated**."
+            prediction_label = f"🤖 - The text is **{confidence:.2f}%** likely **AI generated**."
             model_info = f"**Identified AI Model:** {label_mapping[predicted_class]}"
 
     result_message = f"**Result:**\n\n{prediction_label}"
@@ -47,7 +47,7 @@ def classify_text(text):
         
     return result_message
 
-title = "AI Generated Text Detector"
+title = "AI Text Detector"
 description = """
 **AI detection tool by SzegedAI**
 
@@ -57,6 +57,8 @@ Detect AI-generated texts with precision using the new **ModernBERT** model, fin
 - **✅ Human Verification**: Marks confidently human-written text with a green checkmark.
 
 **Press the button below to classify your text!**
+
+
 """
 
 iface = gr.Interface(
@@ -67,7 +69,7 @@ iface = gr.Interface(
         lines=5,
         elem_id="text_input_box"
     ),
-    outputs=gr.Markdown(),
+    outputs=gr.Markdown(elem_id="result_output_box"),
     title=title,
     description=description,
     allow_flagging="never", 
@@ -77,6 +79,15 @@ iface = gr.Interface(
         border-radius: 10px;
         border: 2px solid #4CAF50;
         font-size: 18px;
+        padding: 10px;
+    }
+    #result_output_box {
+        border-radius: 10px;
+        border: 2px solid #4CAF50;
+        font-size: 18px;
+        padding: 15px;
+        background-color: #2E2E3F;
+        margin-top: 20px;
     }
     body {
         background: #1E1E2F;
@@ -87,13 +98,14 @@ iface = gr.Interface(
     .gradio-container {
         border: 2px solid #4CAF50;
         border-radius: 15px;
-        padding: 20px;
+        padding: 25px;
         box-shadow: 0px 0px 20px rgba(0,255,0,0.6);
     }
     h1, h2 {
         text-align: center;
         font-size: 32px;
         font-weight: bold;
+        margin-bottom: 20px;
     }
     """
 )