Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-from sympy import symbols, solve, Eq, Ge
+from sympy import symbols, solve, S, Interval
 import matplotlib.pyplot as plt
 import numpy as np
 
@@ -133,21 +133,39 @@ def inequality_solver():
         st.write(solution)
         
         # 不等式の解を一次元に描画
-        plot_inequality(solution)
+        plot_inequality(inequality, solution)
 
-def plot_inequality(solution):
-    x = np.linspace(-10, 10, 400)
-    y = np.zeros_like(x)
+def plot_inequality(inequality, solution):
+    x = symbols('x')
+    x_vals = np.linspace(-10, 10, 400)
+    y_vals = np.zeros_like(x_vals)
+
+    fig, ax = plt.subplots(figsize=(10, 2))
+    ax.plot(x_vals, y_vals, 'k-')
+    
+    if isinstance(solution, list) and len(solution) == 2:
+        lower_bound, upper_bound = solution
+        if inequality.func == Ge:
+            ax.fill_between(x_vals, y_vals - 0.1, y_vals + 0.1, where=(x_vals >= lower_bound), color='blue', alpha=0.5)
+        elif inequality.func == Le:
+            ax.fill_between(x_vals, y_vals - 0.1, y_vals + 0.1, where=(x_vals <= upper_bound), color='blue', alpha=0.5)
+    elif isinstance(solution, Interval):
+        start = solution.start
+        end = solution.end
+        if solution.left_open:
+            start += 1e-5
+        if solution.right_open:
+            end -= 1e-5
+        ax.fill_between(x_vals, y_vals - 0.1, y_vals + 0.1, where=(x_vals >= start) & (x_vals <= end), color='blue', alpha=0.5)
+    else:
+        ax.fill_between(x_vals, y_vals - 0.1, y_vals + 0.1, where=(x_vals >= solution), color='blue', alpha=0.5)
     
-    plt.figure(figsize=(10, 1))
-    plt.plot(x, y, 'k-', linewidth=1)
-    plt.fill_between(x, y - 0.1, y + 0.1, where=(x >= solution), color='blue', alpha=0.5)
-    plt.xlim(-10, 10)
-    plt.ylim(-1, 1)
-    plt.yticks([])
-    plt.title('不等式の解のプロット')
+    ax.set_xlim(-10, 10)
+    ax.set_ylim(-1, 1)
+    ax.set_yticks([])
+    ax.set_title('不等式の解のプロット')
     
-    st.pyplot(plt)
+    st.pyplot(fig)
 
 if __name__ == "__main__":
     main()