Update app.py

app.py

@@ -33,49 +33,97 @@ def annot(xcrd, zcrd, txt, xancr='center'):
 
 # The Streamlit app
 def main():
-    st.title('Solar System Visualization')
-    st.write('##### Zoom in to realize how small and insignificant you are!')
+   st.title('Solar System Visualization')
+   st.write('##### Zoom in to realize how small and insignificant you are!')
 
-    # Data for the sizes and distances of the planets (in km)
-    diameter_km = [1392684, 12742, 6780, 142984, 120536, 51118, 49528, 2370]
-    distance_from_sun_km = [0, 57909175, 108208930, 149597890, 227936640, 778412020, 1433449370, 2870671400]
+   # VISUALIZE THE SOLAR SYSTEM
+   # Diameters in km
+   diameter_km = [1392700, 4879, 12104, 12756, 6792, 142984, 120536, 51118, 49528]
+   # Distances from the Sun in millions of km
+   distance_from_sun = [0, 57.9, 108.2, 149.6, 227.9, 778.6, 1433.5, 2872.5, 4495.1]
 
-    # Scale factors for visualization
-    scale_factor_diameter = 1e-5
-    scale_factor_distance = 1e-6
+   # Set up orbit traces
+   traces_orbits = [
+       orbits(distance_from_sun[1] * 1000000, eccentricity=0.205, inclination=7.005),  # Mercury
+       orbits(distance_from_sun[2] * 1000000, eccentricity=0.007, inclination=3.394),  # Venus
+       orbits(distance_from_sun[3] * 1000000, eccentricity=0.017, inclination=0),      # Earth
+       orbits(distance_from_sun[4] * 1000000, eccentricity=0.093, inclination=1.850),  # Mars
+       orbits(distance_from_sun[5] * 1000000, eccentricity=0.048, inclination=1.308),  # Jupiter
+       orbits(distance_from_sun[6] * 1000000, eccentricity=0.056, inclination=2.488),  # Saturn
+       orbits(distance_from_sun[7] * 1000000, eccentricity=0.046, inclination=0.773),  # Uranus
+       orbits(distance_from_sun[8] * 1000000, eccentricity=0.009, inclination=1.774)   # Neptune
+   ]
 
-    layout = go.Layout(
-        title='Solar System',
-        showlegend=False,
-        margin=dict(l=0, r=0, t=0, b=0),
-        scene=dict(
-            xaxis=dict(title='Distance from the Sun (million km)', titlefont_color='black', range=[-500, 500]),
-            yaxis=dict(title='Distance from the Sun (million km)', titlefont_color='black', range=[-500, 500]),
-            zaxis=dict(title='', range=[-500, 500]),
-            annotations=[
-                annot(distance_from_sun_km[0] * scale_factor_distance, 40, 'Sun', xancr='left'),
-                annot(distance_from_sun_km[1] * scale_factor_distance, 5, 'Mercury'),
-                annot(distance_from_sun_km[2] * scale_factor_distance, 9, 'Venus'),
-                annot(distance_from_sun_km[3] * scale_factor_distance, 9, 'Earth'),
-                annot(distance_from_sun_km[4] * scale_factor_distance, 7, 'Mars'),
-                annot(distance_from_sun_km[5] * scale_factor_distance, 30, 'Jupiter'),
-                annot(distance_from_sun_km[6] * scale_factor_distance, 28, 'Saturn'),
-                annot(distance_from_sun_km[7] * scale_factor_distance, 20, 'Uranus'),
-            ]
-        )
-    )
+   # Use the same to draw a few rings for Saturn
+   traces_rings_saturn = [
+       orbits(23 * 120536 / 142984, distance_from_sun[6] * 1000000, inclination=2.488),
+       orbits(24 * 120536 / 142984, distance_from_sun[6] * 1000000, inclination=2.488),
+       orbits(25 * 120536 / 142984, distance_from_sun[6] * 1000000, inclination=2.488),
+       orbits(26 * 120536 / 142984, distance_from_sun[6] * 1000000, inclination=2.488),
+       orbits(27 * 120536 / 142984, distance_from_sun[6] * 1000000, inclination=2.488),
+       orbits(28 * 120536 / 142984, distance_from_sun[6] * 1000000, inclination=2.488)
+   ]
 
-    fig = go.Figure(layout=layout)
+   layout = go.Layout(
+       title='Solar System',
+       showlegend=False,
+       margin=dict(l=0, r=0, t=0, b=0),
+       scene=dict(
+           xaxis=dict(title='Distance from the Sun (km)',
+                      titlefont_color='white',
+                      range=[-7e9, 7e9],
+                      backgroundcolor='black',
+                      color='white',
+                      gridcolor='gray'),
+           yaxis=dict(title='Distance from the Sun (km)',
+                      titlefont_color='white',
+                      range=[-7e9, 7e9],
+                      backgroundcolor='black',
+                      color='white',
+                      gridcolor='gray'),
+           zaxis=dict(title='',
+                      range=[-7e9, 7e9],
+                      backgroundcolor='black',
+                      color='white',
+                      gridcolor='gray'),
+           annotations=[
+               annot(0, 40, 'Sun', xancr='left'),
+               annot(distance_from_sun[1] * 1000000, 5, 'Mercury'),
+               annot(distance_from_sun[2] * 1000000, 9, 'Venus'),
+               annot(distance_from_sun[3] * 1000000, 9, 'Earth'),
+               annot(distance_from_sun[4] * 1000000, 7, 'Mars'),
+               annot(distance_from_sun[5] * 1000000, 30, 'Jupiter'),
+               annot(distance_from_sun[6] * 1000000, 28, 'Saturn'),
+               annot(distance_from_sun[7] * 1000000, 20, 'Uranus'),
+               annot(distance_from_sun[8] * 1000000, 20, 'Neptune'),
+           ]
+       )
+   )
 
-    # Add spheres for the Sun and planets
-    for i in range(len(diameter_km)):
-        fig.add_trace(spheres(diameter_km[i] * scale_factor_diameter, '#ffff00', distance_from_sun_km[i] * scale_factor_distance))
+   fig = go.Figure(layout=layout)
 
-    # Add orbit traces
-    for i in range(1, len(distance_from_sun_km)):
-        fig.add_trace(orbits(distance_from_sun_km[i] * scale_factor_distance, clr='#BEBEBE', wdth=1))
+   # Add spheres for the Sun and planets
+   fig.add_trace(spheres(diameter_km[0] / 2, '#ffff00', 0))  # Sun
+   fig.add_trace(spheres(diameter_km[1] / 2, '#87877d', distance_from_sun[1] * 1000000))  # Mercury
+   fig.add_trace(spheres(diameter_km[2] / 2, '#d23100', distance_from_sun[2] * 1000000))  # Venus
+   fig.add_trace(spheres(diameter_km[3] / 2, '#325bff', distance_from_sun[3] * 1000000))  # Earth
+   fig.add_trace(spheres(diameter_km[4] / 2, '#b20000', distance_from_sun[4] * 1000000))  # Mars
+   fig.add_trace(spheres(diameter_km[5] / 2, '#ebebd2', distance_from_sun[5] * 1000000))  # Jupiter
+   fig.add_trace(spheres(diameter_km[6] / 2, '#ebcd82', distance_from_sun[6] * 1000000))  # Saturn
+   fig.add_trace(spheres(diameter_km[7] / 2, '#37ffda', distance_from_sun[7] * 1000000))  # Uranus
+   fig.add_trace(spheres(diameter_km[8] / 2, '#2500ab', distance_from_sun[8] * 1000000))  # Neptune
 
-    st.plotly_chart(fig)
+   # Add orbit traces
+   for trace in traces_orbits:
+       trace.line.color = '#808080'
+       fig.add_trace(trace)
+
+   # Add rings for Saturn
+   for trace in traces_rings_saturn:
+       trace.line.color = '#827962'
+       fig.add_trace(trace)
+
+   st.plotly_chart(fig)
 
 if __name__ == "__main__":
-    main()
+   main()