Update app.py

app.py

@@ -111,34 +111,42 @@ def predictModel(mlfpath, userID):
     res = model.predict(userID)
     return res
 
-if __name__ == "__main__":
-    st.title("Conversion Prediction App")
-    st.write("Welcome to the Conversion Prediction App. This app uses a Markov chain based classifier to predict whether a customer will convert or not based on their visit history.")
-
-    op = st.sidebar.selectbox("Select Operation", ["Generate Visit History", "Train Model", "Predict"])
-
-    if op == "Generate Visit History":
-        st.write("Enter the parameters to generate the visit history:")
-        numUsers = st.number_input("Number of users", min_value=1, max_value=1000, value=100, step=1)
-        convRate = st.number_input("Conversion Rate (in percentage)", min_value=0, max_value=100, value=10, step=1)
-        label = st.checkbox("Add Labels")
-        st.write("Click the button below to generate the visit history")
-        if st.button("Generate"):
-            genVisitHistory(numUsers, convRate, label)
+if op == "Predict":
+    st.write("Enter the parameters to make a prediction:")
+    userID = st.text_input("User ID")
+    st.write("Click the button below to make a prediction")
+    if st.button("Predict"):
+        prediction = predictModel(mlfpath, userID)
+        st.write("Prediction:", prediction)
+
+# if __name__ == "__main__":
+#     st.title("Conversion Prediction App")
+#     st.write("Welcome to the Conversion Prediction App. This app uses a Markov chain based classifier to predict whether a customer will convert or not based on their visit history.")
+
+#     op = st.sidebar.selectbox("Select Operation", ["Generate Visit History", "Train Model", "Predict"])
+
+#     if op == "Generate Visit History":
+#         st.write("Enter the parameters to generate the visit history:")
+#         numUsers = st.number_input("Number of users", min_value=1, max_value=1000, value=100, step=1)
+#         convRate = st.number_input("Conversion Rate (in percentage)", min_value=0, max_value=100, value=10, step=1)
+#         label = st.checkbox("Add Labels")
+#         st.write("Click the button below to generate the visit history")
+#         if st.button("Generate"):
+#             genVisitHistory(numUsers, convRate, label)
             
-    elif op == "Train Model":
-        st.write("Train the model using the following parameters:")
-        mlfpath = st.text_input("MLF Path")
-        if st.button("Train"):
-            trainModel(mlfpath)
-
-    elif op == "Predict":
-        st.write("Predict using the trained model:")
-        mlfpath = st.text_input("MLF Path")
-        userID = st.text_input("User ID")
-        if st.button("Predict"):
-            result = predictModel(mlfpath, userID)
-            st.write("Prediction Result: ", result)
+#     elif op == "Train Model":
+#         st.write("Train the model using the following parameters:")
+#         mlfpath = st.text_input("MLF Path")
+#         if st.button("Train"):
+#             trainModel(mlfpath)
+
+#     elif op == "Predict":
+#         st.write("Predict using the trained model:")
+#         mlfpath = st.text_input("MLF Path")
+#         userID = st.text_input("User ID")
+#         if st.button("Predict"):
+#             result = predictModel(mlfpath, userID)
+#             st.write("Prediction Result: ", result)
 
 # def main():
 #     st.title("Markov Chain Classifier")