FabioSantos commited on
Commit
eef4676
·
verified ·
1 Parent(s): b1a497d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -4
app.py CHANGED
@@ -4,6 +4,9 @@ import base64
4
  import streamlit as st
5
  import pandas as pd
6
  import json
 
 
 
7
 
8
  client = Groq(
9
  api_key="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",
@@ -57,11 +60,21 @@ def game_generation(client, image_description, age, support_level, stimulus_type
57
 
58
  return chat_completion.choices[0].message.content
59
 
60
- def save_json(idade: str, nivel_suporte: str, tipo_estimulo) -> None:
61
- with scheduler.lock:
 
 
 
 
62
  with JSON_DATASET_PATH.open("a") as f:
63
- json.dump({"Idade": idade, "NivelDeSuporte": nivel_suporte, "TipoDeEstimulo": tipo_estimulo, "datetime": datetime.now().isoformat()}, f)
64
- f.write("\n")
 
 
 
 
 
 
65
 
66
  # 5. Streamlit app
67
  def main():
 
4
  import streamlit as st
5
  import pandas as pd
6
  import json
7
+ from datetime import datetime
8
+ from pathlib import Path
9
+ from threading import Lock
10
 
11
  client = Groq(
12
  api_key="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",
 
60
 
61
  return chat_completion.choices[0].message.content
62
 
63
+ # Suponha que o caminho para o arquivo JSON esteja definido assim
64
+ JSON_DATASET_PATH = Path("dados.json")
65
+ scheduler = Lock() # Exemplo de lock, pode ser substituído pelo seu objeto de scheduler com lock
66
+
67
+ def save_json(idade: str, nivel_suporte: str, tipo_estimulo: str) -> None:
68
+ with scheduler: # Usando o Lock diretamente
69
  with JSON_DATASET_PATH.open("a") as f:
70
+ # Dump dos dados em formato JSON, seguido de uma nova linha
71
+ json.dump({
72
+ "Idade": idade,
73
+ "NivelDeSuporte": nivel_suporte,
74
+ "TipoDeEstimulo": tipo_estimulo,
75
+ "datetime": datetime.now().isoformat()
76
+ }, f)
77
+ f.write("\n") # Para garantir que cada registro fique em uma linha nova)
78
 
79
  # 5. Streamlit app
80
  def main():