Update app.py

app.py

@@ -68,7 +68,6 @@ if tokenizer: # Only set normalizer if tokenizer loaded successfully
         newline_to_space,
         Strip()
     ])
-# --- End Model & Tokenizer Configuration ---
 
 
 title_md = """
@@ -78,19 +77,19 @@ title_md = """
 
 description = """
 <div class="app-description">
-    <p>This tool utilizes the <b>ModernBERT</b> model to discern whether a given text is human-authored or AI-generated. It employs a soft voting ensemble of <b>three</b> models, amalgamating their outputs to enhance detection accuracy.</p>
+    <p>This tool utilizes the <b>ModernBERT</b> model to decide whether a given text is human-authored or AI-generated. It employs a soft voting ensemble of <b>three</b> models to improve detection accuracy.</p>
     <ul class="features-list">
-        <li><span class="icon">✅</span> <strong>Human Verification:</strong> Clearly identifies human-written content.</li>
-        <li><span class="icon">🔍</span> <strong>Model Detection:</strong> Capable of identifying content from over 40 AI models.</li>
-        <li><span class="icon">📈</span> <strong>Accuracy:</strong> Performs optimally with more extensive text inputs.</li>
-        <li><span class="icon">📄</span> <strong>Read more:</strong> Our methodology is detailed in our research paper:
-            <a href="https://aclanthology.org/2025.genaidetect-1.15/" target="_blank" class="learn-more-link"><b> LINK</b></a>.
+        <li><span class="icon">✅</span> <strong>Human Verification: </strong> Clearly identifies human-written content.</li>
+        <li><span class="icon">🔍</span> <strong>Model Detection: </strong> Capable of identifying content from over 40 AI models.</li>
+        <li><span class="icon">📈</span> <strong>Accuracy: </strong> Performs optimally with more extensive text inputs.</li>
+        <li><span class="icon">📄</span> <strong>Read more: </strong> Our methodology is detailed in our research paper: 
+            <a href="https://aclanthology.org/2025.genaidetect-1.15/" target="_blank" class="learn-more-link"> <b> LINK </b></a>.
         </li>
     </ul>
     <p class="instruction-text">Paste your text into the field below to analyze its origin.</p>
 </div>
 """
