Update app.py

app.py

@@ -11,18 +11,27 @@ from sentence_transformers import SentenceTransformer
 from sentence_transformers.util import cos_sim
 from sentence_transformers import CrossEncoder
 
-import mysql.connector
+# import mysql.connector
+import psycopg2
 
 db_user = os.environ["DB_NAME"] 
 
-mydb = mysql.connector.connect(
+# mydb = mysql.connector.connect(
+#   host=os.environ["DB_HOST"],
+#   user=os.environ["DB_USER"],
+#   port=os.environ["DB_PORT"],
+#   password=os.environ["DB_PASSWORD"]
+# )
+
+mydb = psycopg2.connect(
   host=os.environ["DB_HOST"],
   user=os.environ["DB_USER"],
   port=os.environ["DB_PORT"],
   password=os.environ["DB_PASSWORD"]
 )
 
-df_sample = pd.read_sql(f"SELECT * FROM {db_user}.mytable", mydb)
+
+df_sample = pd.read_sql(f"SELECT * FROM {db_user}.sample", mydb)
 df_bobot = pd.read_sql(f"SELECT * FROM {db_user}.bobot", mydb)
 df_bobot['BOBOT'] = [float(x.replace(",",".")) for x in df_bobot['BOBOT']]#.astype(float)
 df_bobot["JENIS_KENDARAAN"] = [x[:-2] + x[-1] if x[-2] == "." else x for x in df_bobot["JENIS_KENDARAAN"]]