Vaibhav84 commited on
Commit
89ac774
·
verified ·
1 Parent(s): a587148

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  from fastapi import FastAPI
2
  import psycopg2
 
3
  from psycopg2 import sql
4
  import pandas as pd
5
  app = FastAPI()
@@ -13,10 +14,14 @@ db_params = {
13
  'user': 'anu',
14
  'password': '5omRLogf9Kdas3zoBFPCT9yrCGU4IbEX',
15
  }
16
-
 
 
 
 
17
  @app.get("/ProfileMatch")
18
  def ProfileMatchResults():
19
  dbQuery = "select * from profilematch"
20
  conn = psycopg2.connect(**db_params)
21
  df = pd.read_sql_query(dbQuery, conn)
22
- return {"data" : df}
 
1
  from fastapi import FastAPI
2
  import psycopg2
3
+ import json
4
  from psycopg2 import sql
5
  import pandas as pd
6
  app = FastAPI()
 
14
  'user': 'anu',
15
  'password': '5omRLogf9Kdas3zoBFPCT9yrCGU4IbEX',
16
  }
17
+ def parse_csv(df):
18
+ res = df.to_json(orient="records")
19
+ parsed = json.loads(res)
20
+ return parsed
21
+
22
  @app.get("/ProfileMatch")
23
  def ProfileMatchResults():
24
  dbQuery = "select * from profilematch"
25
  conn = psycopg2.connect(**db_params)
26
  df = pd.read_sql_query(dbQuery, conn)
27
+ return parse_csv(df)