Update app.py

app.py

@@ -1,96 +1,254 @@
 import streamlit as st
 import os
-from scenario_advisor import get_scenario_based_response
+import google.generativeai as genai
 
-st.set_page_config(page_title="Samvidhan AI", page_icon="📜")
-
-# Get API key from Hugging Face Secrets
-try:
-    api_key = st.secrets["GOOGLE_API_KEY"]
-    st.success("✅ API Key loaded from secrets")
-except Exception as e:
-    st.error("❌ Could not load API key from secrets")
-    st.info("Please configure GOOGLE_API_KEY in Space Settings → Secrets")
-    api_key = None
+st.set_page_config(
+    page_title="Samvidhan AI",
+    page_icon="📜",
+    layout="wide"
+)
 
+# Custom CSS
 st.markdown("""
 <style>
-    .main-header { font-size: 2.5rem; color: #2E3A8C; }
-    .subheader { font-size: 1.5rem; color: #3949AB; }
-    .scenario-box { 
-        background-color: #E3F2FD; 
+    .main-header { 
+        font-size: 2.5rem; 
+        color: #2E3A8C; 
+        text-align: center;
+        margin-bottom: 0.5rem;
+    }
+    .subheader { 
+        font-size: 1.2rem; 
+        color: #3949AB; 
+        text-align: center;
+        margin-bottom: 2rem;
+    }
+    .analysis-box { 
+        background-color: #F8F9FA; 
         padding: 20px; 
         border-radius: 10px; 
+        border-left: 4px solid #2E3A8C;
         margin: 20px 0; 
+    }
+    .example-box {
+        background-color: #E3F2FD;
+        padding: 15px;
+        border-radius: 8px;
+        margin: 10px 0;
         border: 1px solid #BBDEFB;
-        color: #000000 !important;
+    }
+    .warning-box {
+        background-color: #FFF3E0;
+        padding: 15px;
+        border-radius: 8px;
+        border-left: 4px solid #FF9800;
+        margin: 20px 0;
     }
 </style>
 """, unsafe_allow_html=True)
 
-# For now, focus on scenario-based analysis since FAISS index setup is complex
-st.markdown('<p class="main-header">⚖️ Samvidhan AI - Constitutional Analysis</p>', unsafe_allow_html=True)
+# Header
+st.markdown('<p class="main-header">📜 Samvidhan AI</p>', unsafe_allow_html=True)
+st.markdown('<p class="subheader">Constitutional Law Assistant for India</p>', unsafe_allow_html=True)
+
+# Safe API key handling
+api_key = None
+try:
+    api_key = st.secrets.get("GOOGLE_API_KEY", None)
+    if api_key:
+        genai.configure(api_key=api_key)
+        # st.success("✅ API configured successfully")
+except Exception as e:
+    st.error(f"❌ Error accessing API configuration: {str(e)}")
+    st.info("Please ensure GOOGLE_API_KEY is set in Hugging Face Spaces Secrets (Settings → Secrets)")
+
+def get_constitutional_analysis(scenario):
+    """Generate constitutional analysis using Google Gemini"""
+    
+    if not api_key:
+        return "❌ API key not configured. Please contact the space administrator."
+    
+    try:
+        model = genai.GenerativeModel('gemini-1.5-flash')
+        
+        prompt = f"""
+        You are a Constitutional law expert specializing in the Constitution of India. 
+        Provide a comprehensive, factual, and unbiased analysis of the following scenario:
+
+        **Scenario:** {scenario}
 
-# API Key check
+        **Please provide a detailed analysis including:**
+
+        1. **Relevant Constitutional Articles**: List specific articles with exact numbers and brief descriptions
+        2. **Legal Framework**: Explain how these constitutional provisions apply to this scenario
+        3. **Possible Interpretations**: Discuss different legal perspectives and potential outcomes
+        4. **Judicial Precedents**: Mention any relevant Supreme Court or High Court cases if applicable
+        5. **Constitutional Balance**: Explain any competing rights or principles involved
+        6. **Practical Implications**: What this means in real-world legal terms
+        7. **Areas of Uncertainty**: Any ambiguous aspects or evolving jurisprudence
+
+        **Format your response with clear headings and use simple language that a non-lawyer can understand.**
+        **Be objective and present multiple viewpoints where applicable.**
+        """
+        
+        response = model.generate_content(prompt)
+        return response.text
+        
+    except Exception as e:
+        return f"❌ Error generating analysis: {str(e)}\n\nPlease check your API configuration or try again later."
+
+# Main application
 if not api_key:
