Report main

import pandas as pd
from matplotlib import pyplot as plt
from flagship_phones import FlagshipPhone, PhoneCollection

phone_collection = PhoneCollection() # Kreiranje instance PhoneCollection

s24_ultra = FlagshipPhone('Samsung', 'S24 Ultra', 1450, 200, 15)
phone_collection.add_phone(s24_ultra)

honor_magic = FlagshipPhone('Honor', 'Magic V3', 1050, 50, 7)
phone_collection.add_phone(honor_magic)

iphone_16 = FlagshipPhone('Iphone', '16Pro Max', 1200, 48, 3)
phone_collection.add_phone(iphone_16)

huawei_mate = FlagshipPhone('Huawei', 'Mate70 Pro', 1080, 60, 12)
phone_collection.add_phone(huawei_mate)

xiaomi_14 = FlagshipPhone('Xiaomi', '14Ultra', 1000, 50, 9)
phone_collection.add_phone(xiaomi_14)

motorola_edge = FlagshipPhone('Motorola', 'Edge50 Ultra', 1300, 50, 17)
phone_collection.add_phone(motorola_edge)

google_pixel = FlagshipPhone('Google', 'Pixel9 Pro XL', 1050, 50, 5)
phone_collection.add_phone(google_pixel)

z_fold6 = FlagshipPhone('Samsung', 'Z Fold6', 1900, 50, 4)
phone_collection.add_phone(z_fold6)

iphone_15 = FlagshipPhone('Iphone', '15 Pro', 1050, 50, 5)
phone_collection.add_phone(iphone_15)

# Pretvaranje u DataFrame
df = phone_collection.to_dataframe()

# Sacuvati DataFrame u CSV
df.to_csv('Flagship.csv', index=False)

# Priprema podataka za pie chart

sizes = df['Sales']
labels = df['Model']

# Kreiranje pie chart-a

fig = plt.figure(figsize=(15, 10))
plt.pie(sizes, labels=labels, autopct='%1.1f%%', startangle=140)

price_info = [f"{label}: {df['Price'][i]} $" for i, label in enumerate(labels)] # Dodavanje legende
plt.legend(price_info, title="Price ($)", loc="upper right")

plt.title('Distribution of Sales')
plt.axis('equal')  # Održava kružni oblik
plt.show()

# Ispis DataFrame-a u konzoli

print(df)