Upload app.py

app.py

@@ -4,6 +4,7 @@ from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
 import csv
 import os
+import base64
 
 # Configurações iniciais
 repo_id = "xtuner/llava-llama-3-8b-v1_1-gguf"
@@ -34,15 +35,21 @@ def save_to_csv(name, age, support_level, stimulus_type, file_path):
             writer.writerow(["Nome", "Idade", "Nível de Suporte", "Tipo de Estímulo"])
         writer.writerow([name, age, support_level, stimulus_type])
 
-# Função para converter a imagem em uma representação textual (base64 ou descrição simulada)
-import base64
-
-def process_image_to_text(image_path):
-    # Converte a imagem em uma string base64
-    with open(image_path, "rb") as image_file:
-        base64_string = base64.b64encode(image_file.read()).decode('utf-8')
+# Função para converter a imagem em uma string base64
+def process_image_to_text(uploaded_file):
+    # Converte o arquivo UploadedFile em uma string base64
+    base64_string = base64.b64encode(uploaded_file.read()).decode('utf-8')
     return base64_string
 
+# Função para salvar os dados em um CSV com codificação UTF-8
+def save_to_csv(name, age, support_level, stimulus_type, file_path):
+    file_exists = os.path.isfile(file_path)
+    with open(file_path, mode='a', newline='', encoding='utf-8') as file:
+        writer = csv.writer(file)
+        if not file_exists:
+            writer.writerow(["Nome", "Idade", "Nível de Suporte", "Tipo de Estímulo"])
+        writer.writerow([name, age, support_level, stimulus_type])
+
 # Frontend usando Streamlit
 st.title("Sugestão de Brincadeiras para Crianças Autistas")
 
@@ -61,14 +68,14 @@ if uploaded_file is not None:
     # Exibe a imagem enviada
     raw_image = Image.open(uploaded_file)
     st.image(raw_image, caption='Imagem enviada', use_column_width=False)
-    
-    # Processa a imagem para ser incluída no prompt
-    image_text = process_image_to_text(uploaded_file)
+
+    # Converte a imagem em base64
+    image_base64 = process_image_to_text(uploaded_file)
 
     # Botão para gerar a resposta
     if st.button("Gerar Sugestão de Brincadeira"):
         with st.spinner("Processando..."):
-            prompt = f"<|start_header_id|>user<|end_header_id|>\n\n<image>\n{image_text}\nQual brincadeira segura você sugere e explica em forma de uma receita passo a passo que utiliza o objeto dessa imagem para {name}, uma criança de {age} anos com autismo de nível {support_level} que responde bem a estímulos {stimulus_type}?<|eot_id|>\n<|start_header_id|>assistant<|end_header_id|>\n\n"
+            prompt = f"Qual brincadeira segura você sugere e explica em forma de uma receita passo a passo que utiliza o objeto dessa imagem (base64: {image_base64}) para {name}, uma criança de {age} anos com autismo de nível {support_level} que responde bem a estímulos {stimulus_type}?"
 
             # Gera a sugestão de brincadeira
             response = model(
@@ -87,5 +94,4 @@ if uploaded_file is not None:
         save_to_csv(name, age, support_level, stimulus_type, csv_file)
         
         st.success("Sugestão de Brincadeira:")
-        st.write(suggestion)
-
+        st.write(suggestion)