Rmv Bug in Describe

app.py

@@ -127,11 +127,15 @@ def get_table_schema(table):
     return full_path
 
 def describe(df):
-    numerical_info = df.select_dtypes(include=['number']).describe().T.reset_index()
-    numerical_info.rename(columns={'index': 'column'}, inplace=True)
     
-    categorical_info = df.select_dtypes(include=['object']).describe().T.reset_index()
-    categorical_info.rename(columns={'index': 'column'}, inplace=True)
+    numerical_info = pd.DataFrame()
+    categorical_info = pd.DataFrame()
+    if len(df.select_dtypes(include=['number']).columns) >= 1:
+        numerical_info = df.select_dtypes(include=['number']).describe().T.reset_index()
+        numerical_info.rename(columns={'index': 'column'}, inplace=True)
+    if len(df.select_dtypes(include=['object']).columns) >= 1:
+        categorical_info = df.select_dtypes(include=['object']).describe().T.reset_index()
+        categorical_info.rename(columns={'index': 'column'}, inplace=True)
 
     return numerical_info, categorical_info