Update app.py

app.py

@@ -1,108 +1,141 @@
 # app.py
-import os
 import streamlit as st
-from groq import Groq
+import requests
 import time
 import plotly.express as px
+import os
 from dotenv import load_dotenv
 
 # Load environment variables
 load_dotenv()
 
-# Initialize Groq client
-client = Groq(api_key=os.getenv("GROQ_API_KEY"))
-
 # Set up Streamlit page
 st.set_page_config(
     page_title="Legal Rights Explorer",
     page_icon="⚖️",
     layout="wide",
-    initial_sidebar_state="expanded"
+    initial_sidebar_state="collapsed"
 )
 
-# Custom CSS for styling
+# Custom CSS for professional styling
 st.markdown("""
 <style>
     :root {
-        --primary: #4361ee;
-        --secondary: #3f37c9;
-        --accent: #4895ef;
-        --success: #4cc9f0;
-        --dark: #3a0ca3;
-        --light: #f8f9fa;
-        --text: #212529;
+        --primary: #2c3e50;
+        --secondary: #3498db;
+        --accent: #e74c3c;
+        --light: #ecf0f1;
+        --dark: #2c3e50;
+        --success: #27ae60;
+        --card-shadow: 0 6px 20px rgba(0,0,0,0.1);
     }
     
     .stApp {
-        background: linear-gradient(135deg, #f5f7fa 0%, #c3cfe2 100%);
-        color: var(--text);
+        background: linear-gradient(135deg, #f8f9fa 0%, #e9ecef 100%);
+        color: #333;
+        font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
     }
     
     .header {
-        background: linear-gradient(90deg, var(--dark) 0%, var(--primary) 100%);
+        background: linear-gradient(90deg, var(--primary) 0%, var(--secondary) 100%);
         color: white;
-        padding: 1.5rem;
+        padding: 1.8rem 2rem;
         border-radius: 0 0 20px 20px;
-        box-shadow: 0 4px 20px rgba(0,0,0,0.15);
-        margin-bottom: 2rem;
+        box-shadow: var(--card-shadow);
+        margin-bottom: 2.5rem;
     }
     
     .card {
         background: white;
         border-radius: 15px;
-        padding: 1.5rem;
-        box-shadow: 0 6px 15px rgba(0,0,0,0.1);
-        margin-bottom: 1.5rem;
-        transition: transform 0.3s ease;
+        padding: 1.8rem;
+        box-shadow: var(--card-shadow);
+        margin-bottom: 1.8rem;
+        transition: all 0.3s ease;
+        border-left: 4px solid var(--secondary);
     }
     
     .card:hover {
+        box-shadow: 0 12px 30px rgba(0,0,0,0.15);
         transform: translateY(-5px);
-        box-shadow: 0 10px 25px rgba(0,0,0,0.15);
     }
     
     .scenario-btn {
         width: 100%;
-        padding: 1rem;
-        background: linear-gradient(90deg, var(--accent) 0%, var(--success) 100%);
+        padding: 1.1rem;
+        background: linear-gradient(135deg, var(--secondary) 0%, #2980b9 100%);
         color: white;
         border: none;
         border-radius: 12px;
         font-weight: 600;
-        margin: 0.5rem 0;
+        margin: 0.7rem 0;
         cursor: pointer;
         transition: all 0.3s ease;
+        text-align: left;
+        padding-left: 20px;
+        font-size: 1.05rem;
     }
     
     .scenario-btn:hover {
-        background: linear-gradient(90deg, var(--secondary) 0%, var(--primary) 100%);
+        background: linear-gradient(135deg, var(--primary) 0%, #2c3e50 100%);
         transform: scale(1.02);
     }
     
     .response-card {
-        background: rgba(255, 255, 255, 0.95);
-        border-left: 5px solid var(--primary);
-        border-radius: 10px;
-        padding: 1.5rem;
-        margin-top: 1.5rem;
-        box-shadow: 0 4px 15px rgba(0,0,0,0.08);
+        background: white;
+        border-left: 5px solid var(--success);
+        border-radius: 12px;
+        padding: 2rem;
+        margin-top: 2rem;
+        box-shadow: var(--card-shadow);
+        animation: fadeIn 0.6s ease;
+    }
+    
+    .country-card {
+        display: flex;
+        align-items: center;
+        gap: 15px;
+        padding: 1.2rem;
+        border-radius: 12px;
+        background: white;
+        box-shadow: var(--card-shadow);
+        margin-bottom: 15px;
+        cursor: pointer;
+        transition: all 0.3s ease;
+    }
+    
+    .country-card:hover {
+        transform: translateX(5px);
+        box-shadow: 0 8px 25px rgba(0,0,0,0.15);
+    }
+    
+    .country-card.selected {
+        background: linear-gradient(135deg, var(--light) 0%, var(--secondary) 100%);
+        border-left: 4px solid var(--accent);
     }
     
     .footer {
         text-align: center;
-        padding: 1.5rem;
-        margin-top: 2rem;
-        color: var(--text);
-        font-size: 0.9rem;
+        padding: 2rem;
+        margin-top: 3rem;
+        color: var(--dark);
+        font-size: 0.95rem;
+        background: rgba(236, 240, 241, 0.7);
+        border-radius: 15px;
+    }
+    
+    @keyframes fadeIn {
+        from { opacity: 0; transform: translateY(20px); }
+        to { opacity: 1; transform: translateY(0); }
     }
     
     @media (max-width: 768px) {
         .header {
-            padding: 1rem;
+            padding: 1.3rem;
         }
         
         .card {
-            padding: 1rem;
+            padding: 1.3rem;
         }
     }
 </style>
@@ -112,16 +145,16 @@ st.markdown("""
 COUNTRIES = {
     "🇺🇸 United States": "US",
     "🇬🇧 United Kingdom": "UK",
-    "🇨🇦 Canada": "Canada",
-    "🇦🇺 Australia": "Australia",
-    "🇮🇳 India": "India",
-    "🇩🇪 Germany": "Germany",
-    "🇫🇷 France": "France",
-    "🇯🇵 Japan": "Japan",
-    "🇧🇷 Brazil": "Brazil",
-    "🇿🇦 South Africa": "South Africa",
-    "🇪🇸 Spain": "Spain",
-    "🇮🇹 Italy": "Italy"
+    "🇨🇦 Canada": "CA",
+    "🇦🇺 Australia": "AU",
+    "🇮🇳 India": "IN",
+    "🇩🇪 Germany": "DE",
+    "🇫🇷 France": "FR",
+    "🇯🇵 Japan": "JP",
+    "🇧🇷 Brazil": "BR",
+    "🇿🇦 South Africa": "ZA",
+    "🇪🇸 Spain": "ES",
+    "🇮🇹 Italy": "IT"
 }
 
 # Common legal scenarios
@@ -140,166 +173,211 @@ SCENARIOS = [
     "Immigration Rights"
 ]
 
-# LLM models available on Groq
-MODELS = {
-    "Llama3-70b-8192": "llama3-70b-8192",
-    "Llama3-8b-8192": "llama3-8b-8192",
-    "Mixtral-8x7b-32768": "mixtral-8x7b-32768",
-    "Gemma-7b-It": "gemma-7b-it"
-}
-
-# Function to get rights information from Groq API
-def get_legal_rights(country, scenario, model_name):
-    """Get legal rights information using Groq API"""
-    system_prompt = f"""
-    You are an expert legal assistant specializing in {country} law. 
-    Provide clear, accurate information about an individual's rights in the given scenario.
+# Simulated API function
+def get_legal_rights(country, scenario):
+    """Simulate API response with country-specific legal rights"""
+    # In a real implementation, this would call an external API
+    # For this demo, we'll return simulated responses
     
-    Guidelines:
-    - List 5-7 key rights as bullet points
-    - Use plain language understandable to non-lawyers
-    - Include relevant legal references when appropriate
-    - Highlight any critical actions the person should take
-    - Mention any country-specific variations
-    - Keep response under 300 words
-    - Format with emojis for readability
-    """
+    # Simulate API delay
+    time.sleep(1.5)
     
-    user_prompt = f"""
-    Scenario: {scenario}
-    Country: {country}
+    # Base rights that apply to most countries
+    base_rights = [
+        "Right to remain silent",
+        "Right to legal representation",
+        "Right to be treated with dignity and respect",
+        "Right to understand the charges against you",
+        "Right to a fair and impartial process"
+    ]
     
-    Please provide a comprehensive list of rights for this situation in {country}.
-    """
+    # Country-specific additions
+    country_specific = {
+        "US": [
+            "Right to refuse unwarranted searches (4th Amendment)",
+            "Right to due process (5th Amendment)",
+            "Right to a speedy and public trial (6th Amendment)"
+        ],
+        "UK": [
+            "Right to legal advice under the Police and Criminal Evidence Act",
+            "Right to have someone informed of your arrest",
+            "Protection under the Human Rights Act 1998"
+        ],
+        "CA": [
+            "Right to be informed of the reason for detention (Charter of Rights)",
+            "Right to retain and instruct counsel without delay",
+            "Right to habeas corpus"
+        ],
+        "AU": [
+            "Right to silence during police questioning",
+            "Right to communicate with a friend or relative",
+            "Protection under the Australian Human Rights Commission Act"
+        ],
+        "IN": [
+            "Right to free legal aid under Article 39A of the Constitution",
+            "Protection against self-incrimination (Article 20)",
+            "Right to be informed of grounds for arrest (Article 22)"
+        ]
+    }
     
-    try:
-        chat_completion = client.chat.completions.create(
-            messages=[
-                {
-                    "role": "system",
-                    "content": system_prompt
-                },
-                {
-                    "role": "user",
-                    "content": user_prompt
-                }
-            ],
-            model=model_name,
-            temperature=0.3,
-            max_tokens=1024,
-            top_p=1,
-            stream=False,
-            stop=None,
-        )
-        
-        return chat_completion.choices[0].message.content
+    # Scenario-specific additions
+    scenario_specific = {
+        "Traffic Stop by Police": [
+            "Right to see official identification",
+            "Right to know the reason for the stop",
+            "Right to refuse a search without probable cause",
+            "Right to record the interaction"
+        ],
+        "Workplace Discrimination": [
+            "Right to a discrimination-free workplace",
+            "Right to report without retaliation",
+            "Right to reasonable accommodations",
+            "Right to equal pay for equal work"
+        ],
+        "Medical Emergency Rights": [
+            "Right to emergency treatment regardless of insurance",
+            "Right to informed consent",
+            "Right to access medical records",
+            "Right to privacy of health information"
+        ],
+        "Arrest or Detention": [
+            "Right to be informed of charges",
+            "Right to contact your embassy if foreign national",
+            "Right to humane treatment",
+            "Right to challenge the lawfulness of detention"
+        ]
+    }
     
-    except Exception as e:
-        st.error(f"Error fetching data: {str(e)}")
-        return None
-
-# Function to display response with animation
-def display_response(response):
-    """Display the response with typing animation effect"""
-    message_placeholder = st.empty()
-    full_response = ""
+    # Combine rights based on country and scenario
+    rights = base_rights.copy()
+    
+    if country in country_specific:
+        rights.extend(country_specific[country])
+    
+    for key, value in scenario_specific.items():
+        if key in scenario:
+            rights.extend(value)
+            break
     
-    # Simulate stream of response with milliseconds delay
-    for chunk in response.split():
-        full_response += chunk + " "
-        time.sleep(0.05)
-        # Add a blinking cursor to simulate typing
-        message_placeholder.markdown(f'<div class="response-card">{full_response}▌</div>', unsafe_allow_html=True)
+    # Format the response
+    response = f"## 🔍 Your Rights in {country}\n"
+    response += f"### Situation: {scenario}\n\n"
+    response += "Based on the laws of your selected country, you have the following rights:\n\n"
     
-    # Display final message without the cursor
-    message_placeholder.markdown(f'<div class="response-card">{response}</div>', unsafe_allow_html=True)
+    for i, right in enumerate(set(rights), 1):
+        response += f"✅ **{right}**\n\n"
+    
+    response += "\n**Important Notes:**\n"
+    response += "- This information is for educational purposes only\n"
+    response += "- Laws vary by jurisdiction and circumstances\n"
+    response += "- Consult a qualified attorney for legal advice\n"
+    
+    return response
 
 # Main app
 def main():
     # Header section
     st.markdown("""
     <div class="header">
-        <h1>Legal Rights Explorer</h1>
-        <p>Know your rights in any situation, anywhere in the world</p>
+        <h1 style="margin:0;font-size:2.3rem;">Legal Rights Explorer</h1>
+        <p style="margin:0;font-size:1.2rem;opacity:0.9;">Know your rights in any situation, anywhere in the world</p>
     </div>
     """, unsafe_allow_html=True)
     
-    # Sidebar for settings
-    with st.sidebar:
-        st.markdown("## ⚙️ Settings")
-        selected_model = st.selectbox("Select AI Model", list(MODELS.keys()), index=0)
-        
-        st.markdown("---")
-        st.markdown("### About")
-        st.markdown("This app helps you understand your legal rights in various situations across different countries using AI technology.")
-        st.markdown("Powered by Groq API and open-source LLMs.")
-        
-        st.markdown("---")
-        st.markdown("### How to Use")
-        st.markdown("1. Select your country 🇬🇧")
-        st.markdown("2. Choose a legal scenario 🚔")
-        st.markdown("3. Get instant rights information ⚖️")
-        
-        st.markdown("---")
-        st.markdown("Built with ❤️ for legal empowerment")
+    # Initialize session state
+    if 'selected_country' not in st.session_state:
+        st.session_state.selected_country = "🇺🇸 United States"
+    
+    if 'selected_scenario' not in st.session_state:
+        st.session_state.selected_scenario = None
     
     # Main content columns
-    col1, col2 = st.columns([1, 2])
+    col1, col2 = st.columns([1, 2], gap="large")
     
     with col1:
         st.markdown("### 🌍 Select Your Country")
-        country_display = st.selectbox("Choose country", list(COUNTRIES.keys()), index=0)
-        country_code = COUNTRIES[country_display]
+        st.markdown("Choose your country to see jurisdiction-specific rights information")
         
-        st.markdown("### 🚨 Choose a Scenario")
-        
-        # Create buttons for each scenario
-        for scenario in SCENARIOS:
-            if st.button(scenario, key=scenario, use_container_width=True, 
-                         help=f"Click to see your rights for {scenario}"):
-                st.session_state.selected_scenario = scenario
-        
-        # Custom scenario input
-        custom_scenario = st.text_input("Or enter your own scenario:")
-        if custom_scenario:
-            st.session_state.selected_scenario = custom_scenario
+        # Country selection cards
+        for country_display in COUNTRIES.keys():
+            is_selected = st.session_state.selected_country == country_display
+            card_class = "country-card selected" if is_selected else "country-card"
+            
+            st.markdown(f"""
+            <div class="{card_class}" onclick="selectCountry('{country_display}')">
+                <span style="font-size:1.8rem;">{country_display.split()[0]}</span>
+                <span style="font-weight:500;">{country_display.split(' ', 1)[1]}</span>
+            </div>
+            """, unsafe_allow_html=True)
         
-        # Add some visualizations
-        st.markdown("### 📊 Global Rights Awareness")
+        # Country visualization
+        st.markdown("### 📊 Global Rights Index")
         countries = list(COUNTRIES.values())
-        awareness = [80, 75, 70, 65, 60, 55, 50, 45, 40, 35, 30, 25]  # Simulated data
+        awareness = [92, 89, 87, 85, 82, 88, 86, 84, 79, 81, 85, 83]  # Simulated data
         
         fig = px.bar(
             x=countries,
             y=awareness,
-            labels={'x': 'Country', 'y': 'Awareness Level'},
+            labels={'x': 'Country', 'y': 'Rights Awareness'},
             color=awareness,
-            color_continuous_scale='Blues'
+            color_continuous_scale='Blues',
+            height=300
         )
         fig.update_layout(
-            title='Rights Awareness by Country',
             plot_bgcolor='rgba(0,0,0,0)',
-            paper_bgcolor='rgba(0,0,0,0)'
+            paper_bgcolor='rgba(0,0,0,0)',
+            margin=dict(l=0, r=0, t=30, b=0)
         )
         st.plotly_chart(fig, use_container_width=True)
+        
+        # Legal resources
+        with st.expander("📚 Legal Resources"):
+            st.markdown("""
+            - [United Nations Human Rights](https://www.ohchr.org/)
+            - [International Justice Resource Center](https://ijrcenter.org/)
+            - [Global Legal Information Network](https://www.loc.gov/law/help/legal-info.php)
+            - [World Legal Information Institute](https://www.worldlii.org/)
+            """)
     
     with col2:
+        st.markdown("### 🚨 Select a Legal Scenario")
+        st.markdown("Choose a situation to understand your rights and protections")
+        
+        # Create buttons for each scenario
+        cols = st.columns(2)
+        for i, scenario in enumerate(SCENARIOS):
+            with cols[i % 2]:
+                if st.button(
+                    f"**{scenario}**",
+                    key=scenario,
+                    use_container_width=True,
+                    help=f"Click to see your rights for {scenario}"
+                ):
+                    st.session_state.selected_scenario = scenario
+        
+        # Custom scenario input
+        custom_scenario = st.text_input("**Or describe your specific situation:**", placeholder="e.g., 'Rights during home search'")
+        if custom_scenario:
+            st.session_state.selected_scenario = custom_scenario
+        
+        # Response area
         st.markdown("### ⚖️ Your Legal Rights")
         
-        if 'selected_scenario' in st.session_state and st.session_state.selected_scenario:
-            with st.spinner(f"Getting your rights for {st.session_state.selected_scenario} in {country_code}..."):
+        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}..."):
                 response = get_legal_rights(
                     country_code, 
-                    st.session_state.selected_scenario, 
-                    MODELS[selected_model]
+                    st.session_state.selected_scenario
                 )
                 
                 if response:
