Update app.py

app.py

@@ -453,7 +453,7 @@ async def get_document(company_id: str = Query(...), company_name: str = Query(.
             companies.append({
                 "row_id":row[0],
                 "company_id": row[1],
-               #"file_path":row[2],
+                "file_path":row[2],
                 "document_name": row[3],
                 "document_desc": row[4],
                 "department": row[5],
@@ -469,7 +469,55 @@ async def get_document(company_id: str = Query(...), company_name: str = Query(.
     except mysql.connector.Error as err:
         logging.error(f"Database error: {err}")
         raise HTTPException(status_code=500, detail="Internal Server Error")
-    
+        
+ #on update of modal form the data table is refresh the  value in datatable          
+ @app.get("/api/document_update")
+    async def get_document(company_id: str = Query(...)):
+    print(f"Received companyId and name: {company_id},{company_id}")  # Log rec
+    #async def get_data_connectors(company_id: str, company_name: str):
+    logging.info(f"Received request for company_id  and company_id: {company_id},{company_id}")
+    try:
+        cnx = get_db_connection()
+        cursor = cnx.cursor()
+        query = """
+            SELECT kb.kid,kb.company_id, kb.file_path, kb.document_name, kb.document_desc,kb.department,kb.version,kb.vectorDBflag,kb.last_updated
+            FROM redmind_gpt.knowledge_base kb
+            JOIN redmind_gpt.company_detail cd ON kb.company_id = cd.company_id 
+            WHERE kb.company_id = %s 
+        """
+        logging.info(f"Executing query: {query} with company_id: {company_id}")
+        values= (company_id,)
+       # logging.info(f"Query parameters: {params}")
+        print(f"Query parameters: {values}")
+
+        cursor.execute(query, values)  # Pa
+        result = cursor.fetchall()
+        logging.info(f"Query result: {result}")
+        cursor.close
+        cnx.close()
+        companies=[]
+        for row in result:
+            
+            companies.append({
+                "kid":row[0],
+                "company_id": row[1],
+               "file_path":row[2],
+                "document_name": row[3],
+                "document_desc": row[4],
+                "department": row[5],
+                "version": row[6],
+                "vectorDBflag":row[7],
+                "last_updated": row[8]                
+            })
+        if companies:
+            return companies
+        else:
+            logging.warning(f"No document found for company_id: {company_id}")
+            raise HTTPException(status_code=404, detail="Data document not found")
+    except mysql.connector.Error as err:
+        logging.error(f"Database error: {err}")
+        raise HTTPException(status_code=500, detail="Internal Server Error")
+       
 #to get data for view  in knowledgebase
 @app.get("/api/getknowledgebase/{company_id}")
 async def get_company_details(company_id: int):
@@ -503,29 +551,42 @@ async def get_knowledge_from_db(company_id: int):
         raise HTTPException(status_code=500, detail="Failed to fetch company")
     
 # to edit the knowledgebase details    
-@app.put("/api/putknowledgebase/{kid}")
-async def update_company_details( 
-                                  kid: int,
-                                  company_id: str = Form(...),
-                                 # file_name: str = Form(...),
-                                  documentName: str = Form(...),
-                                  documentDescription: str = Form(...),
-                                  department: str = Form(...),
-                                  version: str = Form(...),
-                                  vectorDBFlag: str = Form(...),
-                                  lastUpdated: str = Form(...)):
-    logging.info(f"Received request for company data with ID: {kid}")
+@app.put("/api/putknowledgebase/{companyId}")
+async def update_company_details(
+    request: Request,
+    companyId: int,
+    company_id: str = Form(...),
+    file_path: UploadFile = File(...),
+    documentName: str = Form(...),
+    documentDescription: str = Form(...),
+    department: str = Form(...),
+    version: str = Form(...),
+    vectorDBFlag: str = Form(...),
+    lastUpdated: str = Form(...)
+):
+    logging.info(f"Received request for company data with ID inside edit/update knowledgebase: {companyId}")
+    print(f"Received request for company data with ID inside edit/update knowledgebase file name: {file_path.filename}")
     
-    # upload_folder = "uploads/"
-    # os.makedirs(upload_folder, exist_ok=True)
-    # file_path = os.path.join(upload_folder, uploadFile.filename)
+    # Create the upload folder if it doesn't exist
+    upload_folder = "uploads/"
+    os.makedirs(upload_folder, exist_ok=True)
+
+    # Construct the file path for saving
+    saved_file_path = os.path.join(upload_folder, file_path.filename)
     
-    # with open(file_path, "wb") as buffer:
-    #     shutil.copyfileobj(uploadFile.file, buffer)
+    try:
+        # Save the uploaded file to the server
+        with open(saved_file_path, "wb") as buffer:
+            shutil.copyfileobj(file_path.file, buffer)
+    except Exception as e:
+        logging.error(f"Error saving file: {e}")
+        raise HTTPException(status_code=500, detail="Failed to save file")
 
+    # Prepare the company data dictionary
     company_data = {
+        'kid': companyId,
         'company_id': company_id,
-        #'file_path': file_name,
+        'file_path': saved_file_path,  # Use the path where the file was saved
         'document_name': documentName,
         'document_desc': documentDescription,
         'department': department,
@@ -534,26 +595,40 @@ async def update_company_details(
         'last_updated': lastUpdated
     }
 
-    updated_company = await update_knowledge_in_db(kid, company_data)
+    # Update the knowledge base in the database
+    updated_company = await update_knowledge_in_db(companyId, company_data)
     if not updated_company:
         raise HTTPException(status_code=500, detail="Failed to update company")
     return updated_company
+
 async def update_knowledge_in_db(kid: int, company_data: dict):
     try:
         logging.info(f"Updating knowledge base for ID: {kid}")
         cnx = get_db_connection()
         if cnx is None:
             raise HTTPException(status_code=500, detail="Failed to connect to the database")
-        
+
         cursor = cnx.cursor()
         update_query = """
-        UPDATE u852023448_redmindgptknowledge_base kb
-        SET kb.company_id,kb.document_name=%s, kb.document_desc=%s, kb.department=%s, kb.version=%s, kb.vectorDBflag=%s, kb.last_updated=%s
+        UPDATE redmind_gpt.knowledge_base kb
+        SET kb.company_id = %s, kb.document_name = %s, kb.document_desc = %s, 
+            kb.department = %s, kb.version = %s, kb.vectorDBflag = %s, kb.last_updated = %s
         WHERE kb.kid = %s;
         """
+        logging.info(f"Executing update query: {update_query}")
+        logging.info(f"With values: {(
+            company_data['company_id'],
+            company_data['document_name'],
+            company_data['document_desc'],
+            company_data['department'],
+            company_data['version'],
+            company_data['vectorDBflag'],
+            company_data['last_updated'],
+            kid
+        )}")
+
         cursor.execute(update_query, (
             company_data['company_id'],
-           # company_data['file_path'],
             company_data['document_name'],
             company_data['document_desc'],
             company_data['department'],
@@ -567,12 +642,18 @@ async def update_knowledge_in_db(kid: int, company_data: dict):
         success = cursor.rowcount > 0
         cursor.close()
         cnx.close()
+        
         if not success:
+            logging.info("No rows updated")
             return None
+        logging.info("Update successful")
         return company_data
     except mysql.connector.Error as err:
-        logging.error(f"Error updating company: {err}")
+        logging.error(f"Database error: {err}")
         raise HTTPException(status_code=500, detail="Failed to update company")
+    except Exception as e:
+        logging.error(f"Unexpected error: {e}")
+        raise HTTPException(status_code=500, detail="Unexpected error occurred")
 
 
 def delete_knowledge_from_db(company_id: int) -> bool:
@@ -594,7 +675,8 @@ def delete_knowledge_from_db(company_id: int) -> bool:
     except mysql.connector.Error as err:
         logging.error(f"Error deleting company: {err}")
         raise HTTPException(status_code=500, detail="Failed to delete company")
-    
+        
+  #to perform delete operation in knowlegebase      
 @app.delete("/api/delknowledgebase/{company_id}")
 async def delete_company(company_id: int):
     deletion_success = delete_knowledge_from_db(company_id)
@@ -619,6 +701,7 @@ async def save_data_connectors( request: Request,
                                 password: str=Form(...),
                                 selectedTables: List[str] = Form(...)):
     logging.info(f"Received form submission for database_connectors")
+    print(f"Received form submission for database_connectors")
     try:
         cnx = get_db_connection()
         cursor = cnx.cursor()
@@ -634,7 +717,7 @@ async def save_data_connectors( request: Request,
                 SET databasetype = %s, serverip = %s, port = %s, database_name = %s, username = %s, password = %s, dbtablename = %s
                 WHERE company_id = %s
             """
-            values = (",".join(database), server, port, ",".join(databaseName), username, password, ",".join(selectedTables), company_id)
+            values = (",".join(database), server, port, ",".join(databaseName), username, password or '', ",".join(selectedTables), company_id)
             logging.info(f"Executing update query: {query} with values: {values}")
             cursor.execute(query, values)
             cnx.commit()
@@ -645,7 +728,7 @@ async def save_data_connectors( request: Request,
                 INSERT INTO data_connectors(company_id, databasetype, serverip, port, database_name, username, password, dbtablename)
                 VALUES (%s, %s, %s, %s, %s, %s, %s, %s)
             """
-            values = (company_id, ",".join(database), server, port, ",".join(databaseName), username, password, ",".join(selectedTables))
+            values = (company_id, ",".join(database), server, port, ",".join(databaseName), username, password or '', ",".join(selectedTables))
             logging.info(f"Executing insert query: {query} with values: {values}")
             cursor.execute(query, values)
             cnx.commit()
@@ -653,26 +736,23 @@ async def save_data_connectors( request: Request,
 
         cursor.close()
         cnx.close()
-        logging.info(f"Data_connectors for {database} processed successfully")
-        return RedirectResponse(url="/data_connectors", status_code=302)
-
+        
+        # logging.info(f"Data_connectors for {database} processed successfully")
+        # return JSONResponse(content={"status": "success", "message": "Data saved successfully"}, status_code=200)
+        response = {
+            "msg": "Data saved successfully",
+            "url": "/save_data_connectors",  # The URL you want to redirect to
+            "created": True
+        }
+        return JSONResponse(content=response)
+        
+    
     except mysql.connector.Error as err:
         logging.error(f"Database error: {err}")
-        raise HTTPException(status_code=500, detail="Internal Server Error")
-    #     #databasetype_json=json.dumps(database)
-    #     query = "INSERT INTO data_connectors(company_id,databasetype, serverip, port, database_name,username,password,dbtablename) VALUES (%s,%s, %s, %s, %s,%s,%s,%s)"
-    #     values = (company_id, ",".join(database), server, port, ",".join(databaseName),username,password, ",".join(selectedTables))
-    #     logging.info(f"Executing query: {query} with values: {values}")
-    #     cursor.execute(query, values)
-    #     cnx.commit()
-    #     logging.info(f"Query executed successfully, {cursor.rowcount} row(s) affected")
-    #     cursor.close()
-    #     cnx.close()
-    #     logging.info(f"Data_connectors for {database} inserted successfully")
-    #     return RedirectResponse(url="/data_connectors", status_code=302)
-    # except mysql.connector.Error as err:
-    #     logging.error(f"Database error: {err}")
-    #     raise HTTPException(status_code=500, detail="Internal Server Error")
+        return JSONResponse(content={"status": "error", "message": "Internal Server Error"}, status_code=500)
+    except Exception as e:
+        logging.error(f"Unexpected error: {e}")
+        return JSONResponse(content={"status": "error", "message": "Unexpected Server Error"}, status_code=500)
     
 @app.get("/api/check_data_connectors")
 async def get_data_connectors(company_id: str = Query(...), company_name: str = Query(...)):
@@ -761,7 +841,7 @@ async def get_api_connectors(company_id: str = Query(...), company_name: str = Q
         cnx =get_db_connection()
         cursor = cnx.cursor()
         query = """
-            SELECT ac.company_id, ac.api_name, ac.api_endpoint,ac.auth_token,ac.input_param, ac.output_json, ac.description
+            SELECT ac.id, ac.company_id, ac.api_name, ac.api_endpoint,ac.auth_token,ac.input_param, ac.output_json, ac.description
             FROM u852023448_redmindgpt.api_connectors ac
             JOIN u852023448_redmindgpt.company_detail cd ON ac.company_id = cd.company_id 
             WHERE ac.company_id = %s and cd.company_name=%s
@@ -779,9 +859,10 @@ async def get_api_connectors(company_id: str = Query(...), company_name: str = Q
         companies=[]
         for row in result:
             companies.append({
-             "company_id": row[0],
-                "APIName":row[1],
-                "APIEndpoint": row[2]
+                "row_id":row[0],
+             "company_id": row[1],
+                "APIName":row[2],
+                "APIEndpoint": row[3]
                 # "Auth_Bearer": result[3],
                # "Inputjson": result[4],
                 #"OutputJson": result[5],
@@ -796,7 +877,197 @@ async def get_api_connectors(company_id: str = Query(...), company_name: str = Q
     except mysql.connector.Error as err:
         logging.error(f"Database error: {err}")
         raise HTTPException(status_code=500, detail="Internal Server Error")   
+ #to view the table details in modal
+@app.get("/api/viewapiconnectors/{company_id}")
+async def get_company_details(company_id: int):
 
+    company = await get_api_from_db(company_id)
+    if not company:
+        raise HTTPException(status_code=404, detail="Company not found")
+    return company  
+      
+async def get_api_from_db(company_id: int):
+    try:
+        # Establish a connection to the database
+        cnx = get_db_connection()
+        if cnx is None:
+            raise HTTPException(status_code=500, detail="Failed to connect to the database")
+        cursor = cnx.cursor(dictionary=True)
+        query = "SELECT * FROM api_connectors WHERE id = %s"
+        cursor.execute(query, (company_id,))
+        company = cursor.fetchone()
+        cursor.close()
+        cnx.close()
+        if company:
+           logging.info(f"api details:{company}")
+           return company
+        else:
+            raise HTTPException(status_code=404, detail="Company not found or file not found for the company")
+    except mysql.connector.Error as err:
+        logging.error(f"Error fetching company: {err}")
+        raise HTTPException(status_code=500, detail="Failed to fetch company")
+    #to edit the api details in modal form
+@app.put("/api/editapiconnectors/{companyId}")
+async def update_company_details(
+                             request: Request,
+                             companyId: int,
+                             company_id:str=Form(...),
+                             APIName:str=Form(...),
+                             APIEndpoint:str=Form(...),
+                             Auth_Bearer:str=Form(...),
+                             Inputjson:str=Form(...),
+                             OutputJson:str=Form(...),
+                             Description:str=Form(...)):
+    logging.info(f"Received form submission for database_connectors")
+    logging.info(f"Received request for company data with ID inside edit/update knowledgebase: {companyId}")
+     
+   # Prepare the company data dictionary
+    company_data = {
+        'kid': companyId,
+        'company_id': company_id,
+        'api_name': APIName,
+        'api_endpoint': APIEndpoint,
+        'auth_token': Auth_Bearer,
+        'input_param': Inputjson,
+        'output_json': OutputJson,
+        'description': Description
+    }
+
+    # Update the knowledge base in the database
+    updated_company = await update_api_in_db(companyId, company_data)
+    if not updated_company:
+        raise HTTPException(status_code=500, detail="Failed to update company")
+    return updated_company
+
+async def update_api_in_db(id: int, company_data: dict):
+    try:
+        logging.info(f"Updating api for ID: {id}")
+        cnx = get_db_connection()
+        if cnx is None:
+            raise HTTPException(status_code=500, detail="Failed to connect to the database")
+
+        cursor = cnx.cursor()
+        update_query = """
+        UPDATE redmind_gpt.api_connectors ac
+        SET ac.company_id = %s, ac.api_name = %s, ac.api_endpoint= %s, 
+            ac.auth_token= %s, ac.input_param= %s,ac.output_json = %s, ac.description= %s
+        WHERE ac.id = %s;
+        """
+        logging.info(f"Executing update query: {update_query}")
+        logging.info(f"With values: {(
+            company_data['company_id'],
+            company_data['api_name'],
+            company_data['api_endpoint'],
+            company_data['auth_token'],
+            company_data['input_param'],
+            company_data['output_json'],
+            company_data['description'],
+            id
+        )}")
+
+        cursor.execute(update_query, (
+           company_data['company_id'],
+            company_data['api_name'],
+            company_data['api_endpoint'],
+            company_data['auth_token'],
+            company_data['input_param'],
+            company_data['output_json'],
+            company_data['description'],
+            id
+        ))
+
+        cnx.commit()
+        success = cursor.rowcount > 0
+        cursor.close()
+        cnx.close()
+        
+        if not success:
+            logging.info("No rows updated")
+            return None
+        logging.info("Update successful")
+        return company_data
+    except mysql.connector.Error as err:
+        logging.error(f"Database error: {err}")
+        raise HTTPException(status_code=500, detail="Failed to update company")
+    except Exception as e:
+        logging.error(f"Unexpected error: {e}")
+        raise HTTPException(status_code=500, detail="Unexpected error occurred")
+ 
+#on update of modal form the data table is refreshed to dispalyupdated value in datatable   
+@app.get("/api/api_updatetable")
+async def get_document(company_id: str = Query(...)):
+    print(f"Received companyId and name for api datatable update: {company_id},{company_id}")  # Log rec
+    #async def get_data_connectors(company_id: str, company_name: str):
+    logging.info(f"Received request for company_id  and company_id: {company_id},{company_id}")
+    try:
+        cnx = get_db_connection()
+        cursor = cnx.cursor()
+        query="""  SELECT ac.id,ac.company_id, ac.api_name, ac.api_endpoint,ac.auth_token,ac.input_param, ac.output_json, ac.description
+            FROM redmind_gpt.api_connectors ac
+            JOIN redmind_gpt.company_detail cd ON ac.company_id = cd.company_id 
+            WHERE ac.company_id = %s 
+        """
+        logging.info(f"Executing query: {query} with company_id: {company_id}")
+        values= (company_id,)
+       # logging.info(f"Query parameters: {params}")
+        print(f"Query parameters: {values}")
+
+        cursor.execute(query, values)  # Pa
+        result = cursor.fetchall()
+        logging.info(f"Query result for update table: {result}")
+        cursor.close
+        cnx.close()
+        companies=[]
+        for row in result:
+            
+            companies.append({
+               "row_id":row[0],
+             "company_id": row[1],
+                "api_name":row[2],
+                "api_endpoint": row[3],
+            #      "Auth_Bearer": row[4],
+            #    "Inputjson": row[5],
+            #     "OutputJson": row[6],
+            #     "description": row[7]
+                
+            })
+        if companies:
+            return companies
+        else:
+            logging.warning(f"No document found for company_id: {company_id}")
+            raise HTTPException(status_code=404, detail="Data document not found")
+    except mysql.connector.Error as err:
+        logging.error(f"Database error: {err}")
+        raise HTTPException(status_code=500, detail="Internal Server Error")
+    
+    #to delete api details from db
+@app.delete("/api/deleteapi/{company_id}")
+async def delete_company(company_id: int):
+    deletion_success = delete_api_from_db(company_id)
+    if not deletion_success:
+        raise HTTPException(status_code=404, detail="Company not found or failed to delete")
+    return {"message": "Company deleted successfully"}
+ 
+def delete_api_from_db(company_id: int) -> bool:
+    print(f"Received api for company_id: {company_id}")  # Debug statement
+    logging.info(f"Received request for api for company id: {company_id}")
+    try:
+        # Establish a connection to the database
+        cnx = get_db_connection()
+        if cnx is None:
+            raise HTTPException(status_code=500, detail="Failed to connect to the database")
+        cursor = cnx.cursor()
+        delete_query = "DELETE FROM api_connectors WHERE id = %s"
+        cursor.execute(delete_query, (company_id,))
+        cnx.commit()
+        success = cursor.rowcount > 0
+        cursor.close()
+        cnx.close()
+        return success
+    except mysql.connector.Error as err:
+        logging.error(f"Error deleting company: {err}")
+        raise HTTPException(status_code=500, detail="Failed to delete company")
+    
 
 @app.get("/prompt_template")
 async def prompt_template(request: Request):
@@ -837,7 +1108,7 @@ async def get_prompt_connectors(company_id: str = Query(...), company_name: str
         cnx =get_db_connection()
         cursor = cnx.cursor()
         query = """
-            SELECT pt.company_id,pt.scenario,pt.prompts,pt.comments
+            SELECT pt.id,pt.company_id,pt.scenario,pt.prompts,pt.comments
             FROM u852023448_redmindgpt.prompt_templates pt
             JOIN u852023448_redmindgpt.company_detail cd ON pt.company_id = cd.company_id 
             WHERE pt.company_id = %s and cd.company_name=%s
@@ -855,9 +1126,10 @@ async def get_prompt_connectors(company_id: str = Query(...), company_name: str
         companies=[]
         for row in result:
             companies.append({
-             "company_id": row[0],
-                "scenario":row[1],
-                "prompt": row[2]
+                "row_id":row[0],
+             "company_id": row[1],
+                "scenario":row[2],
+                "prompt": row[3]
                 # "Auth_Bearer": result[3],
                # "Inputjson": result[4],
                 #"OutputJson": result[5],
@@ -872,7 +1144,179 @@ async def get_prompt_connectors(company_id: str = Query(...), company_name: str
     except mysql.connector.Error as err:
         logging.error(f"Database error: {err}")
         raise HTTPException(status_code=500, detail="Internal Server Error")   
+def delete_prompt_template_from_db(row_id: int) -> bool:
+    logging.info(f"Received request for prompt_template company id: {row_id}")
+    logging.info(f"Received request for prompt_template row id: {row_id}")
+
+    try:
+        # Establish a connection to the database
+        cnx = get_db_connection()
+        if cnx is None:
+            raise HTTPException(status_code=500, detail="Failed to connect to the database")
+        cursor = cnx.cursor()
+        delete_query = "DELETE FROM prompt_templates WHERE id = %s"
+
+        logging.info(f"sql delete query for prompt template ===> {delete_query}")
+
+        cursor.execute(delete_query, (row_id,))
+        cnx.commit()
+        success = cursor.rowcount > 0
+        logging.info (f"deleted succesfully ! ===> {success}")
+
+        cursor.close()
+        cnx.close()
+        return success
+    except mysql.connector.Error as err:
+        print('python')
+        logging.error(f"Error deleting company: {err}")
+        raise HTTPException(status_code=500, detail="Failed to delete company")
+
+@app.delete("/api/prompt_template_for_del/{row_id}")
+async def delete_company(row_id: int):
+    deletion_success = delete_prompt_template_from_db(row_id)
+    logging.info(f"company row_id +++> {row_id}")
+    
+    if not deletion_success:
+        raise HTTPException(status_code=404, detail="Company not found or failed to delete")
+    return {"message": "Company deleted successfully"}
+
+# promt_template view function ! ............
+
+#to get data for view  in promt_templae by id
+@app.get("/api/getpromttemplate/{company_id}")
+async def get_promt_company_details(company_id: int):
+    company = await get_promt_from_db(company_id)
+    if not company:
+        raise HTTPException(status_code=404, detail="Company not found")
+    return company  
+      
+async def get_promt_from_db(company_id: int):
+    try:
+        # Establish a connection to the database
+        cnx = get_db_connection()
+        if cnx is None:
+            raise HTTPException(status_code=500, detail="Failed to connect to the database")
+        cursor = cnx.cursor(dictionary=True)
+        query = "SELECT * FROM prompt_templates  WHERE id = %s"
+        logging.info(f"row_id  in db addresss ========>  {company_id}")
+
+        cursor.execute(query, (company_id,))
+        company = cursor.fetchone()
+        cursor.close()
+        cnx.close()
+        if company:
+           logging.info(f"row_id  in db addresss ========>  {company}")
+           return company
+        
+        else:
+            raise HTTPException(status_code=404, detail="Company not found or file not found for the company")
+    except mysql.connector.Error as err:
+        logging.error(f"Error fetching company: {err}")
+        raise HTTPException(status_code=500, detail="Failed to fetch company")
+    
+# Function to update company details
+@app.put("/api/putprompttemplates/{kid}")
+async def update_company_details(
+    kid: int,
+    scenario: str = Form(...),
+    prompt: str = Form(...),
+    comments: str = Form(...)
+):
+    logging.info(f"Received request for company data with ID: {kid}")
+    company_data = {
+        'scenario': scenario,
+        'prompts': prompt,
+        'comments': comments,
+    }
+    updated_company = await update_prompt_in_db(kid, company_data)
+    if not updated_company:
+        raise HTTPException(status_code=500, detail="Failed to update company")
+    return updated_company
 
+# Database query function to update company data
+async def update_prompt_in_db(kid: int, company_data: dict):
+    try:
+        logging.info(f"Updating prompt for ID: {kid}")
+        cnx = get_db_connection()
+        if cnx is None:
+            raise HTTPException(status_code=500, detail="Failed to connect to the database")
+        
+        cursor = cnx.cursor()
+        update_query = """
+        UPDATE redmind_gpt.prompt_templates pt
+        SET pt.scenario=%s, pt.prompts=%s, pt.comments=%s
+        WHERE pt.id = %s;
+        """
+        logging.info(f"row_id in prompt db address ========> {kid}")
+        logging.info(f"SQL update query for company ===> {update_query}")
+        
+        cursor.execute(update_query, (
+            company_data['scenario'],
+            company_data['prompts'],
+            company_data['comments'],
+            kid
+        ))
+
+        cnx.commit()
+        success = cursor.rowcount > 0
+        cursor.close()
+        cnx.close()
+
+        if not success:
+            return None
+        return company_data
+
+    except mysql.connector.Error as err:
+        logging.error(f"Error updating company: {err}")
+        raise HTTPException(status_code=500, detail="Failed to update company")
+    
+# to refresh prompt data table     
+@app.get("/api/prompt_update")
+async def get_document(company_id: str = Query(...)):
+    print(f"Received companyId and name: {company_id},{company_id}")  # Log rec
+    #async def get_data_connectors(company_id: str, company_name: str):
+    logging.info(f"Received request for company_id  and company_id: {company_id},{company_id}")
+    try:
+        cnx = get_db_connection()
+        cursor = cnx.cursor()
+        query = """
+             SELECT pt.id,pt.company_id,pt.scenario,pt.prompts,pt.comments
+            FROM redmind_gpt.prompt_templates pt
+            JOIN redmind_gpt.company_detail cd ON pt.company_id = cd.company_id 
+            WHERE pt.company_id = %s 
+        """
+        logging.info(f"Executing query: {query} with company_id: {company_id}")
+        values= (company_id,)
+       # logging.info(f"Query parameters: {params}")
+        print(f"Query parameters: {values}")
+
+        cursor.execute(query, values)  # Pa
+        result = cursor.fetchall()
+        logging.info(f"Query result: {result}")
+        cursor.close
+        cnx.close()
+        companies=[]
+        for row in result:
+            companies.append({
+                'id':row[0],
+                "company_id": row[1],
+                "scenario":row[2],
+                "prompts": row[3]
+                # "Auth_Bearer": result[3],
+               # "Inputjson": result[4],
+                #"OutputJson": result[5],
+                #"description": result[6]
+                
+            })
+        if companies:
+            logging.info(f"the  primary key id is {companies}")
+            return companies
+        else:
+            logging.warning(f"No document found for company_id: {company_id}")
+            raise HTTPException(status_code=404, detail="Data document not found")
+    except mysql.connector.Error as err:
+        logging.error(f"Database error: {err}")
+        raise HTTPException(status_code=500, detail="Internal Server Error")
 
 @app.get("/chatbot")
 async def chatbot(request: Request):