-    st.error("🔐 Google API Key not configured. Please contact the administrator.")
+    st.error("🔐 **API Key Not Configured**")
+    st.markdown("""
+    <div class="warning-box">
+    <strong>Setup Required:</strong><br>
+    This application requires a Google API Key to function. Please contact the space administrator to configure the GOOGLE_API_KEY in the Hugging Face Spaces secrets.
+    </div>
+    """, unsafe_allow_html=True)
     st.stop()
 
-st.markdown("### Scenario-based Legal Analysis")
-st.markdown("Analyze legal scenarios based on the Indian Constitution")
+# Example scenarios
+st.markdown("## 💡 Example Constitutional Scenarios")
+st.markdown("Choose from these examples or write your own scenario:")
 
 examples = [
-    "A state government passes a law restricting online speech criticizing its ministers, citing maintenance of public order. Is this constitutional?",
-    "A private school refuses admission based on religion. What does the Constitution say?",
-    "Government wants to acquire private land for highway. What are the constitutional provisions?",
-    "A citizen wants to know their fundamental rights under Article 19.",
-    "Can a state government ban a particular religious practice in public places?"
+    {
+        "title": "Free Speech vs Public Order",
+        "scenario": "A state government passes a law restricting online speech criticizing its ministers, citing maintenance of public order. Citizens argue this violates their fundamental right to free speech. Is this constitutional under Indian law?"
+    },
+    {
+        "title": "Religious Freedom in Education",
+        "scenario": "A private Christian school refuses admission to a Muslim student, stating they only admit Christian students to maintain their religious character. The student's parents file a complaint claiming discrimination. What does the Constitution say about this?"
+    },
+    {
+        "title": "Reservation Policy Challenge",
+        "scenario": "A general category student with 95% marks is denied admission to a medical college, while an SC category student with 70% marks gets admission under reservation. The general category student challenges this as violating equality. Is the reservation policy constitutional?"
+    },
+    {
+        "title": "Property Rights & Land Acquisition",
+        "scenario": "The government wants to acquire private agricultural land for a highway project. The farmer refuses to sell, claiming it violates his property rights. The government proceeds with compulsory acquisition. What are the constitutional provisions here?"
+    },
+    {
+        "title": "Emergency Powers",
+        "scenario": "During a natural disaster, a state government declares emergency and suspends all fundamental rights for 30 days, including the right to move freely and conduct business. A business owner challenges this in court. Is such action constitutionally valid?"
+    },
+    {
+        "title": "Right to Privacy vs National Security",
+        "scenario": "The government introduces a law requiring all citizens to provide biometric data and link it to all their activities. Citizens challenge this citing right to privacy. How does the Constitution balance privacy rights with national security?"
+    }
 ]
 
-selected_example = st.selectbox("Choose an example or write your own:", ["Write your own..."] + examples)
+# Create expandable sections for examples
+with st.expander("📋 View Example Scenarios"):
+    for i, example in enumerate(examples, 1):
+        st.markdown(f"**{i}. {example['title']}**")
+        st.markdown(f"<div class='example-box'>{example['scenario']}</div>", unsafe_allow_html=True)
 
-if selected_example != "Write your own...":
-    scenario = st.text_area("Describe your legal scenario", value=selected_example, height=200)
+# Scenario selection
+selected_example = st.selectbox(
+    "Select an example or choose 'Custom' to write your own:",
+    ["Custom Scenario"] + [f"{example['title']}" for example in examples]
+)
+
+# Text area for scenario
+if selected_example == "Custom Scenario":
+    scenario = st.text_area(
+        "📝 Describe your constitutional law scenario:",
+        height=200,
+        placeholder="Example: A government employee wants to participate in a political protest during office hours. The administration prohibits this citing service rules. Can the employee claim this violates their fundamental rights?",
+        help="Describe a situation involving constitutional law, fundamental rights, government powers, or legal disputes."
+    )
 else:
