Update app.py

app.py

@@ -3,28 +3,27 @@ import streamlit as st
 import pandas as pd
 import json
 from datetime import datetime
+from pathlib import Path
 import firebase_admin
 from firebase_admin import credentials, firestore
-import os
 
-from groq import Groq
+# Inicializar o Firebase, se ainda não estiver inicializado
+if not firebase_admin._apps:
+    cred = credentials.Certificate("playbot-f0897-firebase-adminsdk-uay8b-0c87a1f1d0.json")  # Substitua pelo caminho correto
+    firebase_admin.initialize_app(cred)
 
-# Inicialização do cliente Groq
-client = Groq(
-    api_key=os.getenv("GROQ_API"),
-)
+# Cliente Firestore
+db = firestore.client()
 
 llava_model = 'llava-v1.5-7b-4096-preview'
 llama31_model = 'llama-3.1-70b-versatile'
 
-# 1. Inicializar o Firebase Admin SDK
-cred = credentials.Certificate("playbot-f0897-firebase-adminsdk-uay8b-0c87a1f1d0.json")  # Substitua pelo caminho correto
-firebase_admin.initialize_app(cred)
-
-# Conectar ao Firestore
-db = firestore.client()
+# 2. Função para codificação de imagem
+def encode_image(image_path):
+    with open(image_path, "rb") as image_file:
+        return base64.b64encode(image_file.read()).decode('utf-8')
 
-# 2. Função para converter imagem em texto
+# 3. Função para converter imagem em texto
 def image_to_text(client, model, base64_image, prompt):
     chat_completion = client.chat.completions.create(
         messages=[
@@ -45,13 +44,13 @@ def image_to_text(client, model, base64_image, prompt):
     )
     return chat_completion.choices[0].message.content
 
-# 3. Função para gerar uma brincadeira
+# 4. Função para gerar uma brincadeira
 def game_generation(client, image_description, age, support_level, stimulus_type):
     chat_completion = client.chat.completions.create(
         messages=[
             {
                 "role": "system",
-                "content": f"Suggest a safe game for a child of {age} years old, with support level {support_level} and focus on {stimulus_type}. The game should use the object described in the image and should not use objects that are dangerous for children...",
+                "content": f"Suggest a safe game for a child of {age} years old, with support level {support_level} and focus on {stimulus_type}. The game should use the object described in the image and should not use objects that are dangerous for children, for example, scissors, knives, toothpicks, electrical wires, any sharp object cannot be used. If the object is not dangerous, describe the game in the form of a step-by-step recipe. Note: Write in Portuguese.",
             },
             {
                 "role": "user",
@@ -62,21 +61,16 @@ def game_generation(client, image_description, age, support_level, stimulus_type
     )
     return chat_completion.choices[0].message.content
 
-# 4. Função para salvar os dados no Firestore
-def save_to_firestore(idade, nivel_suporte, tipo_estimulo):
-    doc_ref = db.collection('usuarios').add({
-        "Idade": idade,
-        "Nível de Suporte": nivel_suporte,
-        "Tipo de Estimulo": tipo_estimulo,
-        "datetime": datetime.now()
-    })
+# Função para salvar dados no Firebase Firestore
+def save_to_firestore(data):
+    db.collection('usuarios').add(data)
     st.success("Dados registrados no Firebase Firestore com sucesso.")
 
-# 5. Função principal do app Streamlit
+# Função principal do app Streamlit
 def main():
     st.image("autism.png", width=200)
     st.title("PlayBot - Gerador de Brincadeiras", anchor="center")
-    st.write("Conheça o PlayBot, um assistente inteligente que gera brincadeiras...")
+    st.write("Conheça o PlayBot, um assistente inteligente que gera brincadeiras divertidas e educativas para crianças com TEA...")
 
     # Entrada de dados do usuário
     idade = st.number_input("Idade da criança:", min_value=1, max_value=16, step=1)
@@ -98,8 +92,16 @@ def main():
         game_description = game_generation(client, image_description, idade, nivel_suporte, tipo_estimulo)
         st.write(game_description)
 
-        # Salvando dados no Firestore
-        save_to_firestore(idade, nivel_suporte, tipo_estimulo)
+        # Salvando dados no Firebase Firestore
+        data = {
+            "Idade": idade,
+            "Nível de Suporte": nivel_suporte,
+            "Tipo de Estimulo": tipo_estimulo,
+            "Descrição da Brincadeira": game_description,
+            "Timestamp": datetime.now().isoformat()
+        }
+
+        save_to_firestore(data)
 
 # Executar o app
 if __name__ == "__main__":