Db Connection added

README.md

@@ -9,3 +9,5 @@ license: apache-2.0
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+https://vaibhav84-resumeapi.hf.space/docs

UploadFile.py

@@ -0,0 +1,27 @@
+from datetime import date
+import psycopg2
+from psycopg2 import sql
+import pandas as pd
+
+class UploadOpenFile:
+
+    def uploadFile(text,filePath,db_params):
+
+        conn = psycopg2.connect(**db_params)
+        cursor = conn.cursor()    
+        query = "Select max(jdmasterid) from JdMaster"                
+        df = pd.read_sql_query(query, conn)
+        try:
+            MasterId = df.iat[0,0] + 1
+        except:
+            MasterId =1
+
+        #print(MasterId)
+        query =sql.SQL("""INSERT INTO  JDMaster (jdmasterid,jobdescription, filename, UploadedDate, IsDetailsExtracted,IsSkillsExtracted,source) VALUES (%s,%s,%s,%s,%s,%s,%s)""")
+        cursor.execute(query, (MasterId,text,filePath, date.today(),0,0,"JD")) 
+        conn.commit()
+        cursor.close()
+        conn.close()
+        print("File Uploaded...")
+        return "File Uploaded Successfully!!!"
+        

app.py

@@ -4,11 +4,20 @@ import json
 from psycopg2 import sql
 import pandas as pd
 from DbConnection import DbConnection
+from UploadFile import UploadOpenFile
+from pydantic import BaseModel
 app = FastAPI()
+
+class FileDetails(BaseModel):
+    filecontents: str
+    filename: str
+    
+
+
 @app.get("/")
 async def root():
  return {"greeting":"Hello world"}
-
+#https://vaibhav84-resumeapi.hf.space/docs
 db_params = DbConnection.GetDbConnection()
 def parse_csv(df):
     res = df.to_json(orient="records")
@@ -20,4 +29,8 @@ def ProfileMatchResults():
    dbQuery = "select * from profilematch"
    conn = psycopg2.connect(**db_params)   
    df = pd.read_sql_query(dbQuery, conn)
-   return parse_csv(df) 
+   return parse_csv(df) 
+
+@app.post("/UploadFile/")
+def UploadFileDetails(file_data: FileDetails):
+   return UploadOpenFile.uploadFile(file_data.filecontents,file_data.filename,db_params)