Update app.py

app.py

@@ -1,20 +1,28 @@
+import random
 import streamlit as st
-from calculator import calculate as calc
 
 
-def interface():
-  st.title("Simple Calculator")
+def calc(number1, number2, select_op, user_answer):
+    if user_answer == number1 + number2:
+        return 'Верно'
+    elif user_answer == number1 - number2:
+        return 'Верно'
+    elif user_answer == number1 * number2:
+        return 'Верно'
+    elif user_answer == number1 / number2:
+        return 'Верно'
+    else:
+        return 'Неверно'
 
-  operations = ["+", "-", "*", "/"]
 
-  select_op = st.selectbox("Select operation", operations)
+def main():
+    number1 = random.randint(1, 10)
+    number2 = random.randint(1, 10)
+    select_op = random.choice(['+', '-', '*', '/'])
+    st.write(f'{number1} {select_op} {number2} = ')
+    user_answer = st.number_input('Введите ответ')
+    st.write(calc(number1, number2, select_op, user_answer))
 
-  number1 = st.number_input("Enter first number")
 
-  number2 = st.number_input("Enter second number")
-
-  
-
-      answer = calc(number1, number2, select_op)  
-
-      st.write("Answer is", answer)
+if __name__ == '__main__':
+    main()