Ctaake commited on
Commit
5a86dcd
·
verified ·
1 Parent(s): 00b142b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +16 -5
app.py CHANGED
@@ -1,9 +1,19 @@
1
  import gradio as gr
2
- from huggingface_hub import InferenceClient, TextGenerationStreamOutput
3
  import random
4
  from transformers import AutoTokenizer
5
  from mySystemPrompt import SYSTEM_PROMPT, SYSTEM_PROMPT_PLUS,SYSTEM_PROMPT_NOUS
6
  from datetime import datetime
 
 
 
 
 
 
 
 
 
 
7
 
8
 
9
  # Model which is used
@@ -88,10 +98,11 @@ def event_voting(vote_data: gr.LikeData):
88
  if vote_data.liked:
89
  pass
90
  else:
91
- vote_message = f'{datetime.now().strftime(" %I: %M %p on %B %d, %Y")}:\n{vote_data.value}\n'
92
- with open(path_to_log, "a") as f:
93
- f.write(vote_message)
94
-
 
95
 
96
  myAdditionalInputs = [
97
  gr.Textbox(
 
1
  import gradio as gr
2
+ from huggingface_hub import InferenceClient, TextGenerationStreamOutput, CommitSchedulder
3
  import random
4
  from transformers import AutoTokenizer
5
  from mySystemPrompt import SYSTEM_PROMPT, SYSTEM_PROMPT_PLUS,SYSTEM_PROMPT_NOUS
6
  from datetime import datetime
7
+ import json
8
+ from pathlib import Path
9
+ from uuid import uuid4
10
+
11
+
12
+ # For log
13
+ DATASET_DIR =Path("FranziBotLog")
14
+ DATASET_DIR.mkdir(parents=True,exist_ok=True)
15
+ DATASET_PATH = DATASET_DIR / f"train-{uuid4()}.json"
16
+ scheduler = CommitSchedulder(repo_id="FranziBotLog",repo_type="dataset",folder_path=DATASET_DIR,path_in_repo="data")
17
 
18
 
19
  # Model which is used
 
98
  if vote_data.liked:
99
  pass
100
  else:
101
+ with scheduler.lock:
102
+ with DATASET_PATH.open("a") as f:
103
+ json.dump({"timestamp":datetime.now().isoformat(),"message":vote_data.value},f)
104
+ f.write("\n")
105
+
106
 
107
  myAdditionalInputs = [
108
  gr.Textbox(