retrieval location logic

.gitignore

@@ -0,0 +1,5 @@
+## python
+__pycache__/
+
+backend/data/embeddings/mars-mae-vit-b-2m/samples.db
+backend/data/embeddings/mars-mae-vit-b-2m/embeddings.index

frontend/src/components/RetrievalDemo.jsx

@@ -43,10 +43,13 @@ const ThumbnailImage = styled("img")({
   transition: "transform 0.3s ease-in-out",
 })
 
-const parseCoordinates = (filename) => {
+const parseCoordinates = (filename, label) => {
   const match = filename.match(/E(-?\d+)_N(-?\d+)_x(\d+)_y(\d+)/)
   if (!match) return null
 
+  const scale = label.match(/size_(\d+)/)
+  if (!scale) return null
+
   const [_, longitude, latitude, x, y] = match
   const baseLong = Number.parseInt(longitude)
   const baseLat = Number.parseInt(latitude)
@@ -56,12 +59,14 @@ const parseCoordinates = (filename) => {
   const degreesPerPixel = 4 / 47420
   const xOffset = Number.parseInt(x) * degreesPerPixel
   const yOffset = Number.parseInt(y) * degreesPerPixel
+  const centerOffset = (Number.parseInt(scale[1]) / 2 ) * degreesPerPixel
+
 
   return {
     // Format the base tile name according to the specified convention
     baseTile: `MurrayLab_CTX_V01_E${longitude}_N${latitude}_Mosaic`,
-    longitude: baseLong + xOffset,
-    latitude: baseLat + 4 - yOffset,
+    longitude: baseLong + xOffset + centerOffset,
+    latitude: baseLat + 4 - yOffset - centerOffset,
     x: Number.parseInt(x),
     y: Number.parseInt(y),
   }
@@ -170,7 +175,7 @@ const RetrievalDemo = () => {
           <Grid item xs={12} md={6}>
             <Typography variant="body1" paragraph>
               Select a sample image above or upload your own Mars image to find similar images in our database.
-              The database has 2 million images sampled from ±32°N on Mars.
+              The database has 2 million images sampled from ±32°N on Mars, from Murray Lab Global CTX Mosaic.
             </Typography>
             <Box sx={{ display: "flex", gap: 2, flexWrap: "wrap", alignItems: "center" }}>
               <label htmlFor="contained-button-file">
@@ -271,13 +276,13 @@ const RetrievalDemo = () => {
                     <Typography variant="subtitle2" sx={{ display: "none" }} title={result.filename}>
                       File: {result.filename}
                     </Typography>
-                    {parseCoordinates(result.filename) && (
+                    {parseCoordinates(result.filename, result.label) && (
                       <Typography variant="body2" color="text.secondary">
-                        Base tile: {parseCoordinates(result.filename).baseTile}
+                        Base tile: {parseCoordinates(result.filename, result.label).baseTile}
                         <br />
                         Exact location:{" "}
-                        {`${Math.abs(parseCoordinates(result.filename).longitude).toFixed(2)}°${parseCoordinates(result.filename).longitude >= 0 ? "E" : "W"}, `}
-                        {`${Math.abs(parseCoordinates(result.filename).latitude).toFixed(2)}°${parseCoordinates(result.filename).latitude >= 0 ? "N" : "S"}`}
+                        {`${Math.abs(parseCoordinates(result.filename,result.label).longitude).toFixed(2)}°${parseCoordinates(result.filename,result.label).longitude >= 0 ? "E" : "W"}, `}
+                        {`${Math.abs(parseCoordinates(result.filename,result.label).latitude).toFixed(2)}°${parseCoordinates(result.filename,result.label).latitude >= 0 ? "N" : "S"}`}
                       </Typography>
                     )}
                     <Chip