Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -16,7 +16,7 @@ os.environ["OPENAI_API_KEY"] = os.environ['SECRET_CODE']
|
|
| 16 |
DATASET_REPO_URL = "https://huggingface.co/datasets/peterpull/MediatorBot"
|
| 17 |
DATA_FILENAME = "data.txt"
|
| 18 |
INDEX_FILENAME = "index.json"
|
| 19 |
-
|
| 20 |
INDEX_FILE = os.path.join("data", INDEX_FILENAME)
|
| 21 |
|
| 22 |
# we need a write access token.
|
|
@@ -50,17 +50,9 @@ def store_message(chatinput: str, chatresponse: str):
|
|
| 50 |
file.write(f"{datetime.now()},{chatinput},{chatresponse}\n")
|
| 51 |
print(f"Wrote to datafile: {datetime.now()},{chatinput},{chatresponse}\n")
|
| 52 |
|
| 53 |
-
# Push back to hub every 10th time the function is called
|
| 54 |
-
if store_message.count_calls % 10 == 0:
|
| 55 |
-
print("Pushing back to Hugging Face model hub")
|
| 56 |
-
# Call the push_to_hub() function to push the changes to the hub
|
| 57 |
-
push_to_hub(commit_message="Added new chat data")
|
| 58 |
-
store_message.count_calls += 1
|
| 59 |
|
| 60 |
return generate_text()
|
| 61 |
|
| 62 |
-
store_message.count_calls = 1 #initiases the count at one. We want to count how many messages stored before pushing back to repo.
|
| 63 |
-
|
| 64 |
#gets the index file which is the context data
|
| 65 |
def get_index(index_file_path):
|
| 66 |
if os.path.exists(index_file_path):
|
|
@@ -72,7 +64,7 @@ def get_index(index_file_path):
|
|
| 72 |
sys.exit()
|
| 73 |
|
| 74 |
index = get_index(INDEX_FILE)
|
| 75 |
-
|
| 76 |
# passes the prompt to the chatbot
|
| 77 |
def chatbot(input_text, mentioned_person='Mediator John Haynes', confidence_threshold=0.5):
|
| 78 |
prompt = f"You are {mentioned_person}. Answer this: {input_text}. Only reply from the contextual data, or say you don't know. At the end of your answer ask an insightful question."
|
|
|
|
| 16 |
DATASET_REPO_URL = "https://huggingface.co/datasets/peterpull/MediatorBot"
|
| 17 |
DATA_FILENAME = "data.txt"
|
| 18 |
INDEX_FILENAME = "index.json"
|
| 19 |
+
DATA_FILEP = os.path.join("data", DATA_FILENAME)
|
| 20 |
INDEX_FILE = os.path.join("data", INDEX_FILENAME)
|
| 21 |
|
| 22 |
# we need a write access token.
|
|
|
|
| 50 |
file.write(f"{datetime.now()},{chatinput},{chatresponse}\n")
|
| 51 |
print(f"Wrote to datafile: {datetime.now()},{chatinput},{chatresponse}\n")
|
| 52 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
|
| 54 |
return generate_text()
|
| 55 |
|
|
|
|
|
|
|
| 56 |
#gets the index file which is the context data
|
| 57 |
def get_index(index_file_path):
|
| 58 |
if os.path.exists(index_file_path):
|
|
|
|
| 64 |
sys.exit()
|
| 65 |
|
| 66 |
index = get_index(INDEX_FILE)
|
| 67 |
+
|
| 68 |
# passes the prompt to the chatbot
|
| 69 |
def chatbot(input_text, mentioned_person='Mediator John Haynes', confidence_threshold=0.5):
|
| 70 |
prompt = f"You are {mentioned_person}. Answer this: {input_text}. Only reply from the contextual data, or say you don't know. At the end of your answer ask an insightful question."
|