Update app.py

app.py

@@ -10,8 +10,11 @@ from reportlab.platypus import SimpleDocTemplate, Paragraph
 from reportlab.lib.styles import getSampleStyleSheet
 import shutil
 import os
+import plotly.express as px
 
 
+        #-----------------#
+    
 # Function to save results in a PDF file
 def save_results_to_pdf(results_formatados, intervalo_confiança, valores_finais):
     doc = SimpleDocTemplate("resultados.pdf", pagesize=letter)
@@ -34,6 +37,8 @@ def save_results_to_pdf(results_formatados, intervalo_confiança, valores_finais
     
     # Build the PDF
     doc.build(elements)
+    
+        #-----------------#
 
 def renderizar_dataframe(df):
     try:
@@ -49,7 +54,45 @@ def renderizar_dataframe(df):
         return tabela_com_rolagem
     except Exception as e:
         return f"Erro ao processar o DataFrame: {str(e)}"
+    
+        #-----------------#
+    
+def plotar_mapa_com_dois_dataframes(df1, df2):
+    fig1 = px.scatter_mapbox(
+        df1,
+        lat='lat',
+        lon='lon',
+        zoom=15,
+        center={"lat": df1['lat'].mean(), "lon": df1['lon'].mean()},
+        color_discrete_sequence=['#008B8B'],
+    )
+    
+    fig1.update_traces(marker=dict(size=10))  # Define o tamanho dos marcadores para o DataFrame 1
+
+    fig2 = px.scatter_mapbox(
+        df2,
+        lat='lat',
+        lon='lon',
+        color_discrete_sequence=['orange'],
+    )
+
+    fig2.update_traces(marker=dict(size=20))  # Define o tamanho dos marcadores para o DataFrame 2
 
+    # Combine as duas figuras em uma única figura
+    for data in fig2.data:
+        fig1.add_trace(data)
+
+    # Personalize o layout do mapa, se desejar
+    fig1.update_layout(
+        mapbox_style="carto-positron",
+    )
+
+    # Mostrar o mapa
+    fig1.show()
+
+    return fig1
+
+        #-----------------#
 
 # Função de avaliação do imóvel
 def avaliacao_imovel(planilha, num_linhas_desejadas=10):
@@ -95,7 +138,7 @@ def avaliacao_imovel(planilha, num_linhas_desejadas=10):
     df_area_terreno['fat'] = round((df_area_terreno['razao']) ** (df_area_terreno['n']), 2)
     df_area_terreno = df_area_terreno[['fat']]
 
-    #-----------------#
+        #-----------------#
     
     # fator topografia (ftp)
     # dicionário topografia
@@ -228,7 +271,7 @@ def avaliacao_imovel(planilha, num_linhas_desejadas=10):
     result = pd.concat([df_dados, df_transp, df_area_const, df_area_terreno, df_topografia, df_idade_cons, df_padrao, df_vaga, df_exc], axis=1)
     result['Valor_desc'] = round(result['Valor']*(result['fof']), 2)
     result['Vunit'] = round((result['Valor_desc']/result['Área Construída']), 2)
-    result = result[['Atratividade local', 'Área Construída', 'Área Terreno', 'Topografia',
+    result = result[['lat','lon','Atratividade local', 'Área Construída', 'Área Terreno', 'Topografia',
            'Idade aparente e conservação', 'Padrão construtivo', 'Vagas',
            'Coeficiente extra', 'Valor', 'fof','Valor_desc', 'Vunit','fal', 'fac', 'fat', 'ftp', 'fic',
            'fpd', 'fvg', 'fex']]
@@ -501,6 +544,11 @@ def avaliacao_imovel(planilha, num_linhas_desejadas=10):
         result_valores.columns = ['Nome da Coluna', 'Valor']
         result_valores.to_excel(writer, sheet_name='valor', index=False)
         
+        #-----------------#
+       
+        mapa = plotar_mapa_com_dois_dataframes(result, df_avaliando)
+        #mapa = criar_mapa(df_avaliando) #novo
+        
         #-----------------#
     
         # Salve o DataFrame 'result' em uma planilha
@@ -515,7 +563,7 @@ def avaliacao_imovel(planilha, num_linhas_desejadas=10):
         #-----------------#
 
         # Retorna tanto a planilha quanto os resultados formatados
-        return output_file, 'resultados.pdf', result_render, resultados_formatados, intervalo_confiança, valores_finais
+        return output_file, 'resultados.pdf', result_render, resultados_formatados, intervalo_confiança, valores_finais, mapa #novo
         
     
 # Interface do Gradio com input como arquivo XLS ou XLSX
@@ -532,6 +580,7 @@ interface = gr.Interface(
         gr.components.Textbox(label="Resultados estatísticos"),
         gr.components.Textbox(label="Intervalo de confiança de 80%"),
         gr.components.Textbox(label="Valores Calculados"),
+        gr.Plot(label="Geolocalização da amostra"), #novo
     
     ],
     live=True,