import numpy as np

import pandas as pd
import [Link] as plt

def showData():
df = pd.read_csv('States table [Link]')
print(df)
input("Press any key to continue... ")

def dataNoIndex():
df = pd.read_csv('States table [Link]', index_col=0)
print(df)
input("Press any key to continue... ")

def data_sorted():
df = pd.read_csv('States table [Link]')
print(df.sort_values(by=['Confirmed']))
input("Press any key to continue... ")

def write_data():
print("Insert data of particular states in list form:")
di = eval(input("Enter States (list): "))
con_cases = eval(input("Enter confirmed cases (list): "))
rec = eval(input("Enter recovered cases (list): "))
deaths = eval(input("Enter deaths (list): "))
active = eval(input("Enter active cases (list): "))

d={
'States': di,
'Confirmed': con_cases,
'Recovered': rec,
'Deaths': deaths,
'Active': active
}

df = [Link](d)
df.to_csv('States table [Link]', mode='a', index=False, header=False)
print("Data has been added.")
input("Press any key to continue... ")

def edit_data():
df = pd.read_csv('States table [Link]')
di = input("Enter state to edit: ")
 col = input("Enter column name to update: ")
val = input("Enter new value: ")

[Link][df['States'] == di, col] = val

def delete_data():
di = input("Enter state to delete: ")
df = pd.read_csv('States table [Link]')
df = df[df['States'] != di]
df.to_csv('States table [Link]', index=False)
print("Record deleted.")
input("Press any key to continue... ")

def line_chart():
df = pd.read_csv('States table [Link]')
States = df["States"]
Confirmed = df["Confirmed"]
Recovered = df["Recovered"]
Deaths = df["Deaths"]
Active = df["Active"]

Y=0
while Y != 6:
print("\n===============================")
print(" Line Graph Menu")
print("===============================")
print("1. State wise Confirmed Cases")
print("2. State wise Recovered Cases")
print("3. State wise Death Cases")
print("4. State wise Active Cases")
print("5. All data")
print("6. Return to main menu")
Y = int(input("Enter your choice to get line graph: "))

if Y == 1:
[Link]("Confirmed Cases")
[Link]("State Wise Confirmed Cases")
[Link](States, Confirmed, color='b')
[Link]()
elif Y == 2:
[Link]("Recovered Cases")
[Link]("State Wise Recovered Cases")
[Link](States, Recovered, color='g')
 [Link]()
elif Y == 3:
[Link]("Death Cases")
[Link]("State Wise Death Cases")
[Link](States, Deaths, color='r')
[Link]()
elif Y == 4:
[Link]("Active Cases")
[Link]("State Wise Active Cases")
[Link](States, Active, color='c')
[Link]()
elif Y == 5:
[Link]("Number of Cases")
[Link](States, Confirmed, color='b', label="Confirmed")
[Link](States, Recovered, color='g', label="Recovered")
[Link](States, Deaths, color='r', label="Deaths")
[Link](States, Active, color='c', label="Active")
[Link]()
[Link]("All Data Comparison")
[Link]()
elif Y == 6:
print("Line Graph Closed.")
else:
print("Invalid Option! Try Again!")

def bar_chart():
df = pd.read_csv('States table [Link]')
States = df["States"]
Confirmed = df["Confirmed"]
Recovered = df["Recovered"]
Deaths = df["Deaths"]
Active = df["Active"]

Y=0
while Y != 7:
print("\n============================")
print(" Bar Graph Menu")
print("============================")
print("1. Confirmed Cases")
print("2. Recovered Cases")
print("3. Death Cases")
print("4. Active Cases")
print("5. All data")
print("6. Combined Bar Graph")
print("7. Return to main menu")
Y = int(input("Enter your choice to get bar graph: "))
 if Y == 1:
[Link](States, Confirmed, color='b')
[Link]("State Wise Confirmed Cases")
[Link]()
elif Y == 2:
[Link](States, Recovered, color='g')
[Link]("State Wise Recovered Cases")
[Link]()
elif Y == 3:
[Link](States, Deaths, color='r')
[Link]("State Wise Death Cases")
[Link]()
elif Y == 4:
[Link](States, Active, color='c')
[Link]("State Wise Active Cases")
[Link]()
elif Y == 5:
[Link](States, Confirmed, color='b', label="Confirmed")
[Link](States, Recovered, color='g', label="Recovered")
[Link](States, Deaths, color='r', label="Deaths")
[Link](States, Active, color='c', label="Active")
[Link]()
[Link]("All Data Comparison")
[Link]()
elif Y == 6:
D = [Link](len(States))
width = 0.2
[Link](D, Confirmed, width, color='b', label="Confirmed")
[Link](D + 0.2, Recovered, width, color='g', label="Recovered")
[Link](D + 0.4, Deaths, width, color='r', label="Deaths")
[Link](D + 0.6, Active, width, color='c', label="Active")
[Link](D + 0.3, States)
[Link]()
[Link]("Combined Bar Graph")
[Link]()
elif Y == 7:
print("Bar Graph Closed.")
else:
print("Invalid Option! Try Again!")

def main_menu():
ch = 0
while ch != 9:
print("\n============================")
print(" Main Menu")
print("============================")
print("""
1. Show DataFrame
2. Data without index
3. Data sorted by Confirmed cases
4. Add state data
5. Edit a record
6. Delete a record
7. Line Graph
8. Bar Graph
9. Exit
""")
ch = int(input("Enter your choice: "))

if ch == 1:
showData()
elif ch == 2:
dataNoIndex()
elif ch == 3:
data_sorted()
elif ch == 4:
write_data()
elif ch == 5:
edit_data()
elif ch == 6:
delete_data()
elif ch == 7:
line_chart()
elif ch == 8:
bar_chart()
elif ch == 9:
print("Thank you for using the App! Goodbye!")
else:
print("Invalid choice! Try again.")

# Start the main menu