-                    display_response(response)
+                    st.markdown(f'<div class="response-card">{response}</div>', unsafe_allow_html=True)
                 else:
                     st.error("Failed to get response. Please try again.")
         else:
-            st.info("Please select a legal scenario from the left panel to see your rights.")
+            st.info("Please select a legal scenario to see your rights information")
             st.image("https://images.unsplash.com/photo-1589391886645-d51941baf7fb?auto=format&fit=crop&w=600&h=400", 
                      caption="Know Your Rights, Exercise Your Freedom")
             
@@ -309,21 +387,40 @@ def main():
                 <p>Understanding your legal rights empowers you to:</p>
                 <ul>
                     <li>Protect yourself in difficult situations</li>
-                    <li>Make informed decisions</li>
-                    <li>Prevent exploitation</li>
-                    <li>Access justice when needed</li>
-                    <li>Confidently navigate legal systems</li>
+                    <li>Make informed decisions when facing legal issues</li>
+                    <li>Prevent exploitation and rights violations</li>
+                    <li>Access justice when your rights are infringed</li>
+                    <li>Confidently navigate complex legal systems</li>
                 </ul>
+                <p style="margin-top:15px;font-style:italic;">"Knowledge of the law is the first defense against injustice."</p>
             </div>
             """, unsafe_allow_html=True)
     
     # Footer
     st.markdown("""
     <div class="footer">
-        <p>Legal Rights Explorer | For Educational Purposes Only | Not Legal Advice</p>
-        <p>Always consult a qualified attorney for your specific legal situation</p>
+        <p>Legal Rights Explorer | © 2023 | For Educational Purposes Only | Not Legal Advice</p>
+        <p>Always consult a qualified attorney in your jurisdiction for your specific legal situation</p>
     </div>
     """, unsafe_allow_html=True)
+    
+    # JavaScript for country selection
+    st.markdown("""
+    <script>
+    function selectCountry(country) {
+        window.parent.document.querySelectorAll("button[kind='secondary']")[0].click();
+        setTimeout(function() {
+            const countryButtons = window.parent.document.querySelectorAll("button[kind='secondary']");
+            for (let btn of countryButtons) {
+                if (btn.innerText.includes(country)) {
+                    btn.click();
+                    break;
+                }
+            }
+        }, 100);
+    }
+    </script>
+    """, unsafe_allow_html=True)
 
 if __name__ == "__main__":
     main()