VideoAnnotationTool/views.py at master · VideoAnalysis/VideoAnnotationTool · GitHub

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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
from __init__ import app
from flask import render_template, request

import glob
import numpy as np

import json
import cv2
import pickle


# http://127.0.0.1:5000/videoanalysis/getmain?name=junaid

from flask import send_from_directory

baseWeb="/videoanalysis"
usersArr=["junaid","test_user1"] # list(userDict.keys())

@app.route(baseWeb+'/static/<path:path>')
def send_js(path):
    return send_from_directory('static', path)

@app.route(baseWeb+'/', methods=["GET", "POST"])
def index():
    if request.method == 'POST':
        print("post side is not implemented yet")
    else:
        print("get req: ",request.values)
        return render_template('comming_soon.html')

@app.route(baseWeb+'/getvaasui', methods=["GET", "POST"])
def getvaasui():
    if request.method == 'POST':
        print("post side is not implemented yet")
    else:
#        print("get req: ",request.values)
        return render_template('vaasui.html')

@app.route(baseWeb+'/getVideoStats', methods=["POST"])
def getVideoStats():
    print("getVideoStats req: ",request.values)
    response = {"feedback":True}
    return json.dumps(response)

@app.route(baseWeb+'/getmain', methods=["GET", "POST"])
def getmain():
    if request.method == 'POST':
        print("post side is not implemented yet")
    else:
        userGet=request.args.get("name")
        print("request args dataDict getmain: ",userGet)
        if(userGet in usersArr):
            return render_template('vidsum.html')
        else:
            return render_template('access_forbidden.html')
#            response = {"feedback":"User is not authorized for access.!!!"} access_forbidden
#            return json.dumps(response)

@app.route(baseWeb+'/getListOfVideos', methods=["POST"])
def getListOfVideos():
    dataDict=eval(list(request.values.keys())[0])
    print("request args dataDict: ",dataDict)
    userDict=pickle.load(open("video_dict_vidsum.p","rb"))
    videoNames=json.load(open("videoNames.json","r"))
    config=json.load(open("config.json","r"))
    if request.method == 'POST':
         response = {"video_dict":pickle.load(open("video_dict_vidsum.p","rb")),"usersArr":usersArr,"vidUserLs":pickle.load(open("videoAnnotate.p","rb")),"videoNames":videoNames,"impLevel":config["impLevel"],"subtitles":config["subtitles"]}
         print("video dictionary loaded...")
         return json.dumps(response)

@app.route(baseWeb+'/saveAnnotation', methods=["POST"])
def saveAnnotation():
    print("in the saving function...")
#    print("request args saveAnnotation: ",request.args)
    dataDict=eval(list(request.values.keys())[0])
    print("creatorName: ",dataDict["creatorName"])
    print("videoImportance: ",dataDict["videoImportance"])
    print("videoName: ",dataDict["videoName"])
    videoAnnotate=pickle.load(open("videoAnnotate.p","rb"))
    print(" request.method: ", request.method)
    if request.method == 'POST':
        if(dataDict["creatorName"] in videoAnnotate):
            videoAnnotate[dataDict["creatorName"]][dataDict["videoName"]]=dataDict["videoImportance"]
            videoAnnotate["vidUserLs"][dataDict["creatorName"]].append([dataDict["videoName"],dataDict["videoImportance"]])
            print("existing user data added...")
        else:
            videoAnnotate[dataDict["creatorName"]]={}
            videoAnnotate[dataDict["creatorName"]][dataDict["videoName"]]=dataDict["videoImportance"]
            videoAnnotate["vidUserLs"][dataDict["creatorName"]]=[[dataDict["videoName"],dataDict["videoImportance"]]]
            print("new suer data added...")
        pickle.dump(videoAnnotate,open("videoAnnotate.p","wb"))

        print("dictionary saved...")
        response = {"feedback":True}
        return json.dumps(response)
    response = {"feedback":True}
    return json.dumps(response)

def printMyName():
    print("JUNAID")