Update app.py

app.py

@@ -11,10 +11,10 @@ classifier = load_model("ngocminhta/authscan-baseline")
 classifier2 = load_model("ngocminhta/authscan-baseline-machine")
 
 TEXT_CLASS_MAPPING_MACHINE = {
-    'LABEL_0': 'Gemini 1.5 Pro',
-    'LABEL_1': 'Gemini 2.0 Experimental',
-    'LABEL_2': 'GPT-4o Mini',
-    'LABEL_3': 'Llama 3.1 8B'
+    'LABEL_0': 'Gemini 1.5 Family',
+    'LABEL_1': 'Gemini 2.0 Family',
+    'LABEL_2': 'GPT-4o Family',
+    'LABEL_3': 'Llama 3.1 Family'
 }
 
 TEXT_CLASS_MAPPING = {
@@ -49,53 +49,16 @@ def process_result_detection_tab(text, language):
 
 def clear_detection_tab():
     return "", gr.update(interactive=False)
+    
+theme = gr.themes.Soft(
+    primary_hue="teal",
+    font=[gr.themes.GoogleFont('Open Sans'), 'ui-sans-serif', 'system-ui', 'sans-serif'],
+    font_mono=[gr.themes.GoogleFont('Roboto Mono'), 'ui-monospace', 'Consolas', 'monospace'],
+)
 
-css = """
-body.dark-mode {
-    background-color: #2c2c2c;
-    color: white;
-}
-body.light-mode {
-    background-color: white;
-    color: black;
-}
-#dark-mode-toggle {
-    position: fixed;
-    top: 10px;
-    right: 10px;
-    z-index: 1000;
-    padding: 10px 15px;
-    background-color: #007bff;
-    color: white;
-    border: none;
-    border-radius: 5px;
-    cursor: pointer;
-    font-size: 14px;
-}
-#dark-mode-toggle:hover {
-    background-color: #0056b3;
-}
-"""
-
-js = """
-function toggleDarkMode() {
-    const body = document.body;
-    if (body.classList.contains('dark-mode')) {
-        body.classList.remove('dark-mode');
-        body.classList.add('light-mode');
-    } else {
-        body.classList.remove('light-mode');
-        body.classList.add('dark-mode');
-    }
-}
-"""
-
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+with gr.Blocks(theme=theme) as demo:
     gr.Markdown("""<h1><center>AuthScan</center></h1>""")
-    gr.HTML("<script>{}</script>".format(js))
-    gr.HTML(
-        '<button id="dark-mode-toggle" onclick="toggleDarkMode()">Switch</button>'
-    )
+    
     with gr.Tab("Code Detection"):
         with gr.Row():
             language = gr.Dropdown(