Enhance search functionality and UI components

- Added a new example button for "a bursty, star forming galaxy"
- Updated callback function to handle the new button click
- Changed distance logging to query similarity in search results
- Adjusted modal image size for better display

app.py

@@ -127,7 +127,6 @@ def main():
         host=args.host,
         port=args.port
     )
-
 app = create_app('aionsearchmodel.pt')
 server = app.server
 

src/callbacks.py

@@ -170,11 +170,12 @@ def register_callbacks(app, search_service: SearchService):
          Input("example-3", "n_clicks"),
          Input("example-5", "n_clicks"),
          Input("example-6", "n_clicks"),
-         Input("example-7", "n_clicks")],
+         Input("example-7", "n_clicks"),
+         Input("example-8", "n_clicks")],
         [State("search-button", "n_clicks")],
         prevent_initial_call=True
     )
-    def trigger_search_from_examples(click1, click2, click3, click5, click6, click7, current_clicks):
+    def trigger_search_from_examples(click1, click2, click3, click5, click6, click7, click8, current_clicks):
         """Trigger search when example buttons are clicked."""
         ctx = callback_context
         if not ctx.triggered:
@@ -188,7 +189,8 @@ def register_callbacks(app, search_service: SearchService):
             "example-3": "galaxy with stream",
             "example-5": "A violent merger in progress with visible tidal features",
             "example-6": "Low surface brightness",
-            "example-7": "face-on ring galaxy"
+            "example-7": "face-on ring galaxy",
+            "example-8": "a bursty, star forming galaxy"
         }
 
         search_query = example_queries.get(button_id, "")
@@ -472,7 +474,7 @@ def register_callbacks(app, search_service: SearchService):
                 'ra': search_data['ra'],
                 'dec': search_data['dec'],
                 'r_mag': search_data['r_mag'],
-                'distance': search_data['distance'],
+                'query_similarity': search_data['distance'],
                 'cutout_url': search_data['cutout_url']
             })
 
@@ -784,7 +786,7 @@ def build_modal_content(galaxy_info: dict) -> tuple:
     """
     image_element = html.Img(
         src=galaxy_info["cutout_url"],
-        style={"width": "100%", "max-width": "500px", "height": "auto"}
+        style={"width": "100%", "max-width": "350px", "height": "auto"}
     )
 
     # Format primary key label (convert snake_case to Title Case)

src/components.py

@@ -553,7 +553,9 @@ def create_search_container():
                         dbc.Button([html.I(className="fas fa-moon me-2"), "Low surface brightness"],
                                  id="example-6", className="example-button me-2 mb-2", size="sm", color="light"),
                         dbc.Button([html.I(className="fas fa-ring me-2"), "Ring galaxy"],
-                                 id="example-7", className="example-button mb-2", size="sm", color="light")
+                                 id="example-7", className="example-button me-2 mb-2", size="sm", color="light"),
+                        dbc.Button([html.I(className="fas fa-star me-2"), "a bursty, star forming galaxy"],
+                                 id="example-8", className="example-button mb-2", size="sm", color="light")
                     ], className="text-center")
                 ], className="mb-3"),
 
@@ -825,7 +827,7 @@ def create_galaxy_modal():
         ]),
         dbc.ModalFooter([
             dbc.Button(
-                [html.I(className="fas fa-plus-circle me-2"), "Add to Advanced Search"],
+                [html.I(className="fas fa-plus me-2"), "Search for similar"],
                 id="add-to-advanced-search",
                 color="primary",
                 className="me-2"

src/utils.py

@@ -36,7 +36,7 @@ def log_zilliz_query(
     query_time: float,
     cost_vcu: int = 0
 ) -> None:
-    """Log Zilliz queries to a file in logs/ directory.
+    """Print Zilliz query info to terminal instead of saving to file.
 
     Args:
         query_type: Type of query (e.g., "vector_search", "text_search")
@@ -45,17 +45,13 @@ def log_zilliz_query(
         query_time: Query execution time in seconds
         cost_vcu: Cost in vCU units
     """
-    logs_dir = Path("logs")
-    logs_dir.mkdir(parents=True, exist_ok=True)
-
-    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
-    log_file = logs_dir / f"zilliz_query_{timestamp}.json"
+    timestamp = datetime.now().isoformat()
 
     # Convert vCU cost to dollars
     cost_usd = (cost_vcu / 1e6) * VCU_COST_PER_MILLION
 
     log_data = {
-        "timestamp": datetime.now().isoformat(),
+        "timestamp": timestamp,
         "query_type": query_type,
         "query_info": query_info,
         "result_count": result_count,
@@ -64,11 +60,15 @@ def log_zilliz_query(
         "cost_usd": cost_usd
     }
 
-    with open(log_file, 'w') as f:
-        json.dump(log_data, f, indent=2)
+    # Print to terminal
+    print("\n" + "="*80)
+    print(f"ZILLIZ QUERY: {query_type}")
+    print("="*80)
+    print(json.dumps(log_data, indent=2))
+    print("="*80 + "\n")
 
     logger.info(
-        f"Query logged to {log_file} | {result_count} results in {query_time:.3f}s | "
+        f"{result_count} results in {query_time:.3f}s | "
         f"{cost_vcu} vCU (${cost_usd:.6f})"
     )
 
@@ -162,34 +162,19 @@ def log_query_to_csv(
     query_xml: str,
     csv_path: str = "logs/query_log.csv"
 ) -> None:
-    """Log a query to CSV file with datetime and XML string.
+    """Print query XML to terminal instead of saving to file.
 
     Args:
         query_xml: XML string representation of the query
-        csv_path: Path to the CSV log file
+        csv_path: Deprecated parameter (kept for backward compatibility)
     """
-    import csv
-    import os
-
-    # Create logs directory if it doesn't exist
-    log_dir = Path(csv_path).parent
-    log_dir.mkdir(parents=True, exist_ok=True)
-
-    # Prepare log entry
     timestamp = datetime.now().isoformat()
 
-    # Check if file exists to determine if we need to write header
-    file_exists = Path(csv_path).exists()
-
-    # Append to CSV
-    with open(csv_path, 'a', newline='', encoding='utf-8') as f:
-        writer = csv.writer(f)
-
-        # Write header if file is new
-        if not file_exists:
-            writer.writerow(['datetime', 'query'])
-
-        # Write the query log
-        writer.writerow([timestamp, query_xml])
+    # Print query to terminal
+    print("\n" + "="*80)
+    print(f"QUERY EXECUTED AT: {timestamp}")
+    print("="*80)
+    print(query_xml)
+    print("="*80 + "\n")
 
-    logger.info(f"Query logged to {csv_path}")
+    logger.info(f"Query printed to terminal")