Update app.py

app.py

@@ -1,7 +1,70 @@
+import os
+from telethon import TelegramClient
+from telethon.errors import SessionPasswordNeededError
+import networkx as nx
+import matplotlib.pyplot as plt
 import gradio as gr
 
-def greet(name):
-    return f"Hello {name}! Welcome to Telegram Network Mapper."
+# Obtener credenciales de las variables de entorno
+API_ID = os.getenv("API_ID")
+API_HASH = os.getenv("API_HASH")
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
+# Configurar el cliente de Telegram
+client = TelegramClient('session_name', API_ID, API_HASH)
+
+async def fetch_messages(channel_username):
+    try:
+        await client.start()
+        messages = []
+        async for message in client.iter_messages(channel_username, limit=50):
+            messages.append({
+                'id': message.id,
+                'text': message.text[:100] + "..." if message.text else "No text",
+                'date': str(message.date),
+                'forwarded_from': message.forward.sender.username if message.forward else None
+            })
+        return messages
+    except Exception as e:
+        return f"Error fetching messages: {str(e)}"
+
+def build_network(messages):
+    G = nx.DiGraph()
+    for msg in messages:
+        origin = msg['forwarded_from']
+        destination = 'TargetChannel'  # Canal actual
+        if origin and destination:
+            if G.has_edge(origin, destination):
+                G[origin][destination]['weight'] += 1
+            else:
+                G.add_edge(origin, destination, weight=1)
+    return G
+
+def visualize_network(channel_username):
+    try:
+        import asyncio
+        messages = asyncio.run(fetch_messages(channel_username))
+        
+        if isinstance(messages, str):  # Si hay un error, devuelve el mensaje
+            return messages
+        
+        G = build_network(messages)
+        
+        plt.figure(figsize=(10, 8))
+        pos = nx.spring_layout(G)
+        nx.draw(G, pos, with_labels=True, node_size=3000, node_color='lightblue', font_size=10, font_weight='bold')
+        labels = nx.get_edge_attributes(G, 'weight')
+        nx.draw_networkx_edge_labels(G, pos, edge_labels=labels)
+        plt.title("Telegram Channel Network")
+        return plt.gcf()  # Devuelve la figura actual
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Interfaz de Gradio
+demo = gr.Interface(
+    fn=visualize_network,
+    inputs="text",
+    outputs="plot",
+    title="Telegram Network Mapper",
+    description="Enter a Telegram channel username to analyze its message forwarding network."
+)
 demo.launch()