chore: Refactor app.py for improved code organization and readability

app.py

@@ -138,28 +138,38 @@ h1 {
 }
 """
 
-demo = gr.Interface(
-    fn=search_by_text_and_return_images,
-    inputs=[
-        gr.Textbox(
-            label="Enter your cosmic query",
-            placeholder="e.g., alien abduction, crop circles",
-        ),
-        gr.Slider(
-            minimum=1,
-            maximum=10,
-            step=1,
-            label="Number of classified documents",
-            value=5,
-        ),
-    ],
-    outputs=gr.Gallery(label="Declassified UFO Sightings", elem_id="gallery"),
-    title="<h1>🛸 Top Secret UFO Document Search 🛸</h1>",
-    description="<marquee direction='left' scrollamount='5' class='yellow-text'>Uncover the truth that's out there! The government doesn't want you to know!</marquee>",
-    css=css,
-    allow_flagging="never",
-    theme=geocities90s,
-)
+# Replace the demo definition with this Blocks implementation
+with gr.Blocks(css=css, theme=geocities90s) as demo:
+    gr.HTML("<h1>🛸 Top Secret UFO Document Search 🛸</h1>")
+    gr.HTML(
+        "<marquee direction='left' scrollamount='5' class='yellow-text'>Uncover the truth that's out there! The government doesn't want you to know!</marquee>"
+    )
+
+    with gr.Row():
+        with gr.Column(scale=3):
+            query_input = gr.Textbox(
+                label="Enter your cosmic query",
+                placeholder="e.g., alien abduction, crop circles",
+            )
+        with gr.Column(scale=1):
+            num_results = gr.Slider(
+                minimum=1,
+                maximum=10,
+                step=1,
+                label="Number of classified documents",
+                value=5,
+            )
+
+    search_button = gr.Button("Declassify Documents")
+
+    gallery_output = gr.Gallery(label="Declassified UFO Sightings", elem_id="gallery")
+
+    search_button.click(
+        fn=search_by_text_and_return_images,
+        inputs=[query_input, num_results],
+        outputs=gallery_output,
+    )
 
+# Keep the main block
 if __name__ == "__main__":
     demo.launch()