Update app.py

app.py

@@ -3,130 +3,134 @@ import math
 import plotly.graph_objects as go
 import streamlit as st
 
-# Function to create spheres
+# THIS FUNCTION CREATES SHPERES
 def spheres(size, clr, dist=0): 
-    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]])
+    # 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]])
     trace.update(showscale=False)
     return trace
 
-# Function to create orbits
+# THIS FUNCTION CREATES ORBITS
 def orbits(dist, offset=0, clr='white', wdth=2): 
-    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))
+    # 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))
     return trace
 
-# Function to define annotations
+# THIS FUNCTION DEFINES 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))
+    strng=dict(showarrow=False, x=xcrd, 
+               y=0, z=zcrd, text=txt, 
+               xanchor=xancr, font=dict(color='#BEBEBE',size=12), opacity=0.5)
     return strng
 
 # 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!')
 
-   # 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]
-   # Eccentricity and inclination values (in degrees) for each planet's orbit
-   eccentricities = [0, 0.205, 0.007, 0.017, 0.093, 0.048, 0.056, 0.046, 0.009]
-   inclinations = [0, 7.005, 3.394, 0, 1.850, 1.308, 2.488, 0.773, 1.774]
+    # VISUALIZE THE SOLAR SYSTEM
+    diameter_km = [1391016, 4879, 12104, 12756, 6792, 142984, 120536, 51118, 49528]
+    diameter = [((i / 142984) * 20) for i in diameter_km]  # Scale diameters
+    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] * 1000000, eccentricities[1], math.radians(inclinations[1])),  # Mercury
-       orbits(distance_from_sun[2] * 1000000, eccentricities[2], math.radians(inclinations[2])),  # Venus
-       orbits(distance_from_sun[3] * 1000000, eccentricities[3], math.radians(inclinations[3])),  # Earth
-       orbits(distance_from_sun[4] * 1000000, eccentricities[4], math.radians(inclinations[4])),  # Mars
-       orbits(distance_from_sun[5] * 1000000, eccentricities[5], math.radians(inclinations[5])),  # Jupiter
-       orbits(distance_from_sun[6] * 1000000, eccentricities[6], math.radians(inclinations[6])),  # Saturn
-       orbits(distance_from_sun[7] * 1000000, eccentricities[7], math.radians(inclinations[7])),  # Uranus
-       orbits(distance_from_sun[8] * 1000000, eccentricities[8], math.radians(inclinations[8]))   # Neptune
-   ]
+    # 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
+    ]
 
-   # Use the same to draw a few rings for Saturn
-   traces_rings_saturn = [
-       orbits(23 * 120536 / 142984, 0, math.radians(2.488)),
-       orbits(24 * 120536 / 142984, 0, math.radians(2.488)),
-       orbits(25 * 120536 / 142984, 0, math.radians(2.488)),
-       orbits(26 * 120536 / 142984, 0, math.radians(2.488)),
-       orbits(27 * 120536 / 142984, 0, math.radians(2.488)),
-       orbits(28 * 120536 / 142984, 0, math.radians(2.488))
-   ]
+    # Use the same to draw a few rings for Saturn
+    traces_rings_saturn = [
+        orbits(60, distance_from_sun[6], '#827962', 3),
+        orbits(70, distance_from_sun[6], '#827962', 3),
+        orbits(80, distance_from_sun[6], '#827962', 3),
+        orbits(90, distance_from_sun[6], '#827962', 3),
+        orbits(100, distance_from_sun[6], '#827962', 3),
+        orbits(110, distance_from_sun[6], '#827962', 3)
+    ]
 
-   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'),
-           ]
-       )
-   )
+    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'
+                      ),
+            annotations=[
+                annot(distance_from_sun[0], 200, 'Sun', xancr='left'),
+                annot(distance_from_sun[1], 2, 'Mercury'),
+                annot(distance_from_sun[2], 6, 'Venus'),
+                annot(distance_from_sun[3], 6, 'Earth'),
+                annot(distance_from_sun[4], 3, 'Mars'),
+                annot(distance_from_sun[5], 70, 'Jupiter'),
+                annot(distance_from_sun[6], 60, 'Saturn'),
+                annot(distance_from_sun[7], 25, 'Uranus'),
+                annot(distance_from_sun[8], 25, 'Neptune'),
+            ]
+        )
+    )
 
-   fig = go.Figure(layout=layout)
+    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
 
-   # Add spheres for the Sun and planets
-   fig.add_trace(go.Scatter3d(x=[0], y=[0], z=[0], mode='markers', marker=dict(size=diameter_km[0] / 2, color='#ffff00')))  # Sun
-   fig.add_trace(go.Scatter3d(x=[distance_from_sun[1] * 1000000], y=[0], z=[0], mode='markers', marker=dict(size=diameter_km[1] / 2, color='#87877d')))  # Mercury
-   fig.add_trace(go.Scatter3d(x=[distance_from_sun[2] * 1000000], y=[0], z=[0], mode='markers', marker=dict(size=diameter_km[2] / 2, color='#d23100')))  # Venus
-   fig.add_trace(go.Scatter3d(x=[distance_from_sun[3] * 1000000], y=[0], z=[0], mode='markers', marker=dict(size=diameter_km[3] / 2, color='#325bff')))  # Earth
-   fig.add_trace(go.Scatter3d(x=[distance_from_sun[4] * 1000000], y=[0], z=[0], mode='markers', marker=dict(size=diameter_km[4] / 2, color='#b20000')))  # Mars
-   fig.add_trace(go.Scatter3d(x=[distance_from_sun[5] * 1000000], y=[0], z=[0], mode='markers', marker=dict(size=diameter_km[5] / 2, color='#ebebd2')))  # Jupiter
-   fig.add_trace(go.Scatter3d(x=[distance_from_sun[6] * 1000000], y=[0], z=[0], mode='markers', marker=dict(size=diameter_km[6] / 2, color='#ebcd82')))  # Saturn
-   fig.add_trace(go.Scatter3d(x=[distance_from_sun[7] * 1000000], y=[0], z=[0], mode='markers', marker=dict(size=diameter_km[7] / 2, color='#37ffda')))  # Uranus
-   fig.add_trace(go.Scatter3d(x=[distance_from_sun[8] * 1000000], y=[0], z=[0], mode='markers', marker=dict(size=diameter_km[8] / 2, color='#2500ab')))  # Neptune
+    # Add orbit traces
+    for trace in traces_orbits:
+        trace.line.color = '#808080'
+        fig.add_trace(trace)
 
-   # 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)
 
-   # Add rings for Saturn
-   for trace in traces_rings_saturn:
-       trace.line.color = '#827962'
-       fig.add_trace(trace)
-
-   st.plotly_chart(fig)
+    st.plotly_chart(fig)
 
 if __name__ == "__main__":
-   main()
+    main()