Update app.py

app.py

@@ -84,15 +84,23 @@ def encode_docs(docs,maxlen = 64, stride = 32):
    
 def predict(query,data):
     name_to_save = data.name.split("/")[-1].split(".")[0][:-8]
-
+    k=20
     st = str([query,name_to_save])
     st_hashed = str(hashlib.sha256(st.encode()).hexdigest()) #just to speed up examples load
     hist = st + " " + st_hashed 
     now = datetime.now()
     current_time = now.strftime("%H:%M:%S")
     try:
-        df = pd.read_csv("{}.csv".format(st_hashed))
-        return df
+        df = pd.read_csv("HISTORY/{}.csv".format(hash(st)))
+        list_outputs = []
+        for i in range(k):
+            temp = [df.iloc[n] for n in range(k)][i]
+            text = ''
+            text += 'PROBABILITIES: '+ temp.Probabilities + '\n\n' 
+            text += 'ANSWER: ' +temp.Answer + '\n\n' 
+            text += 'PASSAGE: '+temp.Passage + '\n\n' 
+            list_outputs.append(text)
+        return list_outputs
     except Exception as e:
         print(e)
         print(st)
@@ -138,24 +146,14 @@ def predict(query,data):
             probabilities = "P(a|p): {}, P(a|p,q): {}, P(p|q): {}".format(round(ans["score"],5), 
                                                                           round(ans["score"]*probs[i],5), 
                                                                           round(probs[i],5))
-            passage = passage.replace(str(ans["answer"]),str(ans["answer"]).upper()) 
             table["Passage"].append(passage)
-            table["Passage"].append("---")
             table["Answer"].append(str(ans["answer"]).upper())
-            table["Answer"].append("---")
             table["Probabilities"].append(probabilities)
-            table["Probabilities"].append("---")
-            table["Source"].append(names)
-            table["Source"].append("---")
         else:
             table["Passage"].append(passage)
-            table["Passage"].append("---")
             table["Answer"].append("no_answer_calculated")
-            table["Answer"].append("---")
             table["Probabilities"].append("P(p|q): {}".format(round(probs[i],5)))
-            table["Probabilities"].append("---")
-            table["Source"].append(names)
-            table["Source"].append("---")
+    
     df = pd.DataFrame(table)
     print("time: "+ str(time.time()-start))
     
@@ -164,7 +162,17 @@ def predict(query,data):
         f.write(" " + str(current_time))
         f.write("\n")
         f.close()
-    df.to_csv("{}.csv".format(st_hashed), index=False)
+    df.to_csv("HISTORY/{}.csv".format(hash(st)), index=False)
+    
+    list_outputs = []
+    for i in range(k):
+        text = ''
+        temp = [df.iloc[n] for n in range(k)][i]
+        text += 'PROBABILITIES: '+ temp.Probabilities + '\n\n' 
+        text += 'ANSWER: ' +temp.Answer + '\n\n' 
+        text += 'PASSAGE: '+temp.Passage + '\n\n' 
+  
+        list_outputs.append(text)
     
     return df
 
@@ -178,7 +186,7 @@ iface = gr.Interface(examples = [
               gr.inputs.File(),
     ],
     outputs = [
-        gr.outputs.Dataframe(),
+        gr.outputs.Carousel(['text']),
             ],
     description=description,
     title = title,