-    scenario = st.text_area("Describe your legal scenario", height=200, 
-                           placeholder="Example: A government employee wants to know if they can participate in political protests...")
-
-if st.button("🔍 Analyze Scenario", type="primary"):
-    if scenario.strip():
-        with st.spinner("Analyzing constitutional implications..."):
-            try:
-                response = get_scenario_based_response(scenario)
-                st.markdown("### 📜 Constitutional Analysis:")
-                st.write(response)
-                
-            except Exception as e:
-                st.error(f"Error analyzing scenario: {str(e)}")
-                st.info("Please check if the API key is properly configured.")
+    # Find the selected example
+    selected_scenario = next((ex['scenario'] for ex in examples if ex['title'] == selected_example), "")
+    scenario = st.text_area(
+        "📝 Selected scenario (you can modify it):",
+        value=selected_scenario,
+        height=200
+    )
+
+# Analysis button
+col1, col2, col3 = st.columns([1, 2, 1])
+with col2:
+    analyze_button = st.button("🔍 **Analyze Constitutional Implications**", type="primary", use_container_width=True)
+
+# Generate analysis
+if analyze_button:
+    if scenario and scenario.strip():
+        with st.spinner("🔄 Analyzing constitutional aspects... This may take a moment."):
+            analysis = get_constitutional_analysis(scenario.strip())
+            
+            st.markdown("---")
+            st.markdown("## 📜 **Constitutional Analysis**")
+            st.markdown(f'<div class="analysis-box">{analysis}</div>', unsafe_allow_html=True)
+            
+            # Add download option
+            st.download_button(
+                label="📄 Download Analysis",
+                data=f"Constitutional Analysis\n\nScenario: {scenario}\n\nAnalysis:\n{analysis}",
+                file_name="constitutional_analysis.txt",
+                mime="text/plain"
+            )
     else:
         st.warning("⚠️ Please enter a scenario to analyze.")
 
 # Sidebar
 with st.sidebar:
-    st.markdown("### 📜 Samvidhan AI")
-    st.markdown("Constitutional AI Assistant for India")
+    st.markdown("### 📜 About Samvidhan AI")
+    st.info("""
+    **Samvidhan AI** is an AI-powered assistant that helps analyze constitutional law scenarios based on the Indian Constitution.
     
-    st.markdown("**Features:**")
-    st.markdown("- Scenario-based legal analysis")
-    st.markdown("- Constitutional article references")  
-    st.markdown("- Plain language explanations")
-    st.markdown("- Supreme Court precedents")
+    **Features:**
+    - Constitutional article references
+    - Legal precedent analysis
+    - Multiple perspective analysis
+    - Plain language explanations
+    """)
     
-    st.markdown("---")
-    st.markdown("**Disclaimer:**")
-    st.markdown("This tool provides educational information only. For legal advice, consult a qualified lawyer.")
+    st.markdown("### 🎯 How to Use")
+    st.markdown("""
+    1. **Choose** from example scenarios or write your own
+    2. **Click** the analyze button
+    3. **Review** the detailed constitutional analysis
+    4. **Download** the analysis if needed
+    """)
+    
+    st.markdown("### ⚖️ Legal Disclaimer")
+    st.warning("""
+    **Important:** This tool provides educational information only and should not be considered as legal advice. 
+    
+    For actual legal matters, please consult with a qualified lawyer or legal professional.
+    """)
+    
+    st.markdown("### 🔧 System Status")
+    if api_key:
+        if st.button("🧪 Test API Connection"):
+            with st.spinner("Testing..."):
+                try:
+                    test_model = genai.GenerativeModel('gemini-1.5-flash')
+                    test_response = test_model.generate_content("What is Article 21 of the Indian Constitution?")
+                    if test_response.text and len(test_response.text) > 50:
+                        st.success("✅ API connection successful!")
+                    else:
+                        st.warning("⚠️ API response seems incomplete")
+                except Exception as e:
+                    st.error(f"❌ API test failed: {str(e)}")
+    else:
+        st.error("❌ API not configured")
     
     st.markdown("---")
-    if st.button("🧪 Test API Connection"):
-        try:
-            test_response = get_scenario_based_response("Test: What is the Constitution of India?")
-            if test_response and len(test_response) > 50:
-                st.success("✅ API working correctly!")
-            else:
-                st.warning("⚠️ API response seems incomplete")
-        except Exception as e:
-            st.error(f"❌ API test failed: {str(e)}")
+    st.markdown("**Built with:** Streamlit + Google Gemini")
+    st.markdown("**Deployed on:** 🤗 Hugging Face Spaces")
+
+# Footer
+st.markdown("---")
+st.markdown(
+    """
+    <div style='text-align: center; color: #666;'>
+    <p>📜 Samvidhan AI - Making Constitutional Law Accessible</p>
+    <p>Built with ❤️ for constitutional awareness in India</p>
+    </div>
+    """,
+    unsafe_allow_html=True
+)