added logic for graphing

Rishab · Sep 8, 2019 · 28eb5cb · 28eb5cb
1 parent 897a401
commit 28eb5cb
Show file tree

Hide file tree

Showing 5,300 changed files with 1,164,401 additions and 96,701 deletions.
diff --git a/.ipynb_checkpoints/test-checkpoint.ipynb b/.ipynb_checkpoints/test-checkpoint.ipynb
diff --git a/__pycache__/app.cpython-36.pyc b/__pycache__/app.cpython-36.pyc
diff --git a/__pycache__/aws.cpython-36.pyc b/__pycache__/aws.cpython-36.pyc
diff --git a/__pycache__/frames.cpython-36.pyc b/__pycache__/frames.cpython-36.pyc
diff --git a/__pycache__/intelliment.cpython-36.pyc b/__pycache__/intelliment.cpython-36.pyc
diff --git a/app.py b/app.py
@@ -1,8 +1,10 @@
 import os
-#import magic
+from frames import frame_capture
+from aws import build_dataset
 import urllib.request
 from flask import Flask, flash, request, redirect, render_template
 from werkzeug.utils import secure_filename
+from intelliment import generate_graphs
 
 ALLOWED_EXTENSIONS = set(['mp4'])
 
@@ -35,9 +37,17 @@ def upload_file():
             return redirect(request.url)
         if file and allowed_file(file.filename):
             filename = secure_filename(file.filename)
+            print(filename)
             file.save(os.path.join(app.config['UPLOAD_FOLDER'], filename))
             flash('File successfully uploaded')
+            file_path = './static/upload/' + filename
+            frame_capture(file_path)
+            build_dataset()
+            generate_graphs()
             return redirect('/')
+        else:
+            flash('Allowed file types are txt, pdf, png, jpg, jpeg, gif')
+            return redirect(request.url)
 
 
 if __name__ == "__main__":

diff --git a/aws.py b/aws.py
@@ -4,48 +4,50 @@
 import xlsxwriter
 import os
 
-with open('credentials.csv', 'r') as input:
-    next(input)
-    reader = csv.reader(input)
-    for line in reader:
-        access_key_id = line[0]
-        secret_access_key = line[1]
-
-rekognition = boto3.client(
-    "rekognition", aws_access_key_id=access_key_id, aws_secret_access_key=secret_access_key, region_name='us-east-1')
-
-frames = os.listdir('video_frames')
-
-workbook = xlsxwriter.Workbook('sentigrade.xlsx')
-worksheet = workbook.add_worksheet()
-
-worksheet.write('A1', 'Second')
-worksheet.write('B1', 'Image')
-worksheet.write('C1', 'Emotion')
-worksheet.write('D1', 'Confidence')
-
-currRow = 2
-
-for frame in frames:
-    print(frame)
-    photo = 'video_frames/' + frame
-    with open(photo, 'rb') as source_image:
-        source_bytes = source_image.read()
-    response = rekognition.detect_faces(
-        Image={'Bytes': source_bytes}, Attributes=['ALL'])
-
-    worksheet.write('A%d' % currRow, frame.replace('.jpg', ''))
-
-    maxEmotion = response['FaceDetails'][0]['Emotions']
-    maxNum = 0.0
-    for emotion in response['FaceDetails'][0]['Emotions']:
-        if emotion['Confidence'] > maxNum:
-            maxNum = emotion['Confidence']
-            maxEmotion = emotion['Type']
-
-    worksheet.write('B%d' % currRow, frame)
-    worksheet.write('C%d' % currRow, maxEmotion)
-    worksheet.write('D%d' % currRow, maxNum)
-    currRow += 1
-
-workbook.close()
+
+def build_dataset():
+    with open('credentials.csv', 'r') as input:
+        next(input)
+        reader = csv.reader(input)
+        for line in reader:
+            access_key_id = line[0]
+            secret_access_key = line[1]
+
+    rekognition = boto3.client(
+        "rekognition", aws_access_key_id=access_key_id, aws_secret_access_key=secret_access_key, region_name='us-east-1')
+
+    frames = os.listdir('video_frames')
+
+    workbook = xlsxwriter.Workbook('dataset.xlsx')
+    worksheet = workbook.add_worksheet()
+
+    worksheet.write('A1', 'Second')
+    worksheet.write('B1', 'Image')
+    worksheet.write('C1', 'Emotion')
+    worksheet.write('D1', 'Confidence')
+
+    currRow = 2
+
+    for frame in frames:
+        print(frame)
+        photo = 'video_frames/' + frame
+        with open(photo, 'rb') as source_image:
+            source_bytes = source_image.read()
+        response = rekognition.detect_faces(
+            Image={'Bytes': source_bytes}, Attributes=['ALL'])
+
+        worksheet.write('A%d' % currRow, frame.replace('.jpg', ''))
+
+        maxEmotion = response['FaceDetails'][0]['Emotions']
+        maxNum = 0.0
+        for emotion in response['FaceDetails'][0]['Emotions']:
+            if emotion['Confidence'] > maxNum:
+                maxNum = emotion['Confidence']
+                maxEmotion = emotion['Type']
+
+        worksheet.write('B%d' % currRow, frame)
+        worksheet.write('C%d' % currRow, maxEmotion)
+        worksheet.write('D%d' % currRow, maxNum)
+        currRow += 1
+
+    workbook.close()
diff --git a/bin/__pycache__/runxlrd.cpython-36.pyc b/bin/__pycache__/runxlrd.cpython-36.pyc