Ctaake commited on
Commit
912f990
·
verified ·
1 Parent(s): 64849b6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -14
app.py CHANGED
@@ -1,20 +1,17 @@
1
  import gradio as gr
2
- from huggingface_hub import InferenceClient, TextGenerationStreamOutput, CommitScheduler
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("FranziBot_feedback")
14
- DATASET_DIR.mkdir(parents=True,exist_ok=True)
15
- DATASET_PATH = DATASET_DIR / f"train-{uuid4()}.json"
16
- scheduler = CommitScheduler(repo_id="Ctaake/FranziBotLog",repo_type="dataset",folder_path=DATASET_DIR,path_in_repo="data")
17
-
18
 
19
  # Model which is used
20
  checkpoint = "CohereForAI/c4ai-command-r-v01"
@@ -97,11 +94,12 @@ def event_voting(vote_data: gr.LikeData):
97
  if vote_data.liked:
98
  pass
99
  else:
100
- print("Dislike")
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 = [
 
1
  import gradio as gr
2
+ from huggingface_hub import InferenceClient, TextGenerationStreamOutput, CommitScheduler, Repository
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 csv
 
 
8
 
9
 
10
  # For log
11
+ DATASET_REPO_URL = "https://huggingface.co/datasets/ctaake/FranziBotLog"
12
+ DATA_FILENAME = "log.csv"
13
+ DATA_FILE = os.path.join("data", DATA_FILENAME)
14
+ repo = Repository(local_dir="data", clone_from=DATASET_REPO_URL)
 
15
 
16
  # Model which is used
17
  checkpoint = "CohereForAI/c4ai-command-r-v01"
 
94
  if vote_data.liked:
95
  pass
96
  else:
97
+ with open(DATA_FILE, "a") as csvfile:
98
+ writer = csv.DictWriter(csvfile, fieldnames=[ "message", "time"])
99
+ writer.writerow(
100
+ { "message": vote_data.value, "time": str(datetime.now().isoformat())})
101
+ commit_url = repo.push_to_hub()
102
+ print(commit_url)
103
 
104
 
105
  myAdditionalInputs = [