app file

app.py

@@ -34,7 +34,8 @@ def image_mod(link):
     lexrank_switch = True
     html = ''
     images = []
-    html, images = su.getSummary(link, lexrank_switch, rpunkt_switch)
+    json_file = su.getSummary(link, lexrank_switch, rpunkt_switch)
+    #html, images = su.getSummary(link, lexrank_switch, rpunkt_switch)
     #images = su.getSummaryImage(link, lexrank_switch, rpunkt_switch)
     print(html)
 
@@ -52,8 +53,9 @@ def image_mod(link):
  
     print('images',images)
 
-    return {"html": html, "images":images}
+    #return {"html": html, "images":images}
     #return HTMLResponse(content=html)
+    return json_file 
 
 
 @app.get("/html")

summarizer.py

@@ -128,11 +128,11 @@ def getSummary(link, lexrank_switch, rpunkt_switch):
     proc_list = fr.extractImagesFromVideo(dict_timestamp_summary.keys())
     print('frames: '+str(proc_list))
 
-    html_file = ts.convertToJSON(dict_timestamp_summary)
-    images = ts.getImages(dict_timestamp_summary)
-
-    return html_file, images
+    json_file = ts.convertToJSON(dict_timestamp_summary)
+    #images = ts.getImages(dict_timestamp_summary)
 
+    #return json_file, images
+    return json_file
 
 #filename='/Users/hujo/Downloads/Channel_Summaries/wholesaleted.srt.pnct.txt'
 #with open(filename, 'w') as the_file:

transcript.py

@@ -100,15 +100,22 @@ def getSentences(raw_transcript):
     #return dsl
 
 def convertToJSON(dsl):
-    workdir = 'file/workdir/'
+    workdir = './workdir/'
     cnt=1
     json_rows = '['
     for (key,val) in dsl.items():
         image='frame_'+f"{int(cnt):04d}"+'.jpg'
+
+        # open image and convert it to base64 image
+        with open(workdir+image, 'rb') as open_file:
+             byte_content = open_file.read()
+         base64_bytes = base64.b64encode(byte_content)
+         base64_string = base64_bytes.decode('utf-8')
+
         sentence = val 
         row = '{"image_id": "'+str(cnt)+'",'
         row = row + '"timestamp": "'+key+'",'
-        row = row + '"image": "'+workdir+image+'",'
+        row = row + '"image": "'+base64_string+'",'
         row = row + '"caption": "'+sentence+'"},'
         json_rows = json_rows + row
         cnt = cnt+1