Update app.py

app.py

@@ -89,7 +89,12 @@ def calculate_kpis(df):
 def get_pandas_profile(df):
     profile = ProfileReport(df, title="Profiling Report")
     json_profile = profile.to_json()
-    return json_profile
+    dict_p = json.loads(json_profile)
+    keys_to_keep = ['analysis', 'table', 'correlations', 'alerts', 'sample']
+
+# Assuming your dictionary is named 'my_dict'
+    filtered_dict = {key: dict_p[key] for key in keys_to_keep}
+    return filtered_dict
     
 def generateResponse(dataFrame,prompt):
         llm = GoogleGemini(api_key=gemini_api_key)