Update app.py

app.py

@@ -2,51 +2,40 @@ import gradio as gr
 from datetime import datetime
 
 def search(query, collections):
-    # Placeholder function where you'd integrate your custom RAG logic
-    # For now, it just echoes the input back with a mock response
+    # Placeholder function for custom RAG integration
     response = f"Results for query: {query} from collections: {', '.join(collections)}"
     return response
 
 # Date for the index last updated
 last_updated_date = datetime.now().strftime("%B %d, %Y")
 
-# Interface components
-title_block = gr.HTML("""
-                      <h2>FindMyPlot</h2>
-                      <p>Find a plot within experimental high energy physics collider results.<br>
-                      Index last updated {}.</p>
-                      """.format(last_updated_date))
-
-query_input = gr.Textbox(label="Search", placeholder="Enter your query here...", lines=2)
-collections = gr.CheckboxGroup(choices=["ATLAS_Conference_Notes", "ATLAS_Papers", "CMS_Papers", "ATLAS_PUB_Notes", "CMS_Physics_Analysis_Summaries"],
-                               label="Advanced Options: Select Collections",
-                               value=["ATLAS_Conference_Notes", "ATLAS_Papers"], visible=False)  # By default, not showing advanced options
-
-# Defining the function that will toggle visibility of the advanced options
-def toggle_advanced_options(is_clicked):
-    collections.visible = not collections.visible
-    return gr.update(visible=collections.visible)
-
-advanced_options_button = gr.Button("Advanced Options", elem_id="advanced_options_button")
-advanced_options_button.click(fn=toggle_advanced_options, inputs=advanced_options_button, outputs=collections)
-
-# Binding the input, options, and the search function together
-output_chat = gr.Textbox(label="Results", lines=6, interactive=True, elem_id="results_area")
-search_button = gr.Button("Search")
-search_interface = gr.Interface(fn=search, inputs=[query_input, collections], outputs=output_chat)
+# Define the function to toggle the visibility of advanced options
+def toggle_advanced_options(is_clicked, collections_visibility):
+    return not collections_visibility
 
 app = gr.Blocks()
 
 with app:
-    gr.Row(title_block)
-    gr.Row(gr.Column(query_input, advanced_options_button, collections, search_button, output_chat))
-
-# Styling
-app.css = """
-    #results_area { margin-top: 20px; }
-    #advanced_options_button { margin-top: 20px; margin-bottom: 20px; }
-"""
+    gr.Markdown(f"### FindMyPlot\nFind a plot within experimental high energy physics collider results.\nIndex last updated {last_updated_date}.")
+    
+    query_input = gr.Textbox(label="Search", placeholder="Enter your query here...", lines=2)
+    collections = gr.CheckboxGroup(choices=["ATLAS_Conference_Notes", "ATLAS_Papers", "CMS_Papers", "ATLAS_PUB_Notes", "CMS_Physics_Analysis_Summaries"],
+                                   label="Select Collections",
+                                   value=["ATLAS_Conference_Notes", "ATLAS_Papers"],
+                                   visible=False)  # Start hidden
+    
+    advanced_options_button = gr.Button("Advanced Options")
+    search_button = gr.Button("Search")
+    output_chat = gr.Textbox(label="Results", lines=6, interactive=True)
+    
+    advanced_options_button.click(fn=toggle_advanced_options, 
+                                  inputs=advanced_options_button, 
+                                  outputs=collections,
+                                  _js="(is_clicked, collections_visibility) => [!collections_visibility]")
+    
+    search_button.click(fn=search, 
+                        inputs=[query_input, collections], 
+                        outputs=output_chat)
 
-# Launch the app
 if __name__ == "__main__":
     app.launch()