Update mylab/fixed_font_issue.py

mylab/fixed_font_issue.py

@@ -110,6 +110,48 @@ def ask_gpt4o_for_visualization(query, df, llm):
         st.error("⚠️ GPT-4o failed to generate a valid suggestion.")
         return None
 
+# Dynamically generate Plotly visualizations based on GPT-4o suggestions
+def generate_visualization(suggestion, df):
+    chart_type = suggestion.get("chart_type", "bar").lower()
+    x_axis = suggestion.get("x_axis")
+    y_axis = suggestion.get("y_axis")
+    group_by = suggestion.get("group_by")
+
+    # Ensure required inputs are available
+    if not x_axis or not y_axis:
+        st.warning("⚠️ GPT-4o did not provide enough information for the visualization.")
+        return None
+
+    # Dynamically select the Plotly function
+    plotly_function = getattr(px, chart_type, None)
+
+    # Handle unsupported chart types gracefully
+    if not plotly_function:
+        st.warning(f"⚠️ Unsupported chart type '{chart_type}' suggested by GPT-4o.")
+        return None
+
+    # Prepare dynamic parameters for Plotly function
+    plot_args = {
+        "data_frame": df,
+        "x": x_axis,
+        "y": y_axis,
+    }
+    if group_by:
+        plot_args["color"] = group_by
+
+    try:
+        # Generate the dynamic visualization
+        fig = plotly_function(**plot_args)
+        fig.update_layout(
+            title=f"{chart_type.title()} Plot of {y_axis.replace('_', ' ').title()} by {x_axis.replace('_', ' ').title()}",
+            xaxis_title=x_axis.replace('_', ' ').title(),
+            yaxis_title=y_axis.replace('_', ' ').title(),
+        )
+        return fig
+
+    except Exception as e:
+        st.error(f"⚠️ Failed to generate visualization: {e}")
+        return None
 
 # Function to create TXT file
 def create_text_report_with_viz_temp(report, conclusion, visualizations):