-bottom_text = "<p class='footer-text'>SzegedAI</p>" # Simplified footer, as requested
+bottom_text = "<p class='footer-text'>SzegedAI - Mihaly Kiss</p>"
 
 AI_texts = [
     "Camels are remarkable desert animals known for their unique adaptations to harsh, arid environments. Native to the Middle East, North Africa, and parts of Asia, camels have been essential to human life for centuries, serving as a mode of transportation, a source of food, and even a symbol of endurance and survival. There are two primary species of camels: the dromedary camel, which has a single hump and is commonly found in the Middle East and North Africa, and the Bactrian camel, which has two humps and is native to Central Asia. Their humps store fat, not water, as commonly believed, allowing them to survive long periods without food by metabolizing the stored fat for energy. Camels are highly adapted to desert life. They can go for weeks without water, and when they do drink, they can consume up to 40 gallons in one sitting. Their thick eyelashes, sealable nostrils, and wide, padded feet protect them from sand and help them walk easily on loose desert terrain.",
@@ -105,7 +104,7 @@ def classify_text_interface(text):
         return "<p style='text-align: center; color: var(--ai-color);'><strong>Error: Models not loaded. Please check the console.</strong></p>"
 
     cleaned_text = clean_text(text)
-    if not cleaned_text.strip(): # Check cleaned_text here
+    if not cleaned_text.strip(): 
         result_message = "<p style='text-align: center; color: var(--text-secondary);'>Please enter some text to analyze.</p>"
         return result_message
 
@@ -125,29 +124,25 @@ def classify_text_interface(text):
 
     ai_probs = probabilities.clone()
     human_label_index = -1
-    for k, v in label_mapping.items(): # Find the human label index dynamically
+    for k, v in label_mapping.items():
         if v.lower() == 'human':
             human_label_index = k
             break
     
     if human_label_index != -1:
-        ai_probs[human_label_index] = 0 # Zero out human probability for AI sum
+        ai_probs[human_label_index] = 0 
         human_prob_value = probabilities[human_label_index].item() * 100
-    else: # Fallback if 'human' not in label_mapping (should not happen with current map)
+    else: 
         human_prob_value = 0 
         print("Warning: 'human' label not found in label_mapping.")
 
     ai_total_prob = ai_probs.sum().item() * 100
 
-    # Recalculate human_prob based on ai_total_prob if necessary,
-    # or ensure the logic correctly identifies human vs AI majority.
-    # The original logic: human_prob = 100 - ai_total_prob might be confusing if ai_total_prob already excluded human.
-    # Let's use the direct human probability from the model.
     
-    ai_argmax_index = torch.argmax(ai_probs).item() # Argmax over non-human probabilities
+    ai_argmax_index = torch.argmax(ai_probs).item() 
     ai_argmax_model = label_mapping.get(ai_argmax_index, "Unknown AI")
 
-    if human_prob_value > ai_total_prob : # Compare direct human probability with sum of AI probabilities
+    if human_prob_value > ai_total_prob : 
         result_message = (
             f"<p><strong>The text is</strong> <span class='highlight-human'><strong>{human_prob_value:.2f}%</strong> likely <b>Human written</b>.</span></p>"
         )
@@ -171,12 +166,12 @@ modern_css = """
     --border-color: #E0E0E0;
     --input-bg: #FFFFFF;
     --input-focus-border: var(--accent-color);
-    --human-color: #2ECC71; /* Green */
+    --human-color: #2ECC71; 
     --human-bg: rgba(46, 204, 113, 0.1);
-    --ai-color: #E74C3C; /* Red */
+    --ai-color: #E74C3C;
     --ai-bg: rgba(231, 76, 60, 0.1);
     --shadow-color: rgba(44, 62, 80, 0.1);
-    --container-max-width: 800px; /* Increased width */
+    --container-max-width: 800px; 
     --border-radius-md: 8px;
     --border-radius-lg: 12px;
 }
@@ -206,20 +201,19 @@ body {
     border: none;
 }
 
-.form.svelte-633qhp, .block.svelte-11xb1hd, .gradio-html .block { /* More generic selector for Gradio HTML block */
+.form.svelte-633qhp, .block.svelte-11xb1hd, .gradio-html .block { 
     background: none !important;
     border: none !important;
     box-shadow: none !important;
     padding: 0 !important;
 }
 
-/* Title and subtitle are now handled by Markdown with inline styles, h1 here is a fallback or for other h1s */
 h1 { 
     color: var(--text-primary);
     font-size: clamp(24px, 5vw, 30px); 
     font-weight: 700;
     text-align: center;
-    margin-bottom: 20px; /* Adjusted default h1 margin */
+    margin-bottom: 20px;
     letter-spacing: -0.5px;
 }
 
@@ -293,7 +287,7 @@ h1 {
 }
 
 #result_output_box { 
-    background-color: var(--input-bg); /* Ensure background for the box */
+    background-color: var(--input-bg);
     border: 1px solid var(--border-color);
     border-radius: var(--border-radius-md);
     padding: 20px;
@@ -301,15 +295,15 @@ h1 {
     width: 100%;
     box-sizing: border-box;
     text-align: center;
-    font-size: clamp(16px, 3vw, 17px); /* Slightly adjusted font size for results */
+    font-size: clamp(16px, 3vw, 17px);
     box-shadow: 0 4px 8px rgba(0,0,0,0.05);
-    min-height: 80px; /* Give it some min height */
-    display: flex; /* For centering content if needed */
+    min-height: 80px; 
+    display: flex; 
     flex-direction: column;
     justify-content: center;
 }
-#result_output_box p { /* Style paragraphs inside the result box */
-    margin-bottom: 8px; /* Space between lines in result */
+#result_output_box p { 
+    margin-bottom: 8px; 
     line-height: 1.6;
 }
 #result_output_box p:last-child {
@@ -319,22 +313,22 @@ h1 {
 
 .highlight-human, .highlight-ai {
     font-weight: 600;
-    padding: 5px 10px; /* Adjusted padding */
+    padding: 5px 10px; 
     border-radius: var(--border-radius-md);
     display: inline-block;
-    font-size: 1.05em; /* Adjusted size */
+    font-size: 1.05em; 
 }
 
 .highlight-human {
     color: var(--human-color);
     background-color: var(--human-bg);
-    /* border: 1px solid var(--human-color); Removed border for cleaner look */
+    /* border: 1px solid var(--human-color); 
 }
 
 .highlight-ai {
     color: var(--ai-color);
     background-color: var(--ai-bg);
-    /* border: 1px solid var(--ai-color); Removed border for cleaner look */
+    /* border: 1px solid var(--ai-color); 
 }
 
 .tabs > div:first-child button {
@@ -359,14 +353,14 @@ h1 {
     border: 1px solid var(--border-color) !important;
     border-radius: var(--border-radius-md) !important;
     background-color: #fdfdfd !important;
-    margin-top: 10px; /* Add some space above examples */
+    margin-top: 10px; 
 }
 .gr-sample-textbox {
     border: 1px solid var(--border-color) !important;
     border-radius: var(--border-radius-md) !important;
     font-size: 14px !important;
 }
-.gr-accordion > .label-wrap button { /* Style accordion label */
+.gr-accordion > .label-wrap button {
     font-weight: 500 !important;
     color: var(--text-primary) !important;
 }
@@ -401,40 +395,36 @@ h1 {
 iface = gr.Blocks(css=modern_css, theme=gr.themes.Base(font=[gr.themes.GoogleFont("Inter"), "sans-serif"]))
 
 with iface:
-    gr.Markdown(title_md) # Using combined Markdown for title and subtitle
+    gr.Markdown(title_md)
     gr.Markdown(description)
 
     text_input = gr.Textbox(
         label="",
         placeholder="Type or paste your content here...",
         elem_id="text_input_box",
-        lines=7 # Adjusted lines
+        lines=10 
     )
     result_output = gr.HTML(elem_id="result_output_box")
 
-    # Only set up the change function if models are loaded
     if all([tokenizer, model_1, model_2, model_3]):
         text_input.change(classify_text_interface, inputs=text_input, outputs=result_output)
     else:
-        # Display a persistent error if models couldn't load
         gr.HTML("<div id='result_output_box'><p style='color: var(--ai-color); text-align: center;'><strong>Application Error: Models could not be loaded. Please check the server console for details.</strong></p></div>")
 
+    with gr.Accordion("AI Text Examples", open=False):
+        gr.Examples(
+            examples=AI_texts,
+            inputs=text_input,
+            label="",
+            elem_classes="gr-examples"
+        )
 
-    with gr.Row():
-        with gr.Column(scale=1):
-             with gr.Accordion("AI Text Examples", open=False):
-                gr.Examples(
-                    examples=AI_texts,
-                    inputs=text_input,
-                    label="", # Label removed as accordion title is enough
-                )
-        with gr.Column(scale=1):
-            with gr.Accordion("Human Text Examples", open=False):
-                gr.Examples(
-                    examples=Human_texts,
-                    inputs=text_input,
-                    label="", # Label removed
-                )
+    with gr.Accordion("Human Text Examples", open=False):
+        gr.Examples(
+            examples=Human_texts,
+            inputs=text_input,
+            label="", 
+        )
 
     gr.Markdown(bottom_text, elem_id="bottom_text")