Stats

app.py

@@ -7,6 +7,7 @@ import pdb
 import json
 from twc_embeddings import HFModel,SimCSEModel,SGPTModel
 import torch
+import requests
 
 
 MAX_INPUT = 100
@@ -20,28 +21,32 @@ use_case = {"1":"Finding similar phrases/sentences","2":"Retrieving semantically
 use_case_url = {"1":"https://huggingface.co/spaces/taskswithcode/semantic_similarity","2":"https://huggingface.co/spaces/taskswithcode/semantic_search","3":""}
 
 
+APP_NAME = "hf/semantic_search"
+INFO_URL = "http://www.taskswithcode.com/stats/"
 
 
 from transformers import BertTokenizer, BertForMaskedLM
 
 
 
-view_count_file = "view_count.txt"
 
-def get_views():
+def get_views(action):
     ret_val = 0
     if ("view_count" not in st.session_state):
         try:
-           data = int(open(view_count_file).read().strip("\n"))
+           app_info = {'name': APP_NAME,"action":action}
+           res = requests.post(INFO_URL, json = app_info).json()
+           print(res)
+           data = res["count"]
         except:
            data = 0
-        data += 1
-        with open(view_count_file,"w") as fp:
-            fp.write(str(data))
         ret_val = data
         st.session_state["view_count"] = data
     else:
         ret_val = st.session_state["view_count"]
+        if (action != "init"):
+           app_info = {'name': APP_NAME,"action":action}
+           res = requests.post(INFO_URL, json = app_info).json()
     return "{:,}".format(ret_val)
         
 
@@ -155,6 +160,8 @@ def display_results(orig_sentences,main_index,results,response_info,app_mode):
     main_sent = main_sent + "\n" + '\n'.join(body_sent)
     st.markdown(main_sent,unsafe_allow_html=True)
     st.session_state["download_ready"] = json.dumps(download_data,indent=4)
+    get_views("submit")
+    
 
 
 def init_session():
@@ -172,7 +179,7 @@ def app_main(app_mode,example_files,model_name_files):
   curr_use_case = use_case[app_mode].split(".")[0]
   st.markdown("<h5 style='text-align: center;'>Compare popular/state-of-the-art models for tasks using sentence embeddings</h5>", unsafe_allow_html=True)
   st.markdown(f"<div style='color: #4f4f4f; text-align: left'>Use cases for sentence embeddings<br/>&nbsp;&nbsp;&nbsp;•&nbsp;&nbsp;{use_case['1']}<br/>&nbsp;&nbsp;&nbsp;•&nbsp;&nbsp;<a href=\'{use_case_url['2']}\' target='_blank'>{use_case['2']}</a><br/>&nbsp;&nbsp;&nbsp;•&nbsp;&nbsp;{use_case['3']}<br/><i>This app illustrates <b>'{curr_use_case}'</b> use case</i></div>", unsafe_allow_html=True)
-  st.markdown(f"<div style='color: #9f9f9f; text-align: right'>views:&nbsp;{get_views()}</div>", unsafe_allow_html=True)
+  st.markdown(f"<div style='color: #9f9f9f; text-align: right'>views:&nbsp;{get_views('init')}</div>", unsafe_allow_html=True)
 
 
   try: