Update app.py

app.py

@@ -131,7 +131,7 @@ def verify_user(username: str, password: str):
     try:
         cnx = get_db_connection()
         cursor = cnx.cursor()
-        query = "SELECT role FROM user_detail WHERE username = %s AND password = %s"
+        query = "SELECT role,company_id FROM user_detail WHERE username = %s AND password = %s"
         values = (username, password)
         cursor.execute(query, values)
         result = cursor.fetchone()
@@ -139,7 +139,7 @@ def verify_user(username: str, password: str):
         cnx.close()
         if result is not None:
             logging.info(f"User {username} logged in successfully")
-            return "success",result[0]
+            return "success",result[0],result[1]
         else:
             logging.info(f"User {username} login failed")
             return "failure"