Update app.py

app.py

@@ -87,28 +87,28 @@ if st.session_state.df is not None and st.session_state.show_preview:
     st.dataframe(st.session_state.df.head())
 
 
-"""def ask_gpt4o_for_visualization(query, df, llm):
-    columns = ', '.join(df.columns)
-    prompt = f"""
-    Analyze the query and suggest one or more relevant visualizations.
-    Query: "{query}"
-    Available Columns: {columns}
-    Respond in this JSON format (as a list if multiple suggestions):
-    [
-      {{
-        "chart_type": "bar/box/line/scatter",
-        "x_axis": "column_name",
-        "y_axis": "column_name",
-        "group_by": "optional_column_name"
-      }}
-    ]
-    """
-    response = llm.generate(prompt)
-    try:
-        return json.loads(response)
-    except json.JSONDecodeError:
-        st.error("⚠️ GPT-4o failed to generate a valid suggestion.")
-        return None"""
+#def ask_gpt4o_for_visualization(query, df, llm):
+#    columns = ', '.join(df.columns)
+#    prompt = f"""
+#    Analyze the query and suggest one or more relevant visualizations.
+#    Query: "{query}"
+#    Available Columns: {columns}
+#    Respond in this JSON format (as a list if multiple suggestions):
+#    [
+#      {{
+#       "chart_type": "bar/box/line/scatter",
+#        "x_axis": "column_name",
+#        "y_axis": "column_name",
+#        "group_by": "optional_column_name"
+#      }}
+#    ]
+#    """
+#    response = llm.generate(prompt)
+#    try:
+#        return json.loads(response)
+#    except json.JSONDecodeError:
+#        st.error("⚠️ GPT-4o failed to generate a valid suggestion.")
+#        return None
 
 def ask_gpt4o_for_visualization(query, df, llm, retries=2):
     import json