Update app.py

app.py

@@ -1,8 +1,3 @@
-
-
-
-
-
 import gradio as gr
 from detector import CustomDetector
 import logging
@@ -100,6 +95,13 @@ button {
     font-weight: 500 !important;
 }
 
+/* Accordion */
+.gr-accordion {
+    margin-top: 1rem;
+    border: 1px solid #d1d5db;
+    border-radius: 6px;
+}
+
 /* Responsive design */
 @media (max-width: 600px) {
     .gradio-container {
@@ -113,20 +115,34 @@ button {
 }
 """
 
+# Citation for the expandable tab
+citation_markdown = """
+## Citation
+
+Please cite our work as:
+
+**Zero-Shot Statistical Tests for LLM-Generated Text Detection using Finite Sample Concentration Inequalities**  
+Tara Radvand, Mojtaba Abdolmaleki, Mohamed Mostagir, Ambuj Tewari  
+[arXiv:2501.02406](https://arxiv.org/abs/2501.02406)  
+Year: 2025
+"""
+
 # Set up the Gradio interface
-iface = gr.Interface(
-    fn=detect_text,
-    inputs=gr.Textbox(
+with gr.Blocks(css=custom_css, theme=None) as iface:
+    gr.Markdown("# AI-Generated Text Detector")
+    gr.Markdown("Enter text to detect if it was generated by an AI model. Powered by a custom detector using tiiuae/falcon-rw-1b.")
+    gr.Textbox(
         lines=5,
         placeholder="Enter text here to check if it's AI-generated...",
         label="Input Text"
-    ),
-    outputs=gr.Textbox(label="Detection Result"),
-    title="AI-Generated Text Detector",
-    description="Enter text to detect if it was generated by an AI model. Powered by a custom detector using tiiuae/falcon-rw-1b.",
-    css=custom_css,
-    theme=None
-)
+    )
+    output = gr.Textbox(label="Detection Result")
+    gr.Button("Detect").click(
+        fn=detect_text,
+        inputs=gr.Textbox(),
+        outputs=output
+    )
+    gr.Accordion("Citation", open=False).markdown(citation_markdown)
 
 # Launch the app
 if __name__ == "__main__":