Spaces:
Sleeping
Sleeping
File size: 1,234 Bytes
6ec8cbf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
import pymysql
import json
import pandas as pd
import re
import streamlit as st
# Database connection
def initialize_database():
# Database Connection
db_params = {"host": st.secrets["host"],
"user": st.secrets["user"],
"password": st.secrets["password"],
"port": st.secrets["port"],
"database": st.secrets["database"]
}
db = pymysql.connect(**db_params)
return db
def execute_query(query):
db = initialize_database()
cursor = db.cursor()
try:
cursor.execute(query)
description = cursor.description
result = cursor.fetchall() # Fetch all rows from the result set
db.commit()
return description, result
except Exception as e:
print("Error executing query:", e)
db.rollback()
return None # Return None if an error occurs
finally:
db.close()
def _get_details_mantra_json(self, query):
description, data = execute_query(query)
df = pd.DataFrame(data)
df.columns = [x[0] for x in description]
mantra_json = df['mantra_json'].values[0]
cleaned_data = re.sub('<[^<]+?>', '', mantra_json)
return json.loads(cleaned_data)
|