Refactor and enhance data import process for Supabase integration

- Improved function documentation and comments for clarity.
- Updated `create_table_if_not_exists` to accept DataFrame directly and added logging for SQL execution.
- Enhanced `import_data_to_supabase` to include a wait time for schema caching and improved error handling during data insertion.
- Modified Gradio interface labels for better user experience.

app.py

@@ -12,15 +12,18 @@ import supabase
 # クライアントの初期化
 supabase_client = supabase.create_client(SUPABASE_URL, SUPABASE_KEY)
 
-import math
-from io import StringIO
-
 import gradio as gr
 import pandas as pd
+import math
+import time
+from io import StringIO
+import supabase 
 
-
+# =============================
+# 型変換ヘルパー
+# =============================
 def pandas_dtype_to_supabase_type(dtype):
-    """Maps pandas dtypes to Supabase(PostgreSQL) types."""
+    """Maps pandas data types to Supabase SQL data types."""
     if pd.api.types.is_integer_dtype(dtype):
         return "integer"
     elif pd.api.types.is_float_dtype(dtype):
@@ -32,14 +35,15 @@ def pandas_dtype_to_supabase_type(dtype):
     else:
         return "text"
 
-
-def create_table_if_not_exists(db_name: str, df: pd.DataFrame):
+# =============================
+# テーブル作成関数
+# =============================
+def create_table_if_not_exists(db_name, dataframe):
     """
-    Supabase 上にテーブルが無ければ作成します。
-    'exec_sql(sql text)' という RPC を作成してある前提です。
+    Creates a table in Supabase if it doesn't exist, using RPC `exec_sql`.
     """
     columns = []
-    for col_name, dtype in df.dtypes.items():
+    for col_name, dtype in dataframe.dtypes.items():
         supabase_type = pandas_dtype_to_supabase_type(dtype)
         columns.append(f'"{col_name}" {supabase_type}')
 
@@ -50,9 +54,11 @@ def create_table_if_not_exists(db_name: str, df: pd.DataFrame):
         {columns_sql}
     );
     """
+    print(f"Executing SQL for table creation:\n{create_table_sql}")
 
     try:
         response = supabase_client.rpc("exec_sql", {"sql": create_table_sql}).execute()
+
         resp_data = getattr(response, "data", None)
         resp_error = getattr(response, "error", None)
 
@@ -65,11 +71,17 @@ def create_table_if_not_exists(db_name: str, df: pd.DataFrame):
             return True, f"Table '{db_name}' created or already exists. (no data returned)"
 
     except Exception as e:
+        print(f"Exception during table creation RPC call: {e}")
         return False, f"Exception during table creation: {e}"
 
-
+# =============================
+# DataFrame の JSON 変換ヘルパー
+# =============================
 def clean_dataframe_for_json(df: pd.DataFrame):
-    """DataFrame → JSON互換の list[dict] に変換"""
+    """
+    Convert DataFrame to JSON-compliant list of dicts
+    (replace NaN, NaT, inf, -inf with None).
+    """
     records = df.to_dict(orient="records")
     cleaned = []
     for row in records:
@@ -84,52 +96,51 @@ def clean_dataframe_for_json(df: pd.DataFrame):
         cleaned.append(new_row)
     return cleaned
 
-
+# =============================
+# メイン処理関数
+# =============================
 def import_data_to_supabase(text_input: str, db_name: str):
-    """
-    タブ区切りの貼り付けテキストを DataFrame に変換し、
-    Supabase のテーブルへ挿入する。
-    """
+    db_name = db_name.strip().lower()  # 小文字に統一
     full_message = []
 
-    if not (text_input and text_input.strip()):
-        return "Please paste tab-separated data."
-    if not (db_name and db_name.strip()):
-        return "Please provide a table name."
-
     try:
-        # 1) タブ区切り読み込み、'null' を欠損値扱い
+        # 1. DataFrame 読み込み（タブ区切り & null対応）
         df = pd.read_csv(
             StringIO(text_input),
             sep="\t",
             na_values=["null", "NULL", "NaN", ""],
-            keep_default_na=True,
+            keep_default_na=True
         )
 
-        # 2) テーブル作成
-        ok, msg = create_table_if_not_exists(db_name.strip(), df)
-        full_message.append(f"Table Creation Status: {msg}")
+        # 2. テーブル作成
+        table_creation_success, table_creation_msg = create_table_if_not_exists(db_name, df)
+        full_message.append(f"Table Creation Status: {table_creation_msg}")
 
-        if not ok:
+        if not table_creation_success:
             return "\n".join(full_message)
 
-        # 3) JSON 互換にクリーニング
+        # 3. 数秒待機（スキーマキャッシュ反映待ち）
+        time.sleep(3)
+
+        # 4. JSON対応にクリーニング（NaN, NaT, inf → None）
         data = clean_dataframe_for_json(df)
+
         if not data:
-            full_message.append("Data Insertion Status: Input text empty or not parsed. No data inserted.")
+            full_message.append("Data Insertion Status: No data to insert.")
             return "\n".join(full_message)
 
-        # 4) データ挿入
-        response = supabase_client.table(db_name.strip()).insert(data).execute()
-        resp_data = getattr(response, "data", None)
-        resp_error = getattr(response, "error", None)
+        # 5. データ挿入
+        print(f"Attempting to insert {len(data)} rows into table '{db_name}'.")
+        response = supabase_client.table(db_name).insert(data).execute()
 
-        if resp_error:
+        if getattr(response, "error", None):
             full_message.append(
-                f"Data Insertion Status: Error inserting data - {getattr(resp_error, 'message', str(resp_error))}"
+                f"Data Insertion Status: Error inserting data - {response.error}"
+            )
+        elif getattr(response, "data", None):
+            full_message.append(
+                f"Data Insertion Status: Successfully inserted {len(response.data)} rows."
             )
-        elif resp_data is not None:
-            full_message.append(f"Data Insertion Status: Successfully inserted {len(resp_data)} rows.")
         else:
             full_message.append("Data Insertion Status: Insert executed, but no data returned.")
 
@@ -138,22 +149,18 @@ def import_data_to_supabase(text_input: str, db_name: str):
 
     return "\n".join(full_message)
 
-
+# =============================
 # Gradio アプリ
+# =============================
 iface = gr.Interface(
     fn=import_data_to_supabase,
     inputs=[
-        gr.Textbox(
-            label="貼り付けデータ（タブ区切り）",
-            lines=12,
-            placeholder="項目名\t下限\t上限\n曝気空気量（微小動物槽-1）\t11.3\t16.5\n…"
-        ),
-        gr.Textbox(label="Supabase テーブル名（例: thresholds）"),
+        gr.Textbox(label="Paste TSV Data Here", lines=15, placeholder="ここにタブ区切りデータを貼り付け"),
+        gr.Textbox(label="Supabase Database Table Name")
     ],
     outputs="text",
-    title="Supabase Data Importer (Paste TSV)",
-    description="Excel からコピーしたタブ区切りデータを貼り付け、Supabase に保存します。",
+    title="Supabase Data Importer",
+    description="タブ区切りデータを貼り付け、Supabase データベースのテーブルにインポートします。テーブルが存在しない場合は新しく作成します。"
 )
 
-if __name__ == "__main__":
-    iface.launch()
+iface.launch()