Refactor data import functionality to support tab-separated input and improve Supabase integration

- Updated `import_data_to_supabase` to accept tab-separated text input instead of CSV files.
- Enhanced table creation logic to handle DataFrame directly and improved error handling.
- Modified Gradio interface to reflect changes in input method and updated labels for clarity.
- Cleaned up code comments and improved function documentation for better understanding.

app.py

@@ -7,17 +7,20 @@ SUPABASE_URL = os.environ.get('SUPABASE_URL')
 
 SUPABASE_KEY = os.environ.get('SUPABASE_KEY')
 
-import gradio as gr
-import pandas as pd
+import supabase
+
+# クライアントの初期化
+supabase_client = supabase.create_client(SUPABASE_URL, SUPABASE_KEY)
+
 import math
+from io import StringIO
 
-# ▼ Supabaseクライアントは事前に初期化しておく必要があります
-# from supabase import create_client
-# supabase_client = create_client(SUPABASE_URL, SUPABASE_KEY)
+import gradio as gr
+import pandas as pd
 
 
 def pandas_dtype_to_supabase_type(dtype):
-    """Maps pandas data types to Supabase SQL data types."""
+    """Maps pandas dtypes to Supabase(PostgreSQL) types."""
     if pd.api.types.is_integer_dtype(dtype):
         return "integer"
     elif pd.api.types.is_float_dtype(dtype):
@@ -30,12 +33,13 @@ def pandas_dtype_to_supabase_type(dtype):
         return "text"
 
 
-def create_table_if_not_exists(db_name, dataframe):
+def create_table_if_not_exists(db_name: str, df: pd.DataFrame):
     """
-    Creates a table in Supabase if it doesn't exist, using RPC `exec_sql`.
+    Supabase 上にテーブルが無ければ作成します。
+    'exec_sql(sql text)' という RPC を作成してある前提です。
     """
     columns = []
-    for col_name, dtype in dataframe.dtypes.items():
+    for col_name, dtype in df.dtypes.items():
         supabase_type = pandas_dtype_to_supabase_type(dtype)
         columns.append(f'"{col_name}" {supabase_type}')
 
@@ -46,11 +50,9 @@ def create_table_if_not_exists(db_name, 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)
 
@@ -63,15 +65,11 @@ def create_table_if_not_exists(db_name, 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}"
 
 
 def clean_dataframe_for_json(df: pd.DataFrame):
-    """
-    Convert DataFrame to JSON-compliant list of dicts
-    (replace NaN, NaT, inf, -inf with None).
-    """
+    """DataFrame → JSON互換の list[dict] に変換"""
     records = df.to_dict(orient="records")
     cleaned = []
     for row in records:
@@ -87,36 +85,42 @@ def clean_dataframe_for_json(df: pd.DataFrame):
     return cleaned
 
 
-def import_data_to_supabase(csv_file, db_name):
+def import_data_to_supabase(text_input: str, db_name: str):
     """
-    Imports data from a CSV file to a Supabase table,
-    creating the table if it doesn't exist.
+    タブ区切りの貼り付けテキストを DataFrame に変換し、
+    Supabase のテーブルへ挿入する。
     """
-    global supabase_client
     full_message = []
 
-    if not csv_file or not db_name:
-        return "Please upload a CSV file and provide a table name."
+    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. CSV読込
-        df = pd.read_csv(csv_file.name)
-
-        # 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}")
+        # 1) タブ区切り読み込み、'null' を欠損値扱い
+        df = pd.read_csv(
+            StringIO(text_input),
+            sep="\t",
+            na_values=["null", "NULL", "NaN", ""],
+            keep_default_na=True,
+        )
+
+        # 2) テーブル作成
+        ok, msg = create_table_if_not_exists(db_name.strip(), df)
+        full_message.append(f"Table Creation Status: {msg}")
+
+        if not ok:
+            return "\n".join(full_message)
 
-        # 3. JSON対応にクリーニング（NaN, NaT, inf → None）
+        # 3) JSON 互換にクリーニング
         data = clean_dataframe_for_json(df)
-
         if not data:
-            full_message.append("Data Insertion Status: CSV file empty or not parsed. No data inserted.")
+            full_message.append("Data Insertion Status: Input text empty or not parsed. No data inserted.")
             return "\n".join(full_message)
 
-        # 4. データ挿入
-        print(f"Attempting to insert {len(data)} rows into table '{db_name}'.")
-        response = supabase_client.table(db_name).insert(data).execute()
-
+        # 4) データ挿入
+        response = supabase_client.table(db_name.strip()).insert(data).execute()
         resp_data = getattr(response, "data", None)
         resp_error = getattr(response, "error", None)
 
@@ -124,7 +128,7 @@ def import_data_to_supabase(csv_file, db_name):
             full_message.append(
                 f"Data Insertion Status: Error inserting data - {getattr(resp_error, 'message', str(resp_error))}"
             )
-        elif resp_data:
+        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.")
@@ -139,12 +143,17 @@ def import_data_to_supabase(csv_file, db_name):
 iface = gr.Interface(
     fn=import_data_to_supabase,
     inputs=[
-        gr.File(label="Upload CSV File"),
-        gr.Textbox(label="Supabase Database Table Name")
+        gr.Textbox(
+            label="貼り付けデータ（タブ区切り）",
+            lines=12,
+            placeholder="項目名\t下限\t上限\n曝気空気量（微小動物槽-1）\t11.3\t16.5\n…"
+        ),
+        gr.Textbox(label="Supabase テーブル名（例: thresholds）"),
     ],
     outputs="text",
-    title="Supabase Data Importer",
-    description="CSVファイルをアップロードし、データをSupabaseデータベースのテーブルにインポートします。指定されたテーブルが存在しない場合は新しく作成します。"
+    title="Supabase Data Importer (Paste TSV)",
+    description="Excel からコピーしたタブ区切りデータを貼り付け、Supabase に保存します。",
 )
 
-iface.launch(mcp_server=True)
+if __name__ == "__main__":
+    iface.launch()