-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathapp.py
69 lines (55 loc) · 1.99 KB
/
app.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
# from flask import Flask
from flask import Flask, request, jsonify, redirect, render_template, url_for
import os
import firebase_admin
from firebase_admin import credentials, firestore
# from model.gk import *
from model.Processing import predicted_image
app = Flask(__name__)
cred = credentials.Certificate("serviceAccountKey.json")
default_app = firebase_admin.initialize_app(cred)
db = firestore.client()
@app.route("/", methods=['GET', 'POST'])
def predict():
# if request.method == 'POST':
# user_file = request.files['file']
# temp = request.files['file']
# # check if the post request has the file part
# if 'file' not in request.files:
# return "No file found"
#
#
# elif user_file.filename == "":
# return "file name not found …"
# else:
# path = os.path.join(os.getcwd() + '\\modules\\static\\' + user_file.filename)
# user_file.save(path)
# a = " , "
# return jsonify({'message': 'Hello, World!'})
# else:
# return jsonify({'message': 'Hello, World!'})
if request.method == 'POST':
# check if the post request has the file part
if 'file' not in request.files:
return "someting went wrong 1"
user_file = request.files['file']
temp = request.files['file']
if user_file.filename == '':
return "file name not found ..."
else:
path = os.path.join(os.getcwd()+ user_file.filename)
user_file.save(path)
a = " , "
doc = db.collection(u'Real').document(u'r1')
string1= user_file.filename[0:5]
doc.update({
string1:(a.join(predicted_image(path))),
})
else:
doc = db.collection(u'Sample').document(u's1')
doc.update({
u's3': "abc",
})
return "Ash"
if __name__ == "__main__":
app.run()