Update app.py

app.py

@@ -52,33 +52,27 @@ st.markdown('<p class="main-header">📜 Samvidhan AI</p>', unsafe_allow_html=Tr
 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:
+api_key = st.secrets.get("GOOGLE_API_KEY", os.getenv("GOOGLE_API_KEY", ""))
+
+if api_key:
+    try:
         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)")
+    except Exception as e:
+        st.error(f"❌ Error configuring API: {str(e)}")
+else:
+    st.warning("⚠️ API key not found. The analysis feature will be disabled.")
 
 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}
-
         **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
@@ -86,32 +80,13 @@ def get_constitutional_analysis(scenario):
         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("🔐 **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()
+        return f"❌ Error generating analysis: {str(e)}"
 
 # Example scenarios
-st.markdown("## 💡 Example Constitutional Scenarios")
-st.markdown("Choose from these examples or write your own scenario:")
-
 examples = [
     {
         "title": "Free Speech vs Public Order",
@@ -139,54 +114,35 @@ examples = [
     }
 ]
 
-# Create expandable sections for examples
+# Example section
+st.markdown("## 💡 Example Constitutional Scenarios")
 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)
 
-# 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]
+    ["Custom Scenario"] + [ex["title"] for ex 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."
-    )
+    scenario = st.text_area("📝 Describe your constitutional law scenario:", height=200)
 else:
-    # 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
-    )
+    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."):
+if st.button("🔍 **Analyze Constitutional Implications**", use_container_width=True):
+    if scenario.strip():
+        with st.spinner("🔄 Analyzing constitutional aspects..."):
             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}",
+                data=f"Scenario:\n{scenario}\n\nAnalysis:\n{analysis}",
                 file_name="constitutional_analysis.txt",
                 mime="text/plain"
             )
@@ -195,60 +151,25 @@ if analyze_button:
 
 # Sidebar
 with st.sidebar:
-    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.
-    
-    **Features:**
-    - Constitutional article references
-    - Legal precedent analysis
-    - Multiple perspective analysis
-    - Plain language explanations
-    """)
-    
-    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")
+    st.info("**Samvidhan AI** helps analyze constitutional law scenarios in India.")
     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)}")
+            try:
+                model = genai.GenerativeModel('gemini-1.5-flash')
+                resp = model.generate_content("What is Article 21 of the Indian Constitution?")
+                if resp.text and len(resp.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("---")
-    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
-)
+        st.error("❌ API not configured. Analysis disabled.")
+
+# Hugging Face auto-run support
+if __name__ == "__main__":
+    import sys
+    port = int(os.environ.get("PORT", 7860))
+    sys.argv = ["streamlit", "run", "app.py", "--server.port", str(port), "--server.address", "0.0.0.0"]
+    from streamlit.web import cli as stcli
+    sys.exit(stcli.main())