Fixing bug

app.py

@@ -2,17 +2,20 @@ import os
 import sys
 import json
 import tempfile
+
 import pandas as pd
 import gradio as gr
+from PIL import Image
 
-# 1) Ajuste de path ANTES de importar smi_ted_light
+# 1) Ajusta o path antes de importar o loader
 BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 INFERENCE_PATH = os.path.join(BASE_DIR, "smi-ted", "inference")
 sys.path.insert(0, INFERENCE_PATH)
 
+# 2) Importa o loader do SMI-TED Light
 from smi_ted_light.load import load_smi_ted
 
-# 2) Carrega o modelo
+# 3) Carrega o modelo
 MODEL_DIR = os.path.join(INFERENCE_PATH, "smi_ted_light")
 model = load_smi_ted(
     folder=MODEL_DIR,
@@ -20,7 +23,7 @@ model = load_smi_ted(
     vocab_filename="bert_vocab_curated.txt",
 )
 
-# 3) Função que retorna STRING JSON + gr.update para o CSV
+# 4) Função que gera o embedding e cria o CSV temporário
 def gerar_embedding_e_csv(smiles: str):
     smiles = smiles.strip()
     if not smiles:
@@ -28,46 +31,52 @@ def gerar_embedding_e_csv(smiles: str):
         return json.dumps(erro), gr.update(visible=False)
 
     try:
+        # Gera o vetor
         vetor = model.encode(smiles, return_torch=True)[0].tolist()
-        # monta CSV
+        # Grava CSV
         df = pd.DataFrame([vetor])
         tmp = tempfile.NamedTemporaryFile(suffix=".csv", delete=False)
         df.to_csv(tmp.name, index=False)
         tmp.close()
-        # retorna JSON-string e torna o link visível
+        # Retorna JSON em string e ativa o link de download
         return json.dumps(vetor), gr.update(value=tmp.name, visible=True)
     except Exception as e:
         erro = {"erro": str(e)}
         return json.dumps(erro), gr.update(visible=False)
 
-# 4) Interface Blocks
+# 5) Monta a interface com Blocks
 with gr.Blocks() as demo:
     gr.Markdown(
         """
-        ## SMI-TED Embedding Generator  
-        Cole uma sequência SMILES e receba:
-        1. Uma **string JSON** com o vetor (Textbox)  
-        2. Um link para **baixar o CSV** (File)  
+        # SMI-TED Embedding Generator  
+        Cole uma sequência SMILES e:
+        - Veja o vetor embedding (JSON)  
+        - Baixe-o em CSV  
         """
     )
 
     with gr.Row():
-        inp_smiles = gr.Textbox(label="SMILES", placeholder="Ex.: CCO")
-        btn = gr.Button("Gerar Embedding")
+        smiles_in = gr.Textbox(label="SMILES", placeholder="Ex.: CCO")
+        gerar_btn = gr.Button("Gerar Embedding")
+
     with gr.Row():
-        out_text = gr.Textbox(
-            label="Embedding (JSON)", 
-            interactive=False, 
-            lines=4, 
-            placeholder='Vai aparecer aqui o vetor como JSON...'
+        embedding_out = gr.Textbox(
+            label="Embedding (JSON)",
+            interactive=False,
+            lines=4,
+            placeholder="O vetor aparecerá aqui…"
+        )
+        download_csv = gr.File(
+            label="Baixar CSV",
+            visible=False
         )
-        out_file = gr.File(label="Download do CSV", visible=False)
 
-    btn.click(
+    # Conecta botão à função que tem dois outputs
+    gerar_btn.click(
         fn=gerar_embedding_e_csv,
-        inputs=inp_smiles,
-        outputs=[out_text, out_file]
+        inputs=smiles_in,
+        outputs=[embedding_out, download_csv]
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0")

requirements.txt

@@ -5,6 +5,5 @@ numpy==1.26.4
 pandas==1.4.0           
 tqdm>=4.66.4
 rdkit>=2024.3.5
-gradio==4.32.0
-gradio-client==0.2.0
+gradio>=4.33.1
 huggingface-hub