Update app.py

app.py

@@ -276,34 +276,11 @@ def mergeCloseBoxes(pr, bb, wr, threshold):
 
   return final_bbox, final_preds, final_words
 
-def createDataframe(preds, words):
-  df = pd.DataFrame(columns = ['song name', 'artist', 'year', 'album', 'genres', 'song writer', 'lyrics', 'others'])
-  if (len(preds) > 0):
-      flag_label = preds[0]
-      #print(preds)
-      #print(words)
-      #print('@@@@@')
-      #print(flag_label)
-      row_number = -1
-      for i in range(len(preds)):
-          #print('i is: {}'.format(i))
-          if (preds[i] == flag_label):
-              row_number = row_number + 1
-              df.at[row_number, preds[i]] = words[i]
-              #print('row number is: {}'.format(row_number))
-              continue
-
-          else:
-            #print('row_number {} is <= of df.shape {}'.format(row_number, df.shape[0]))
-            #print(pd.isna(df[preds[i]].iloc[row_number]))
-            #print(pd.isna(df[preds[i]].iloc[row_number]))
-            if(pd.isna(df[preds[i]].iloc[row_number])):
-              df.at[row_number, preds[i]] = words[i]
-            else:
-              row_number = row_number + 1
-              df.at[row_number, preds[i]] = words[i]
-  
-  return df
+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
 
 def isInside(w, z):
     # return True if w is inside z, if z is inside w return false