Fixed custom color issue (converted HEX to RGB for Matplotlib)

app.py

@@ -1,17 +1,23 @@
 import numpy as np
 import matplotlib.pyplot as plt
 import gradio as gr
-import matplotlib.colors as mcolors
 import os  
 
+# Function to convert HEX color to RGB tuple
+def hex_to_rgb(hex_color):
+    """Convert HEX color (#RRGGBB) to Matplotlib RGB tuple (R, G, B)."""
+    hex_color = hex_color.lstrip("#")  # Remove #
+    rgb = tuple(int(hex_color[i:i+2], 16)/255 for i in (0, 2, 4))  # Convert to (R, G, B)
+    return rgb  # Matplotlib expects RGB (not HEX)
+
 def plot_function(func_str, x_min, x_max, resolution, color, linestyle, grid):
     try:
-        # Ensure color is in proper HEX format
+        # Ensure color is valid; fallback to black if empty
         if not color or not color.startswith("#") or len(color) != 7:
             color = "#000000"  # Default to black
 
-        # Convert to Matplotlib-compatible RGBA format
-        color = mcolors.to_rgba(color)  
+        # Convert HEX to RGB
+        color_rgb = hex_to_rgb(color)
 
         x_values = np.linspace(x_min, x_max, resolution)
         functions = func_str.split(",")
@@ -23,8 +29,8 @@ def plot_function(func_str, x_min, x_max, resolution, color, linestyle, grid):
             func = lambda x: eval(func_text, {"x": x, "np": np})
             y_values = func(x_values)
 
-            # Use RGBA color format (Matplotlib-compatible)
-            plt.plot(x_values, y_values, label=f"f(x) = {func_text}", color=color, linestyle=linestyle)
+            # Use RGB color for Matplotlib
+            plt.plot(x_values, y_values, label=f"f(x) = {func_text}", color=color_rgb, linestyle=linestyle)
 
         plt.xlabel("x")
         plt.ylabel("f(x)")