Update app.py

app.py

@@ -5,7 +5,7 @@ from zipfile import ZipFile
 from huggingface_hub import InferenceClient
 
 # Configuração do modelo e token de autenticação
-MODEL_NAME = "microsoft/codebert-base"  # Substituímos pelo modelo CodeBERT
+MODEL_NAME = "microsoft/codebert-base"  # Modelo CodeBERT
 TOKEN = os.getenv("HF_TOKEN", "")  # Defina a variável de ambiente HF_TOKEN com seu token de acesso
 client = InferenceClient(model=MODEL_NAME, token=TOKEN)
 
@@ -52,19 +52,17 @@ def chat_with_codebert(chat_history, user_input):
         return f"Erro ao processar a resposta da IA: {e}"
 
 # Interface de upload
-def upload_and_analyze(zip_file):
+def upload_and_analyze(zip_file_path):
     # Salvar o arquivo ZIP enviado
     if not os.path.exists(UPLOAD_DIR):
         os.makedirs(UPLOAD_DIR)
-    zip_path = os.path.join(UPLOAD_DIR, zip_file.name)
-    with open(zip_path, "wb") as f:
-        f.write(zip_file.read())
     
     # Processar o ZIP para análise
     try:
-        analysis = process_uploaded_folder(zip_path)
+        analysis = process_uploaded_folder(zip_file_path)
     finally:
-        os.remove(zip_path)  # Remover o arquivo ZIP após processamento
+        if os.path.exists(zip_file_path):
+            os.remove(zip_file_path)  # Remover o arquivo ZIP após processamento
     return analysis
 
 # Interface gráfica com Gradio
@@ -72,7 +70,7 @@ with gr.Blocks() as demo:
     gr.Markdown("# Analisador de APK com CodeBERT 🚀")
     
     with gr.Tab("Upload e Análise"):
-        upload_input = gr.File(label="Envie um arquivo ZIP com o código Java", type="file")
+        upload_input = gr.File(label="Envie um arquivo ZIP com o código Java", type="filepath")  # Corrigido para usar `filepath`
         analysis_output = gr.Textbox(label="Resultado da Análise", lines=20)
         analyze_button = gr.Button("Analisar")
         analyze_button.click(upload_and_analyze, inputs=upload_input, outputs=analysis_output)