supabase image url and file name mapping

supabase_service.py

@@ -18,30 +18,43 @@ BUCKET_NAME = "csvcharts"
 async def upload_image_to_supabase(file_path: str, file_name: str) -> str:
     """
     Uploads an image to Supabase Storage and returns the public URL.
-    
-    :param file_path: Path to the image file on your local machine.
-    :param file_name: Name to save the file as in Supabase Storage.
-    :return: Public URL of the uploaded image.
+    Also saves the mapping between public_url and chart_name in the database.
     """
-    # Check if the file exists
     if not os.path.exists(file_path):
         raise FileNotFoundError(f"The file {file_path} does not exist.")
 
-    # Read the file in binary mode
     with open(file_path, "rb") as f:
         file_data = f.read()
 
-    # Upload the file to Supabase Storage
     try:
         res = supabase.storage.from_(BUCKET_NAME).upload(file_name, file_data)
-        # print("Upload response:", res)  # Debugging: Print the response
+        print("Upload response:", res)
     except Exception as e:
-        print(f"Error uploading file: {e}")
         raise Exception(f"Failed to upload file: {e}")
 
-    # Get the public URL of the uploaded file
     public_url = supabase.storage.from_(BUCKET_NAME).get_public_url(file_name)
-    # print("Public URL:", public_url)  # Debugging: Print the public URL
+    print("Public URL:", public_url)
+
+    # Save the mapping to the database
+    try:
+        supabase.table("chart_mappings").insert({
+            "public_url": public_url,
+            "chart_name": file_name
+        }).execute()
+    except Exception as e:
+        raise Exception(f"Failed to save mapping to database: {e}")
 
     return public_url
 
+async def get_chart_name_by_url(public_url: str) -> str:
+    """
+    Retrieves the chart_name using the public_url from the database.
+    """
+    try:
+        response = supabase.table("chart_mappings").select("chart_name").eq("public_url", public_url).execute()
+        if response.data:
+            return response.data[0]["chart_name"]
+        else:
+            return None
+    except Exception as e:
+        raise Exception(f"Failed to retrieve chart_name: {e}")