fix: changes

models/continent.py

@@ -2,4 +2,21 @@ import pandas as pd
 
 def read_country_data(path_world):
     """Reads CSV data into a Pandas DataFrame"""
-    return pd.read_csv(path_world)
+    return pd.read_csv(path_world)
+
+def get_continents(data_frame):
+    continents = data_frame["Continent"].unique()
+    return continents
+
+def get_continent_stats(data_frame, attribute, stat_type):
+    continents_stats = data_frame.groupby("Continent")[attribute].agg(total_attribute = 'sum')
+    if stat_type == "highest":
+        continent = continents_stats.idxmax().item()
+        value = continents_stats.max().item()
+        return {continent: value}
+    elif stat_type == "lowest":
+        continent = continents_stats.idxmin().item()
+        value = continents_stats.min().item()
+        return {continent: value}
+
+    

models/country.py

@@ -0,0 +1,27 @@
+import pandas as pd
+def get_country(data_frame):
+    country = data_frame["Country"].values
+    return country
+
+def get_country_stats(data_frame, attribute, stat_type):
+    if stat_type == "highest":
+        idx = data_frame[attribute].idxmax()
+    if stat_type == "lowest":
+        idx = data_frame[attribute].idxmin()
+
+    country= data_frame["Country"][idx]
+    attribute_result = data_frame[attribute][idx]
+    result = {country: attribute_result.item()}
+    return result
+
+
+    
+#     continents_stats = data_frame.[attribute].gg(total_attribute = 'sum')
+#     if stat_type == "highest":
+#         continent = continents_stats.idxmax().item()
+#         value = continents_stats.max().item()
+#         return {continent: value}
+#     elif stat_type == "lowest":
+#         continent = continents_stats.idxmin().item()
+#         value = continents_stats.min().item()
+#         return {continent: value}

routes/population_routes.py

@@ -1,8 +1,10 @@
 from fastapi import APIRouter, HTTPException
 import logging
 import pandas as pd
-from models.continent import read_country_data
+from models.continent import read_country_data, get_continents, get_continent_stats
 from models.validation_population import ContinentPopulationStats
+from models.country import get_country, get_country_stats
+
 
 router = APIRouter()
 
@@ -13,37 +15,30 @@ DATA_PATH = "data/world_population.csv"
 countries_df = read_country_data(DATA_PATH)
 logger.info("✅ Data loaded successfully from CSV.")
 
-def get_continent_countries(continent_name: str):
-    filtered_countries = countries_df[countries_df["Continent"].str.lower() == continent_name.lower()]
-    if filtered_countries.empty:
-        logger.warning(f"⚠️ Continent '{continent_name}' not found.")
-        raise HTTPException(status_code=404, detail=f"Continent '{continent_name}' not found.")
-    return filtered_countries
 
 @router.get("/")
 def home():
-    return {"message": "Welcome to the World Population API! Use /continent/{continent_name} to get population statistics."}
-
-
-@router.get("/continent/{continent_name}", response_model=ContinentPopulationStats)
-def get_continent_population_data(continent_name: str):
-    try:
-        continent_countries = get_continent_countries(continent_name)
-
-        max_pop_country = continent_countries.loc[continent_countries["Population"].idxmax()]
-        min_pop_country = continent_countries.loc[continent_countries["Population"].idxmin()]
-        avg_population = continent_countries["Population"].mean()
-        avg_pop_country = continent_countries.iloc[(continent_countries["Population"] - avg_population).abs().idxmin()]
-
-        return {
-            "continent": continent_name,
-            "max_population": int(max_pop_country["Population"]),
-            "min_population": int(min_pop_country["Population"]),
-            "avg_population": round(avg_population, 2),
-            "max_population_country": max_pop_country["Country"],
-            "min_population_country": min_pop_country["Country"],
-            "avg_population_country": avg_pop_country["Country"]
-        }
-    except Exception as e:
-        logger.error(f"❌ Error: {str(e)}")
-        raise HTTPException(status_code=500, detail="Internal Server Error")
+    return {"message": f"Welcome to the World Population"}
+
+
+@router.get("/continents")
+def display_continents():
+    result = get_continents(countries_df)
+    return result.tolist()
+@router.get("/country")
+def display_country():
+    res_country =get_country(countries_df)
+    return res_country.tolist()
+
+@router.get("/continent_stats/{attribute}/{stat_type}")
+def display_stats(attribute :str, stat_type: str):
+    continent_stars = get_continent_stats(countries_df, attribute, stat_type )
+    return continent_stars
+
+
+@router.get("/country_stats/{attribute}/{stat_type}")
+def display_stats(attribute :str, stat_type: str):
+    country_stars = get_country_stats(countries_df, attribute, stat_type )
+    return country_stars
+
+