Update app.py

app.py

@@ -3,135 +3,79 @@ import math
 import plotly.graph_objects as go
 import streamlit as st
 
-# THIS FUNCTION CREATES SHPERES
+# Function to create spheres
 def spheres(size, clr, dist=0): 
-    # Set up 100 points. First, do angles
-    theta = np.linspace(0,2*np.pi,100)
-    phi = np.linspace(0,np.pi,100)
-    # Set up coordinates for points on the sphere
-    x0 = dist + size * np.outer(np.cos(theta),np.sin(phi))
-    y0 = size * np.outer(np.sin(theta),np.sin(phi))
-    z0 = size * np.outer(np.ones(100),np.cos(phi))
-    # Set up trace
-    trace= go.Surface(x=x0, y=y0, z=z0, colorscale=[[0,clr], [1,clr]])
+    theta = np.linspace(0, 2*np.pi, 100)
+    phi = np.linspace(0, np.pi, 100)
+    x0 = dist + size * np.outer(np.cos(theta), np.sin(phi))
+    y0 = size * np.outer(np.sin(theta), np.sin(phi))
+    z0 = size * np.outer(np.ones(100), np.cos(phi))
+    trace = go.Surface(x=x0, y=y0, z=z0, colorscale=[[0, clr], [1, clr]])
     trace.update(showscale=False)
     return trace
 
-# THIS FUNCTION CREATES ORBITS
+# Function to create orbits
 def orbits(dist, offset=0, clr='white', wdth=2): 
-    # Initialize empty lists for eac set of coordinates
-    xcrd=[]
-    ycrd=[]
-    zcrd=[]
-    # Calculate coordinates
-    for i in range(0,361):
-        xcrd=xcrd+[(round(np.cos(math.radians(i)),5)) * dist + offset]
-        ycrd=ycrd+[(round(np.sin(math.radians(i)),5)) * dist]
-        zcrd=zcrd+[0]
-    trace = go.Scatter3d(x=xcrd, y=ycrd, z=zcrd, marker=dict(size=0.1), line=dict(color=clr,width=wdth))
+    xcrd = []
+    ycrd = []
+    zcrd = []
+    for i in range(0, 361):
+        xcrd.append((round(np.cos(math.radians(i)), 5)) * dist + offset)
+        ycrd.append((round(np.sin(math.radians(i)), 5)) * dist)
+        zcrd.append(0)
+    trace = go.Scatter3d(x=xcrd, y=ycrd, z=zcrd, marker=dict(size=0.1), line=dict(color=clr, width=wdth))
     return trace
 
-# THIS FUNCTION DEFINES ANNOTATIONS
+# Function to define annotations
 def annot(xcrd, zcrd, txt, xancr='center'):
-    strng=dict(showarrow=False, x=xcrd, 
-               y=0, z=zcrd, text=txt, 
-               xanchor=xancr, font=dict(color='#BEBEBE',size=12), opacity=0.5)
+    strng = dict(showarrow=False, x=xcrd, y=0, z=zcrd, text=txt, xanchor=xancr, font=dict(color='#BEBEBE', size=12))
     return strng
 
-
 # The Streamlit app
 def main():
     st.title('Solar System Visualization')
     st.write('##### Zoom in to realize how small and insignificant you are!')
 
-    # VISUALIZE THE SOLAR SYSTEM
-    diameter_km = [200000, 4878, 12104, 12756, 6787, 142796, 120660, 51118, 48600]
-    diameter = [((i / 12756) * 2) for i in diameter_km]
-    distance_from_sun = [0, 57.9, 108.2, 149.6, 227.9, 778.6, 1433.5, 2872.5, 4495.1]
-
-    # Set up orbit traces
-    traces_orbits = [
-        orbits(distance_from_sun[1]), # Mercury
-        orbits(distance_from_sun[2]), # Venus
-        orbits(distance_from_sun[3]), # Earth
-        orbits(distance_from_sun[4]), # Mars
-        orbits(distance_from_sun[5]), # Jupiter
-        orbits(distance_from_sun[6]), # Saturn
-        orbits(distance_from_sun[7]), # Uranus
-        orbits(distance_from_sun[8])  # Neptune
-    ]
+    # 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]
 
-    # Use the same to draw a few rings for Saturn
-    traces_rings_saturn = [
-        orbits(23, distance_from_sun[6], '#827962', 3),
-        orbits(24, distance_from_sun[6], '#827962', 3),
-        orbits(25, distance_from_sun[6], '#827962', 3),
-        orbits(26, distance_from_sun[6], '#827962', 3),
-        orbits(27, distance_from_sun[6], '#827962', 3),
-        orbits(28, distance_from_sun[6], '#827962', 3)
-    ]
+    # Scale factors for visualization
+    scale_factor_diameter = 1e-5
+    scale_factor_distance = 1e-6
 
-    layout=go.Layout(
-        title = 'Solar System',
+    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', 
-                       titlefont_color='black', 
-                       range=[-7000,7000], 
-                       backgroundcolor='black',
-                       color='black',
-                       gridcolor='black'),
-            yaxis=dict(title='Distance from the Sun',
-                       titlefont_color='black',
-                       range=[-7000,7000],
-                       backgroundcolor='black',
-                       color='black',
-                       gridcolor='black'
-                      ),
-            zaxis=dict(title='', 
-                       range=[-7000,7000],
-                       backgroundcolor='black',
-                       color='white', 
-                       gridcolor='black'
-                      ),
+        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[0], 40, 'Sun', xancr='left'),
-                annot(distance_from_sun[1], 5, 'Mercury'),
-                annot(distance_from_sun[2], 9, 'Venus'),
-                annot(distance_from_sun[3], 9, 'Earth'),
-                annot(distance_from_sun[4], 7, 'Mars'),
-                annot(distance_from_sun[5], 30, 'Jupiter'),
-                annot(distance_from_sun[6], 28, 'Saturn'),
-                annot(distance_from_sun[7], 20, 'Uranus'),
-                annot(distance_from_sun[8], 20, 'Neptune'),
+                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'),
             ]
         )
     )
 
     fig = go.Figure(layout=layout)
-    
+
     # Add spheres for the Sun and planets
-    fig.add_trace(spheres(diameter[0], '#ffff00', distance_from_sun[0])) # Sun
-    fig.add_trace(spheres(diameter[1], '#87877d', distance_from_sun[1])) # Mercury
-    fig.add_trace(spheres(diameter[2], '#d23100', distance_from_sun[2])) # Venus
-    fig.add_trace(spheres(diameter[3], '#325bff', distance_from_sun[3])) # Earth
-    fig.add_trace(spheres(diameter[4], '#b20000', distance_from_sun[4])) # Mars
-    fig.add_trace(spheres(diameter[5], '#ebebd2', distance_from_sun[5])) # Jupiter
-    fig.add_trace(spheres(diameter[6], '#ebcd82', distance_from_sun[6])) # Saturn
-    fig.add_trace(spheres(diameter[7], '#37ffda', distance_from_sun[7])) # Uranus
-    fig.add_trace(spheres(diameter[8], '#2500ab', distance_from_sun[8])) # Neptune
+    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))
 
     # 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:
-        fig.add_trace(trace)
+    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))
 
     st.plotly_chart(fig)
 
 if __name__ == "__main__":
-    main()
+    main()