Spaces:
Running
on
Zero
Running
on
Zero
test
Browse files
polls/databases/insert_line.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import pandas as pd
|
2 |
import re
|
3 |
-
from sqlalchemy import create_engine, inspect, Table, MetaData, Column, String, text
|
4 |
import psycopg2
|
5 |
|
6 |
# チャット履歴ファイルを読み込む関数(最初の3行をスキップ)
|
@@ -41,7 +41,7 @@ def connect_to_db():
|
|
41 |
return conn
|
42 |
|
43 |
# ファイルパスの設定
|
44 |
-
input_file_path = '/
|
45 |
|
46 |
# チャット履歴を読み込み、特定の列の個人情報をマスク
|
47 |
chat_history_df = load_chat_history(input_file_path)
|
@@ -55,7 +55,9 @@ engine = create_engine('postgresql+psycopg2://miyataken999:yz1wPf4KrWTm@ep-odd-m
|
|
55 |
if not inspect(engine).has_table("fasis_chat_history"):
|
56 |
metadata = MetaData()
|
57 |
Table('fasis_chat_history', metadata,
|
58 |
-
|
|
|
|
|
59 |
metadata.create_all(engine)
|
60 |
|
61 |
# マスクされたデータをPostgreSQLにインポート
|
|
|
1 |
import pandas as pd
|
2 |
import re
|
3 |
+
from sqlalchemy import create_engine, inspect, Table, MetaData, Column, Integer, String, text, Sequence
|
4 |
import psycopg2
|
5 |
|
6 |
# チャット履歴ファイルを読み込む関数(最初の3行をスキップ)
|
|
|
41 |
return conn
|
42 |
|
43 |
# ファイルパスの設定
|
44 |
+
input_file_path = 'path/to/your/chat_history.csv' # エクスポートされたチャット履歴ファイルのパス
|
45 |
|
46 |
# チャット履歴を読み込み、特定の列の個人情報をマスク
|
47 |
chat_history_df = load_chat_history(input_file_path)
|
|
|
55 |
if not inspect(engine).has_table("fasis_chat_history"):
|
56 |
metadata = MetaData()
|
57 |
Table('fasis_chat_history', metadata,
|
58 |
+
Column('id', Integer, Sequence('user_id_seq'), primary_key=True, autoincrement=True),
|
59 |
+
*(Column(name, String) for name in masked_chat_history_df.columns),
|
60 |
+
Column('AnswerField', String))
|
61 |
metadata.create_all(engine)
|
62 |
|
63 |
# マスクされたデータをPostgreSQLにインポート
|