Untitled

from graphviz import Digraph
import matplotlib.pyplot as plt
import pandas as pd
import numpy as np

# Create Network Diagram
network = Digraph('Network', node_attr={'shape': 'circle'})

network.node('A', 'Start')
network.node('B', 'Site Exploration')
network.node('C', 'Feasibility Study')
network.node('D', 'Construction')
network.node('E', 'Commissioning')
network.node('F', 'Project Closeout')
network.node('G', 'End')

network.edges(['A-B', 'B-C', 'C-D', 'D-E', 'E-F', 'F-G'])
network.render('network_diagram', format='png', cleanup=True)

# Create Gantt Chart Data
data = {
    'Task': ['Site Exploration', 'Feasibility Study', 'Construction', 'Commissioning', 'Project Closeout'],
    'Start': [0, 2, 5, 12, 20],
    'Duration': [2, 3, 7, 4, 4]
}
df = pd.DataFrame(data)

df['Finish'] = df['Start'] + df['Duration']
fig, ax = plt.subplots(figsize=(10, 5))
colors = plt.cm.Paired(np.linspace(0, 1, len(df)))

for i, task in enumerate(df['Task']):
    ax.barh(task, df['Duration'][i], left=df['Start'][i], color=colors[i])

ax.set_xlabel("Time (Months)")
ax.set_title("Gantt Chart for M3 Greenfield Mining Project")
plt.grid(axis='x', linestyle='--', alpha=0.7)
plt.savefig('gantt_chart.png')