Update app.py

app.py

@@ -87,8 +87,6 @@ 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"""
@@ -187,7 +185,21 @@ def ask_gpt4o_for_visualization(query, df, llm, retries=2):
                 if all(k in suggestions for k in ["chart_type", "x_axis", "y_axis"]):
                     return [suggestions]
                 else:
-                    st.warning("⚠️ GPT-4o's suggestion is
+                    st.warning("⚠️ GPT-4o's suggestion is incomplete.")
+                    return None
+
+        except json.JSONDecodeError:
+            st.warning(f"⚠️ Attempt {attempt + 1}: GPT-4o returned invalid JSON.")
+        except Exception as e:
+            st.error(f"⚠️ Error during GPT-4o call: {e}")
+        
+        # Retry if necessary
+        if attempt < retries:
+            st.info("🔄 Retrying visualization suggestion...")
+
+    st.error("❌ Failed to generate a valid visualization after multiple attempts.")
+    return None
+
 
 
 def add_stats_to_figure(fig, df, y_axis, chart_type):