Notes

import pandas as pd
import math

# Load data from CSV
d = pd.read_csv(r"C:Userskuldeep_aryaDownloadsextract_response (2).csv")

# Ensure the CSV has the 'response' column
if 'response' not in d.columns:
raise ValueError("The 'response' column is missing from the CSV file.")

# Convert the 'response' column to a list of dictionaries
data = d['response'].dropna().apply(pd.eval).tolist()

# Static edges to be added to the script
static_edges = [
('student_section', 'student_section_assignment', 'student_section_id', 'student_section_id'),
('student_section_assignment', 'student_section', 'student_section_id', 'student_section_id'),
('student_section', 'student_section_discussion_topic', 'student_section_id', 'student_section_id'),
('student_section_discussion_topic', 'student_section', 'student_section_id', 'student_section_id'),
('course', 'section', 'course_id', 'course_id')
]

def create_generate_edge_script(data, output_script):
with open(output_script, 'w') as script_file:
# Writing import statements and setup code for the script
script_file.write('import networkx as nxn')
script_file.write('G = nx.DiGraph()n')

# Add static edges
for source, destination, source_on, destination_on in static_edges:
script_file.write(f'G.add_edge("{source}", "{destination}", source_on="{source_on}", destination_on="{destination_on}")n')

# Write dynamic edges from filtered CSV data
for row in data:
# Skip rows that are not dictionaries or contain NaN values
if not isinstance(row, dict):
continue

# Extract values and handle NaN
source = row.get('source')
destination = row.get('destination')
source_on = row.get('source_on')
destination_on = row.get('destination_on')

# Skip if any of the required fields is NaN
if any(pd.isna(x) for x in [source, destination, source_on, destination_on]):
continue

# Ensure all values are strings
if all(isinstance(x, str) for x in [source, destination, source_on, destination_on]):
script_file.write(f'G.add_edge("{source}", "{destination}", source_on="{source_on}", destination_on="{destination_on}")n')

print(f"Script '{output_script}' generated successfully.")

# Generate the script
script_filename = 'generate_edge.py'
create_generate_edge_script(data, script_filename)