Update app.py

app.py

@@ -1,7 +1,7 @@
 # 必要なライブラリをインポート
 import streamlit as st
 import random
-from sympy import symbols, expand, Eq
+from sympy import symbols, factor, Eq, solve
 
 # Streamlitアプリのタイトルを設定
 st.title("数学問題アプリ")
@@ -12,26 +12,29 @@ st.sidebar.markdown("$2x$と書きたい場合は「$2*x$」と、記述する
 st.sidebar.markdown("割り算は`/`を使うように。掛け算は`*`を使うように。")
 
 # メインのコンテンツを作成
-selected_function = st.sidebar.selectbox("問題形式を選択", ["因数分解","方程式"])
+selected_function = st.sidebar.selectbox("問題形式を選択", ["因数分解", "方程式"])
 x = symbols("x")
+
+
 def factor_expression(expression):
     x = symbols('x')
     factored_expression = factor(expression)
 
     st.success(f"因数分解結果: {factored_expression}")
     st.latex(factored_expression)
+
+
 def generate_equation():
-    
     x = symbols('x')
     generated_problems = []
     # 例: x**2 + 5*x + 6
-    wh=random.choice(["1","2"])
-    if wh==1:
+    wh = random.choice([1, 2])
+    if wh == 1:
         a = random.randint(min_value, max_value)
         b = random.randint(min_value, max_value)
         init_problem = f"(x + {a})*(x + {b})"
         AExpression = expand(init_problem)
-    if wh==2:
+    if wh == 2:
         a = random.randint(min_value, max_value)
         b = random.randint(min_value, max_value)
         init_problem = f"{a}*x + {b}"
@@ -45,6 +48,7 @@ def generate_equation():
 
     return equation
 
+
 # 因数分解
 if selected_function == "因数分解":
     min_value = st.number_input("乱数の最小値を入力してください:", value=-10)
@@ -63,6 +67,7 @@ if selected_function == "因数分解":
         if expression:
             st.success(f"問題: {expression}")
             st.latex(expression)
+
 if selected_function == "方程式":
     min_value = st.number_input("乱数の最小値を入力してください:", value=-10)
     max_value = st.number_input("乱数の最大値を入力してください:", value=10)
@@ -72,7 +77,7 @@ if selected_function == "方程式":
     if generate_button:
         # 方程式の生成
         equation = generate_equation()
-    
+
         # 生成した方程式の表示
         st.success(f"生成した方程式: {equation}")
         st.latex(equation)