ConstantCoder commited on
Commit
72cd03d
·
verified ·
1 Parent(s): 68e4cd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -22
app.py CHANGED
@@ -1,27 +1,31 @@
1
- #!/usr/bin/env python
2
- # coding: utf-8
3
-
4
- # In[1]:
5
-
6
-
7
- from flask import Flask, send_from_directory
8
 
9
  app = Flask(__name__)
10
 
11
- @app.route('/')
12
- def index():
13
- return send_from_directory('.', 'index.html')
14
-
15
- @app.route('/<path:path>')
16
- def static_files(path):
17
- return send_from_directory('.', path)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
18
 
19
  if __name__ == '__main__':
20
- app.run(debug=True)
21
-
22
-
23
- # In[ ]:
24
-
25
-
26
-
27
-
 
1
+ from flask import Flask, request, jsonify
2
+ from sklearn.tree import DecisionTreeClassifier
3
+ import numpy as np
 
 
 
 
4
 
5
  app = Flask(__name__)
6
 
7
+ # Basit örnek veriler (Bu veriler yerel olarak tutulabilir)
8
+ # Örnek veri formatı: [phishing_response, ransomware_response, ddos_response, malware_response]
9
+ example_data = [
10
+ [0, 0, 1, 1],
11
+ [1, 1, 0, 0],
12
+ [1, 0, 1, 0],
13
+ [0, 1, 0, 1]
14
+ ]
15
+
16
+ # Örnek sonuçlar: 1 = güvenli, 0 = güvensiz
17
+ example_labels = [1, 0, 1, 0]
18
+
19
+ # Model eğitimi
20
+ model = DecisionTreeClassifier()
21
+ model.fit(example_data, example_labels)
22
+
23
+ @app.route('/analyze', methods=['POST'])
24
+ def analyze():
25
+ data = request.json
26
+ response = [data['phishing'], data['ransomware'], data['ddos'], data['malware']]
27
+ prediction = model.predict([response])
28
+ return jsonify({'prediction': int(prediction[0])})
29
 
30
  if __name__ == '__main__':
31
+ app.run(debug=True)