app file

summarizer.py

@@ -128,7 +128,7 @@ def getSummary(link, lexrank_switch, rpunkt_switch):
     proc_list = fr.extractImagesFromVideo(dict_timestamp_summary.keys())
     print('frames: '+str(proc_list))
 
-    html_file = ts.convertToHTML(dict_timestamp_summary)
+    html_file = ts.convertToJSON(dict_timestamp_summary)
     images = ts.getImages(dict_timestamp_summary)
 
     return html_file, images

transcript.py

@@ -99,6 +99,25 @@ def getSentences(raw_transcript):
 
     #return dsl
 
+def convertToJSON(dsl):
+    workdir = 'file/workdir/'
+    cnt=1
+    json_rows = '['
+    for (key,val) in dsl.items():
+        image='frame_'+f"{int(cnt):04d}"+'.jpg'
+        sentence = val 
+        row = '{"Image ID": '+str(cnt)+','
+        row = row + '"Timestamp": '+key+','
+        row = row + '"Image": '+workdir+image+','
+        row = row + '"Caption": '+sentence+'},'
+        json_rows = json_rows + row
+        cnt = cnt+1
+    # remove the comma from the last item
+    json_rows = json_rows[:-1] + ']'
+
+    return json_rows
+
+
 def convertToHTML(dsl):
     #workdir = 'file/workdir/'
     workdir = '../workdir/'