Update app.py

app.py

@@ -1,36 +1,18 @@
 import subprocess
 import pkg_resources
 import sys
-import time
 
-def update_gradio():
-    """Atualiza o Gradio para a versão mais recente."""
+def ensure_latest_gradio():
+    """Verifica e atualiza o Gradio silenciosamente."""
     try:
         subprocess.check_call([sys.executable, '-m', 'pip', 'install', '--upgrade', 'gradio'])
         print("Gradio atualizado com sucesso!")
-        print("Reiniciando aplicação...")
-        time.sleep(2)
-        python = sys.executable
-        subprocess.Popen([python] + sys.argv)
-        sys.exit(0)
     except Exception as e:
         print(f"Erro ao atualizar Gradio: {e}")
 
-# Verifica e atualiza o Gradio se necessário
-try:
-    current_version = pkg_resources.get_distribution('gradio').version
-    latest_version = subprocess.check_output([sys.executable, '-m', 'pip', 'index', 'versions', 'gradio']).decode().split('\n')[0].split(' ')[-1]
-    
-    if current_version != latest_version:
-        print(f"Atualizando Gradio: {current_version} -> {latest_version}")
-        update_gradio()
-    else:
-        print(f"Gradio já está na versão mais recente: {current_version}")
-except:
-    print("Instalando Gradio...")
-    update_gradio()
-
-# Importações necessárias
+# Atualiza o Gradio uma única vez no início
+ensure_latest_gradio()
+
 import gradio as gr
 import rdflib
 import requests
@@ -40,17 +22,15 @@ from io import BytesIO
 import base64
 
 def load_names_from_url(jsonld_url):
-    """Carrega e extrai os nomes do arquivo JSON-LD a partir de uma URL."""
+    """Carrega e extrai os nomes do arquivo JSON-LD."""
     try:
         response = requests.get(jsonld_url)
         data = response.json()
-        
         names = []
         for item in data:
             if 'name' in item:
                 names.append(item['name'])
-        
-        return sorted(names)  # Retorna nomes ordenados alfabeticamente
+        return sorted(names)  # Retorna nomes ordenados
     except Exception as e:
         print(f"Erro ao carregar dados: {e}")
         return []
@@ -103,12 +83,12 @@ def run_query_and_visualize(selected_location, jsonld_url):
             return f'<div style="color: red; text-align: center; padding: 20px;">{G}</div>'
         
         plt.figure(figsize=(15, 10))
-        pos = nx.spring_layout(G)
+        pos = nx.spring_layout(G, k=1, iterations=50)
         
         nx.draw_networkx_nodes(G, pos, node_size=3000, node_color="skyblue", alpha=0.9)
         nx.draw_networkx_edges(G, pos, width=2, alpha=0.5, edge_color='gray')
-        nx.draw_networkx_labels(G, pos, labels=nx.get_node_attributes(G, 'label'), font_size=9, font_color="black")
-        nx.draw_networkx_edge_labels(G, pos, edge_labels=nx.get_edge_attributes(G, 'label'), font_size=9, font_color="red")
+        nx.draw_networkx_labels(G, pos, labels=nx.get_node_attributes(G, 'label'), font_size=9)
+        nx.draw_networkx_edge_labels(G, pos, edge_labels=nx.get_edge_attributes(G, 'label'), font_size=9)
 
         plt.title(f"Grafo para: {selected_location}", size=15)
         plt.axis('off')
@@ -130,31 +110,26 @@ jsonld_url = 'https://huggingface.co/spaces/histlearn/ShowGraph/raw/main/datafil
 names = load_names_from_url(jsonld_url)
 
 # Interface Gradio
-with gr.Blocks() as demo:
-    gr.Markdown("""
-    # Visualização de Grafos de Localidades
-    Selecione um local para visualizar seu grafo de relacionamentos.
-    """)
+with gr.Blocks(css="footer {display: none}") as demo:
+    gr.Markdown("# Visualização de Grafos de Localidades")
+    gr.Markdown("Selecione um local para visualizar seu grafo de relacionamentos e citações.")
 
     with gr.Column():
         selected_location = gr.Dropdown(
             choices=names,
             label="Selecione o Local",
-            info="Escolha um local para ver suas relações"
+            value=names[0] if names else None
         )
-        run_button = gr.Button("Visualizar Grafo", variant="primary")
+        view_button = gr.Button("Visualizar Grafo", variant="primary")
 
-    graph_output = gr.HTML(
-        value='<div style="text-align: center; padding: 20px;">Selecione um local e clique em "Visualizar Grafo"</div>'
-    )
-
-    def on_run_button_click(selected_location):
-        return run_query_and_visualize(selected_location, jsonld_url)
+        result_html = gr.HTML(
+            value='<div style="text-align: center; padding: 20px;">Selecione um local e clique em "Visualizar Grafo"</div>'
+        )
 
-    run_button.click(
-        fn=on_run_button_click,
-        inputs=[selected_location],
-        outputs=graph_output
+    view_button.click(
+        fn=run_query_and_visualize,
+        inputs=[selected_location, gr.Textbox(value=jsonld_url, visible=False)],
+        outputs=result_html
     )
 
 if __name__ == "__main__":