Update mylab/fixed_font_issue.py

mylab/fixed_font_issue.py

@@ -86,6 +86,31 @@ if st.session_state.df is not None and st.session_state.show_preview:
     st.subheader("📂 Dataset Preview")
     st.dataframe(st.session_state.df.head())
 
+# Ask GPT-4o for Visualization Suggestions
+def ask_gpt4o_for_visualization(query, df, llm):
+    columns = ', '.join(df.columns)
+    prompt = f"""
+    Analyze the query and suggest the best visualization.
+
+    Query: "{query}"
+    Available Columns: {columns}
+
+    Respond in this JSON format:
+    {{
+      "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
+
+
 # Function to create TXT file
 def create_text_report_with_viz_temp(report, conclusion, visualizations):
     content = f"### Analysis Report\n\n{report}\n\n### Visualizations\n"