log download

code/qa.py

@@ -2,6 +2,7 @@ import streamlit as st
 from transformers import pipeline
 import pandas as pd
 import datetime
+import base64 
 
 @st.cache(allow_output_mutation = True)
 def load_model():
@@ -20,7 +21,15 @@ def load_model():
 #     st.markdown('Page viewed = {} times.'.format(1))
 
 
-st.sidebar.markdown('![Visitor count](https://shields-io-visitor-counter.herokuapp.com/badge?page=https://share.streamlit.io/https://huggingface.co/spaces/aakashgoel12/nlp1&label=VisitorsCount&labelColor=000000&logo=GitHub&logoColor=FFFFFF&color=1D70B8&style=for-the-badge)')
+def csv_downloader(data):
+    csvfile = data.to_csv()
+    b64 = base64.b64encode(csvfile.encode()).decode()
+    new_filename = "results/df_log_file.tsv"
+    st.markdown("#### Download File ###")
+    href = f'<a href="data:file/tsv;base64,{b64}" download="{new_filename}">Click Here!!</a>'
+    st.markdown(href,unsafe_allow_html=True)
+
+st.markdown('![Visitor count](https://shields-io-visitor-counter.herokuapp.com/badge?page=https://share.streamlit.io/https://huggingface.co/spaces/aakashgoel12/nlp1&label=VisitorsCount&labelColor=000000&logo=GitHub&logoColor=FFFFFF&color=1D70B8&style=for-the-badge)')
 
 def load_file():
     """Load text from file"""
@@ -75,12 +84,12 @@ if raw_text != None and raw_text != '':
             res_df = pd.DataFrame({"TimeStamp":[str(datetime.datetime.now())],\
                 "Question":[question],\
                 "Input":[str(raw_text)],"Answer":[str(answer["answer"])],"Score":[str(answer["score"])]})
-            st.write("PASS1")
+            #st.write("PASS1")
         except:
             st.write("FAIL1")
         try:
             res_df.to_csv(output_path, mode='a', index=False, sep="\t", header= not os.path.exists(output_path))
-            st.write("PASS2")
+            #st.write("PASS2")
         except:
             st.write("FAIL2")
         import os