asteroid factor

app.py

@@ -1,29 +1,37 @@
+# -*- coding: utf-8 -*-
+ 
 import gradio as gr
-import orjson
+import json
 import os
 import re
 import requests
 
-NASA_API_KEY = os.getenv("NASA_API_KEY") 
 
-def calc_asteroid_factor(json, weight=1.0):
+
+def calc_asteroid_factor(data, weight=1.0):
     """
-    Calculates the astoriod doom probability.
+    Calculates the asteroid doom probability.
 
     Args:
-        json (object): The data structure returned frm NASA
+        data (object): The data structure returned from NASA
         weight (float): The weight to apply to the hazard factor.
     
     Returns:
         Weighted factor for Near Earth Object impact for the next seven days.    
     """
-    # vibe coded, replace with real code later
-    matches = re.findall(r'"is_potentially_hazardous_asteroid"\s*:\s*(true|false)', json) 
-    hazard_list = list(filter(lambda x: x == "true", matches))
-    return (len(hazard_list)/len(matches) * weight) # Calculate the hazard factor as a ratio of hazardous to total asteroids
 
+    objects = [obj for sublist in data["near_earth_objects"].values() for obj in sublist] 
+
+    num_objects = len(objects)
+
+    hazardous_objects = [obj for sublist in data["near_earth_objects"].values() for obj in sublist if obj["is_potentially_hazardous_asteroid"] == True]
+
+    num_hazardous_objects = len(hazardous_objects)
+
+    return (num_hazardous_objects/num_objects * weight) # hazard factor 
 
-def fetch_nasa():
+
+def fetch_asteroid_data(url, api_key):
     """
     Fetches data from the NASA Near Earth Object Web Service.
     
@@ -31,22 +39,18 @@ def fetch_nasa():
         Near Earth Objects for the next seven days.    
     """
 
-    url = "https://api.nasa.gov/neo/rest/v1/feed"
-
-    rawdata = requests.get(
+    request_data = requests.get(
         url,
-        params={"api_key": NASA_API_KEY},  
+        params={"api_key": api_key},  
     )
 
-    if rawdata.status_code != 200:
-        return "Error fetching data from NASA API."
+    if request_data.status_code != 200:    
+        return None
+    else:    
+        return json.loads(request_data.content)
     
-    json = orjson.loads(rawdata.content)
-
-    return json
 
-
-def calc_factor():
+def calc_doom_probability():
     """
     Calculates the overall doom probability.
     
@@ -54,7 +58,17 @@ def calc_factor():
        The overall doom probability for the next seven days.    
     """
 
-    return calc_asteroid_factor(fetch_nasa(), weight=1.0)
+    nasa_url = "https://api.nasa.gov/neo/rest/v1/feed"
+    nasa_api_key = os.getenv("NASA_API_KEY") 
+
+    if not nasa_api_key:
+        return "ERROR: NASA_API_KEY not set."
+    else: 
+        nasa_data = fetch_asteroid_data(nasa_url, nasa_api_key)
+        if not nasa_data:
+            return "ERROR: Unable to fetch data from NASA API."
+        else:
+            return calc_asteroid_factor(nasa_data, weight=1.0)
         
-demo = gr.Interface(fn=calc_factor, inputs=None, outputs="text")
+demo = gr.Interface(fn=calc_doom_probability, inputs=None, outputs="text")
 demo.launch(mcp_server=True)

requirements.txt

@@ -1 +0,0 @@
-orjson