Update app.py

app.py

@@ -25,7 +25,8 @@ async def fetch_messages(channel_username):
             })
         return messages
     except Exception as e:
-        return f"Error fetching messages: {str(e)}"
+        print(f"Error fetching messages: {str(e)}")
+        return []  # Devuelve una lista vacía en caso de error
 
 def build_network(messages):
     G = nx.DiGraph()
@@ -44,8 +45,11 @@ def visualize_network(channel_username):
         import asyncio
         messages = asyncio.run(fetch_messages(channel_username))
         
-        if isinstance(messages, str):  # Si hay un error, devuelve el mensaje
-            return messages
+        if not messages:  # Si no hay mensajes, devuelve un gráfico vacío
+            G = nx.DiGraph()
+            plt.figure(figsize=(10, 8))
+            plt.title("No data available")
+            return plt.gcf()
         
         G = build_network(messages)
         
@@ -57,7 +61,10 @@ def visualize_network(channel_username):
         plt.title("Telegram Channel Network")
         return plt.gcf()  # Devuelve la figura actual
     except Exception as e:
-        return f"Error: {str(e)}"
+        print(f"Error in visualize_network: {str(e)}")
+        plt.figure(figsize=(10, 8))
+        plt.title("An error occurred")
+        return plt.gcf()
 
 # Interfaz de Gradio
 demo = gr.Interface(