Update app.py

app.py

@@ -3,6 +3,7 @@ import json
 from datetime import datetime, timedelta
 import gradio as gr
 import pandas as pd
+import traceback
 import plotly.express as px
 import plotly.graph_objects as go
 
@@ -16,60 +17,72 @@ class NasaSsdCneosApi:
         self.ca_url = "https://ssd-api.jpl.nasa.gov/cad.api"
 
     def get_fireballs(self, limit=10, date_min=None, energy_min=None):
-        params = {'limit': limit}
-        if date_min:
-            params['date-min'] = date_min
-        if energy_min:
-            params['energy-min'] = energy_min
+        try:
+            params = {'limit': limit}
+            if date_min:
+                params['date-min'] = date_min
+            if energy_min:
+                params['energy-min'] = energy_min
 
-        response = requests.get(self.fireball_url, params=params)
-        if response.status_code == 200:
+            response = requests.get(self.fireball_url, params=params)
+            response.raise_for_status()
             return response.json()
-        return None
+        except Exception as e:
+            print("Fireball API Error:", e)
+            traceback.print_exc()
+            return None
 
     def get_close_approaches(self, dist_max=None, date_min=None, date_max=None,
                               h_min=None, h_max=None, v_inf_min=None, v_inf_max=None,
                               limit=10):
-        params = {'limit': limit, 'dist-max': dist_max, 'date-min': date_min,
-                  'date-max': date_max, 'h-min': h_min, 'h-max': h_max,
-                  'v-inf-min': v_inf_min, 'v-inf-max': v_inf_max, 'sort': 'date'}
-        # Remove None values
-        params = {k: v for k, v in params.items() if v is not None}
+        try:
+            params = {'limit': limit, 'dist-max': dist_max, 'date-min': date_min,
+                      'date-max': date_max, 'h-min': h_min, 'h-max': h_max,
+                      'v-inf-min': v_inf_min, 'v-inf-max': v_inf_max, 'sort': 'date'}
+            params = {k: v for k, v in params.items() if v is not None}
 
-        response = requests.get(self.ca_url, params=params)
-        if response.status_code == 200:
+            response = requests.get(self.ca_url, params=params)
+            response.raise_for_status()
             return response.json()
-        return None
+        except Exception as e:
+            print("Close Approaches API Error:", e)
+            traceback.print_exc()
+            return None
 
     def format_response(self, data, format_type):
-        if not data:
-            return None
+        try:
+            if not data:
+                return None
 
-        fields = data.get('fields')
-        rows = data.get('data')
+            fields = data.get('fields')
+            rows = data.get('data')
 
-        if not fields or not rows:
-            return None
+            if not fields or not rows:
+                return None
 
-        df = pd.DataFrame([dict(zip(fields, row)) for row in rows])
+            df = pd.DataFrame([dict(zip(fields, row)) for row in rows])
 
-        if format_type == 'fireballs':
-            return df.rename(columns={
-                'date': 'Date/Time', 'energy': 'Energy (kt)',
-                'impact-e': 'Impact Energy (10^10 J)', 'lat': 'Latitude',
-                'lon': 'Longitude', 'alt': 'Altitude (km)',
-                'vel': 'Velocity (km/s)'
-            })
+            if format_type == 'fireballs':
+                return df.rename(columns={
+                    'date': 'Date/Time', 'energy': 'Energy (kt)',
+                    'impact-e': 'Impact Energy (10^10 J)', 'lat': 'Latitude',
+                    'lon': 'Longitude', 'alt': 'Altitude (km)',
+                    'vel': 'Velocity (km/s)'
+                })
 
-        elif format_type == 'close_approaches':
-            return df.rename(columns={
-                'des': 'Object', 'orbit_id': 'Orbit ID', 'cd': 'Time (TDB)',
-                'dist': 'Nominal Distance (au)', 'dist_min': 'Minimum Distance (au)',
-                'dist_max': 'Maximum Distance (au)', 'v_rel': 'Velocity (km/s)',
-                'h': 'H (mag)'
-            })
+            elif format_type == 'close_approaches':
+                return df.rename(columns={
+                    'des': 'Object', 'orbit_id': 'Orbit ID', 'cd': 'Time (TDB)',
+                    'dist': 'Nominal Distance (au)', 'dist_min': 'Minimum Distance (au)',
+                    'dist_max': 'Maximum Distance (au)', 'v_rel': 'Velocity (km/s)',
+                    'h': 'H (mag)'
+                })
 
-        return df
+            return df
+        except Exception as e:
+            print("Data formatting error:", e)
+            traceback.print_exc()
+            return None