Update app.py

app.py

@@ -1,8 +1,8 @@
-# zodiac_app.py (Streamlit version)
 import streamlit as st
 import pandas as pd
 import datetime
 import matplotlib.pyplot as plt
+import numpy as np
 import io
 
 # Data: zodiac signs with date ranges and personality traits
@@ -56,8 +56,37 @@ def get_zodiac_sign(month: int, day: int) -> str:
                 return sign
     return None
 
+# Function to create radar chart
+def create_radar_chart(traits, values, sign):
+    num_vars = len(traits)
+
+    # Compute the angle for each trait (360 degrees divided by number of traits)
+    angles = np.linspace(0, 2 * np.pi, num_vars, endpoint=False).tolist()
+
+    # The plot needs to be circular, so we repeat the first value to close the loop
+    values += values[:1]
+    angles += angles[:1]
+
+    # Create the radar chart
+    fig, ax = plt.subplots(figsize=(6, 6), subplot_kw=dict(polar=True))
+
+    # Plot the data
+    ax.fill(angles, values, color='blue', alpha=0.25)
+    ax.plot(angles, values, color='blue', linewidth=2)  # Outline of the chart
+
+    # Set the labels for each axis (the traits)
+    ax.set_yticklabels([])  # Remove the radial ticks
+    ax.set_xticks(angles[:-1])  # Set the ticks for each trait
+    ax.set_xticklabels(traits, color='black', fontsize=12)
+
+    # Add the title
+    ax.set_title(f'{sign} Personality Traits', size=14, color='black', fontweight='bold')
+
+    # Show the radar chart
+    return fig
+
 # Streamlit UI
-st.title('Zodiac App')
+st.title('Zodiac Matcher')
 
 # Horoscope Section
 st.header('Get Your Horoscope and Personality Traits')
@@ -71,6 +100,15 @@ if birthdate:
     st.subheader(f"Your zodiac sign is: {sign}")
     st.write(f"Personality Traits: {', '.join(traits)}")
 
+    # Display Radar Chart for Zodiac Sign
+    trait_values = [9, 10, 8, 9]  # Example trait values for the radar chart (can be adjusted)
+    fig = create_radar_chart(traits, trait_values, sign)
+    
+    buf = io.BytesIO()
+    fig.savefig(buf, format='png')
+    buf.seek(0)
+    st.image(buf)
+
 # Compatibility Section
 st.header('Check Compatibility Between Two Zodiac Signs')