Update app.py

app.py

@@ -1,13 +1,13 @@
-from groq import Groq
 import base64
 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
-import sqlite3
-from sqlalchemy import create_engine
+
+from groq import Groq
 
 # Inicialização do cliente Groq
 client = Groq(
@@ -17,12 +17,14 @@ client = Groq(
 llava_model = 'llava-v1.5-7b-4096-preview'
 llama31_model = 'llama-3.1-70b-versatile'
 
-# 2. 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')
+# 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)
 
-# 3. Função para converter imagem em texto
+# Conectar ao Firestore
+db = firestore.client()
+
+# 2. Função para converter imagem em texto
 def image_to_text(client, model, base64_image, prompt):
     chat_completion = client.chat.completions.create(
         messages=[
@@ -43,13 +45,13 @@ def image_to_text(client, model, base64_image, prompt):
     )
     return chat_completion.choices[0].message.content
 
-# 4. Função para gerar uma brincadeira
+# 3. 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, 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.",
+                "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...",
             },
             {
                 "role": "user",
@@ -60,11 +62,21 @@ def game_generation(client, image_description, age, support_level, stimulus_type
     )
     return chat_completion.choices[0].message.content
 
-# Função principal do app Streamlit
+# 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()
+    })
+    st.success("Dados registrados no Firebase Firestore com sucesso.")
+
+# 5. 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 divertidas e educativas para crianças com TEA...")
+    st.write("Conheça o PlayBot, um assistente inteligente que gera brincadeiras...")
 
     # Entrada de dados do usuário
     idade = st.number_input("Idade da criança:", min_value=1, max_value=16, step=1)
@@ -86,22 +98,10 @@ def main():
         game_description = game_generation(client, image_description, idade, nivel_suporte, tipo_estimulo)
         st.write(game_description)
 
-        # Salvando dados no banco de dados
-        data = {
-            "Idade": idade,
-            "Nível de Suporte": nivel_suporte,
-            "Tipo de Estimulo": tipo_estimulo
-        }
-        df = pd.DataFrame([data])
-
-        # Conexão com o banco de dados SQLite
-        engine = create_engine('sqlite:///dados_usuarios.db')
-
-        # Inserindo os dados na tabela 'usuarios'
-        df.to_sql('usuarios', con=engine, if_exists='append', index=False)
-
-        st.success("Dados registrados no banco de dados com sucesso.")
+        # Salvando dados no Firestore
+        save_to_firestore(idade, nivel_suporte, tipo_estimulo)
 
 # Executar o app
 if __name__ == "__main__":
     main()
+