Update main.py

main.py

@@ -1,5 +1,8 @@
 from flask import Flask, request, jsonify, render_template
 from flask_cors import CORS
+import numpy as np
+import pandas as pd
+from sklearn.linear_model import LogisticRegression
 
 app = Flask(__name__)
 app.static_folder = 'static'
@@ -10,8 +13,32 @@ app.secret_key = "flask-nielit-2023"
 CORS(app)
 
 @app.route('/')
-def index():
-    return render_template('index.html')
+def iris():
+  return render_template("index.html")
+
+@app.route('/irisf', methods=["POST"])
+def page():
+  swidth=eval(request.form.get("swidth"))
+  sheight=eval(request.form.get("sheight"))
+  pwidth=eval(request.form.get("pwidth"))
+  pheight=eval(request.form.get("pheight"))
+  
+  url="https://raw.githubusercontent.com/lovnishverma/datasets/main/iris.csv"
+  
+  data=pd.read_csv(url, header=None)
+  flower=data.values
+  
+  #Split
+  x=flower[:,:4]
+  y=flower[:,-1]
+  
+  model=LogisticRegression()
+  model.fit(x,y)
+  
+  arr=model.predict([[swidth,sheight,pwidth,pheight]])
+
+  return render_template("index.html", data=str(arr[0]))
+
 
 if __name__ == '__main__':
     app.run()