Update app.py

app.py

@@ -280,7 +280,9 @@ def createDataframe(predictions, words):
     result1, result2 = zip(*sorted(zip(predictions, words), key=lambda x: x[0]))
     d = {'key':result1,'value':result2}
     df = pd.DataFrame(d)
-    return df
+    df1 = df[df['key'] != 'lyrics']
+    df2 = df[df['key'] == 'lyrics']
+    return df1, df2
 
 def isInside(w, z):
     # return True if w is inside z, if z is inside w return false
@@ -344,7 +346,8 @@ def mergeImageVertical(a):
   return imgs_comb
 
 def completepreprocess(pdffile):
-  myDataFrame = pd.DataFrame()
+  dfOne = pd.DataFrame()
+  dfTwo = pd.DataFrame()
   a=[]
   doc = fitz.open(pdffile)
   for i in range(0,len(doc)):
@@ -366,11 +369,12 @@ def completepreprocess(pdffile):
       pred_list.append(iob_to_label(number))
     _bbox, _preds, _words = process_form(pred_list, words, bbox)
     print('page: ' + str(i) + '  ' + str(len(_preds))+ '  ' + str(len(_words)))
-    df = createDataframe(_preds, _words)
-    myDataFrame=myDataFrame.append(df)
+    df1, df2 = createDataframe(_preds, _words)
+    dfOne=dfOne.append(df1)
+    dfTwo=dfOne.append(df2)
 
   im2=mergeImageVertical(a)  
-  return im2,myDataFrame
+  return im2,dfOne,dfTwo
 
 
 title = "Interactive demo: Music Information Extraction model"
@@ -384,7 +388,7 @@ iface = gr.Interface(fn=completepreprocess,
                      #inputs=gr.inputs.Image(type="pil",optional=True,label="upload file"),
                      inputs=gr.File(label="PDF"),
                      #inputs=gr.inputs.Image(type="pil")
-                     outputs=[gr.outputs.Image(type="pil", label="annotated image"),"dataframe"] ,
+                     outputs=[gr.outputs.Image(type="pil", label="annotated image"),"dataframe", "dataframe"] ,
                      title=title,
                      description=description,
                      examples=examples,