# DO NOT REMOVE OR MODIFY
# Import the library that lets you connect Colab with Google Drive
from google.colab import drive
drive.mount('/content/drive')

# ------------------------------------------------------------
# Get the email of the user running the notebook
# and create a folder:
# MyDrive/Taming Data Dragons/
# ------------------------------------------------------------
import os
import subprocess
from google.colab import auth

# Base path in Google Drive
user_path = "/content/drive/MyDrive/Taming Data Dragons/"

# Create the directory if it doesn't exist
os.makedirs(user_path, exist_ok=True)

print(f"User folder ready at: {user_path}")

"""
Small-World Network Demo (Watts–Strogatz Model)

This cell:
1. Creates a small-world network using NetworkX
2. Plots it at double size and 300 DPI
3. Saves the figure to Google Drive in:
   MyDrive/Taming Data Dragons/small_world_network.png
"""

import os
import networkx as nx
import matplotlib.pyplot as plt

# ------------------------------------------------------------
# Output path in Google Drive
# ------------------------------------------------------------
os.makedirs(user_path, exist_ok=True)
output_path = os.path.join(user_path, "small_world_network.png")

# ------------------------------------------------------------
# Parameters for the Watts–Strogatz small-world model
# ------------------------------------------------------------
n = 30   # Number of nodes
k = 4    # Each node connected to k nearest neighbors
p = 0.1  # Rewiring probability

# ------------------------------------------------------------
# Create the small-world network
# ------------------------------------------------------------
G = nx.watts_strogatz_graph(n=n, k=k, p=p)

# ------------------------------------------------------------
# Circular layout for visualization
# ------------------------------------------------------------
pos = nx.circular_layout(G)

# ------------------------------------------------------------
# Plot: double size + 300 DPI
# Original was ~8x8, so we go 16x16
# ------------------------------------------------------------
plt.figure(figsize=(8, 8), dpi=100)

nx.draw(
    G,
    pos,
    with_labels=False,
    node_size=300,
    alpha=0.9
)

plt.title("Small-World Network (Watts–Strogatz Model)")

# ------------------------------------------------------------
# Save to Google Drive
# ------------------------------------------------------------
plt.savefig(output_path, dpi=300, bbox_inches="tight")
plt.show()

print(f"Saved figure to: {output_path}")