Update app.py

app.py

@@ -5,6 +5,7 @@ import time
 import plotly.express as px
 from groq import Groq
 from dotenv import load_dotenv
+import pycountry
 
 # Load environment variables
 load_dotenv()
@@ -107,8 +108,8 @@ st.markdown("""
     .country-btn {
         display: flex;
         align-items: center;
-        justify-content: center;
-        gap: 10px;
+        justify-content: flex-start;
+        gap: 12px;
         width: 100%;
         padding: 1.2rem;
         border-radius: 15px;
@@ -120,6 +121,7 @@ st.markdown("""
         border: 2px solid transparent;
         font-weight: 600;
         font-size: 1.1rem;
+        text-align: left;
     }
     
     .country-btn:hover {
@@ -133,6 +135,12 @@ st.markdown("""
         box-shadow: 0 8px 25px rgba(0,0,0,0.12);
     }
     
+    .country-flag {
+        font-size: 1.8rem;
+        min-width: 40px;
+        text-align: center;
+    }
+    
     .footer {
         text-align: center;
         padding: 2.5rem;
@@ -202,6 +210,11 @@ st.markdown("""
         .response-card {
             padding: 1.8rem;
         }
+        
+        .country-flag {
+            font-size: 1.5rem;
+            min-width: 35px;
+        }
     }
     
     @media (max-width: 576px) {
@@ -229,6 +242,11 @@ st.markdown("""
         .response-card {
             padding: 1.5rem;
         }
+        
+        .country-flag {
+            font-size: 1.3rem;
+            min-width: 30px;
+        }
     }
 </style>
 """, unsafe_allow_html=True)
@@ -236,7 +254,7 @@ st.markdown("""
 # Country data with flags
 COUNTRIES = {
     "🇺🇸 United States": "US",
-    "🇬🇧 United Kingdom": "UK",
+    "🇬🇧 United Kingdom": "GB",
     "🇨🇦 Canada": "CA",
     "🇦🇺 Australia": "AU",
     "🇮🇳 India": "IN",
@@ -276,11 +294,20 @@ MODELS = {
     "Mixtral-8x7b (Balanced)": "mixtral-8x7b-32768"
 }
 
+# Function to get country name from code
+def get_country_name(code):
+    try:
+        return pycountry.countries.get(alpha_2=code).name
+    except:
+        return code
+
 # Function to get rights information from Groq API
 def get_legal_rights(country, scenario, model_name):
     """Get legal rights information using Groq API"""
+    country_name = get_country_name(country)
+    
     system_prompt = f"""
-    You are an expert legal assistant specializing in {country} law. 
+    You are an expert legal assistant specializing in {country_name} law. 
     Provide clear, accurate information about an individual's rights in the given scenario.
     
     Guidelines:
@@ -297,9 +324,9 @@ def get_legal_rights(country, scenario, model_name):
     
     user_prompt = f"""
     Scenario: {scenario}
-    Country: {country}
+    Country: {country_name}
     
-    Please provide a comprehensive list of rights for this situation in {country}.
+    Please provide a comprehensive list of rights for this situation in {country_name}.
     """
     
     try:
@@ -350,7 +377,7 @@ def main():
     st.markdown("""
     <div class="header">
         <h1 style="margin:0;font-size:2.8rem;">Legal Rights Explorer Pro</h1>
-        <p style="margin:0;font-size:1.3rem;opacity:0.9;margin-top:12px;">Know your rights • Protect your freedom</p>
+        <p style="margin:0;font-size:1.3rem;opacity:0.9;margin-top:12px;">Know Your Rights, Anywhere. Anytime.</p>
     </div>
     """, unsafe_allow_html=True)
     
@@ -393,9 +420,10 @@ def main():
             for country_display in countries_col1:
                 is_selected = st.session_state.selected_country == country_display
                 btn_class = "country-btn selected" if is_selected else "country-btn"
+                flag, name = country_display.split(" ", 1)
                 
                 if st.button(
-                    country_display,
+                    f'<div class="country-flag">{flag}</div> {name}',
                     key=f"btn_{country_display}",
                     use_container_width=True
                 ):
@@ -408,9 +436,10 @@ def main():
             for country_display in countries_col2:
                 is_selected = st.session_state.selected_country == country_display
                 btn_class = "country-btn selected" if is_selected else "country-btn"
+                flag, name = country_display.split(" ", 1)
                 
                 if st.button(
-                    country_display,
+                    f'<div class="country-flag">{flag}</div> {name}',
                     key=f"btn_{country_display}",
                     use_container_width=True
                 ):
@@ -424,7 +453,7 @@ def main():
         awareness = [92, 89, 87, 85, 82, 88, 86, 84, 79, 81, 85, 83, 80, 78, 90, 85]  # Simulated data
         
         fig = px.bar(
-            x=countries,
+            x=[get_country_name(code) for code in countries],
             y=awareness,
             labels={'x': 'Country', 'y': 'Rights Awareness'},
             color=awareness,
@@ -482,7 +511,7 @@ def main():
         if st.session_state.selected_scenario:
             country_code = COUNTRIES[st.session_state.selected_country]
             
-            with st.spinner(f"��� Analyzing your rights for '{st.session_state.selected_scenario}' in {country_code}..."):
+            with st.spinner(f"🔍 Analyzing your rights for '{st.session_state.selected_scenario}' in {st.session_state.selected_country}..."):
                 response = get_legal_rights(
                     country_code, 
                     st.session_state.selected_scenario,