Update app.py

app.py

@@ -6,11 +6,13 @@ from gradio import components
 from gradio import Interface
 import xlsxwriter
 
-# função da tratamento de dados por fatores
-def avaliacao_imovel(planilha):
+# Função de avaliação do imóvel
+def avaliacao_imovel(planilha, num_linhas_desejadas=5):
     # Lendo a aba 'avaliando' da planilha
     df_avaliando = pd.read_excel(planilha.name, 'avaliando')
-    df_dados = pd.read_excel(planilha.name, 'dados')
+    
+    # Lendo a aba 'dados' da planilha, limitando o número de linhas
+    df_dados = pd.read_excel(planilha.name, 'dados').iloc[:int(num_linhas_desejadas)]
 
         #-----------------#
     
@@ -399,6 +401,7 @@ interface = gr.Interface(
     fn=avaliacao_imovel,
     inputs=[
         gr.components.File(label="Upload planilha", type="file"),
+        gr.inputs.Number(label="Número de linhas desejadas", default=5),
     ],
     outputs=[
         gr.components.File(label="Download planilha"),