test layout 2

app.py

@@ -41,25 +41,22 @@ class ArxivSearch:
             gr.Markdown("Search arXiv papers by keyword and embedding model.")
 
             self.plot_output = gr.Plot()
-            self.output_md = gr.Markdown()
             
             with gr.Row():
                 self.query_box = gr.Textbox(lines=1, placeholder="Enter your search query", label="Query")
                 self.embedding_dropdown.render()
                 self.plot_button.render()
+                with gr.Column():
+                    self.search_button = gr.Button("Search")
 
+            self.output_md = gr.Markdown()
+            
             self.query_box.submit(
                 self.search_function,
                 inputs=[self.query_box, self.embedding_dropdown],
                 outputs=self.output_md
             )
-            # Add a search button
-            self.search_button = gr.Button("Search")
-            self.search_button.click(
-                self.search_function,
-                inputs=[self.query_box, self.embedding_dropdown],
-                outputs=self.output_md
-            )
+
             self.embedding_dropdown.change(
                 self.search_function,
                 inputs=[self.query_box, self.embedding_dropdown],
@@ -70,6 +67,11 @@ class ArxivSearch:
                 inputs=[self.embedding_dropdown],
                 outputs=self.plot_output
             )
+            self.search_button.click(
+                self.search_function,
+                inputs=[self.query_box, self.embedding_dropdown],
+                outputs=self.output_md
+            )
 
         # self.iface = gr.Interface(
         #     fn=self.search_function,