import os import re import streamlit as st import google.generativeai as genai # Backend / Model Logic GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY") if not GOOGLE_API_KEY: st.error("Google API Key not found. Add it in Hugging Face → Settings → Secrets.") st.stop() genai.configure(api_key=GOOGLE_API_KEY) _model = genai.GenerativeModel("gemini-2.5-flash-lite") st.set_page_config( page_title="AI Lawyer — Indian Law Assistant", page_icon="⚖️", layout="wide", initial_sidebar_state="expanded", ) # ╔══════════════════════════════════════════════════════════════════╗ # ║ BACKEND / MODEL LOGIC ║ # ╚══════════════════════════════════════════════════════════════════╝ SYSTEM_PROMPT = """You are an expert AI lawyer specializing in Indian law. You have deep knowledge of the Indian Penal Code (IPC), Code of Criminal Procedure (CrPC), Constitution of India, Civil Procedure Code (CPC), and all major Indian statutes. Rules: - Answer clearly and precisely with relevant section numbers where applicable. - If a question is outside Indian law, politely redirect. - Structure long answers with headings when needed. - Always end with a brief disclaimer: "This is for informational purposes only and not a substitute for professional legal advice." """ def markdown_to_html(text: str) -> str: """Convert markdown response to styled HTML for chat bubble rendering.""" lines = text.split("\n") html = [] in_ul = False for line in lines: # ### Heading 3 if line.startswith("### "): if in_ul: html.append(""); in_ul = False heading = re.sub(r"\*\*(.+?)\*\*", r"\1", line[4:].strip()) html.append(f'

{heading}

') # ## Heading 2 elif line.startswith("## "): if in_ul: html.append(""); in_ul = False heading = re.sub(r"\*\*(.+?)\*\*", r"\1", line[3:].strip()) html.append(f'

{heading}

') # # Heading 1 elif line.startswith("# "): if in_ul: html.append(""); in_ul = False heading = re.sub(r"\*\*(.+?)\*\*", r"\1", line[2:].strip()) html.append(f'

{heading}

') # Bullet points * or - elif re.match(r"^\s*[\*\-]\s+", line): if not in_ul: html.append("

{item}

"); in_ul = False html.append("

") # Normal paragraph else: if in_ul: html.append(""); in_ul = False line = re.sub(r"\*\*(.+?)\*\*", r"\1", line) line = re.sub(r"\*(.+?)\*", r"\1", line) html.append(f"

{line}

") if in_ul: html.append("") return "".join(html) def ask_llm(chat_history: list) -> str: """Send the full conversation history to Gemini and return the reply.""" messages = [] for msg in chat_history: role = "user" if msg["role"] == "user" else "model" messages.append({"role": role, "parts": [msg["content"]]}) # Prepend system instruction as first user/model exchange full_messages = [ {"role": "user", "parts": [SYSTEM_PROMPT]}, {"role": "model", "parts": ["Understood. I am your AI Lawyer specializing in Indian law. How can I assist you today?"]}, ] + messages try: response = _model.generate_content(full_messages) return response.text except Exception as e: err = str(e) if "429" in err or "quota" in err.lower() or "ResourceExhausted" in err: return "⚠️ **API Quota Exceeded** — The Gemini free-tier limit has been reached. Please wait a few minutes and try again." return f"⚠️ **An error occurred:** {err}" # ╔══════════════════════════════════════════════════════════════════╗ # ║ UI / STREAMLIT ║ # ╚══════════════════════════════════════════════════════════════════╝ # ── Session State ───────────────────────────────────────────────── if "chat_history" not in st.session_state: st.session_state.chat_history = [] # ── CSS ─────────────────────────────────────────────────────────── st.markdown(""" """, unsafe_allow_html=True) # ── SIDEBAR ─────────────────────────────────────────────────────── with st.sidebar: st.markdown('', unsafe_allow_html=True) st.markdown('

Indian Law Assistant

', unsafe_allow_html=True) st.markdown('

About

', unsafe_allow_html=True) st.markdown("""

Ask any question related to Indian law — IPC, CrPC, Constitution, CPC, and more.

The AI will provide precise answers with relevant section references.

""", unsafe_allow_html=True) st.markdown('

', unsafe_allow_html=True) st.markdown('

Topics Covered

', unsafe_allow_html=True) topics = ["⚖️ Indian Penal Code", "📜 Constitution of India", "🔍 CrPC", "📋 Civil Procedure Code", "🏛️ Contract Act", "👨‍👩‍👧 Family Law", "🏘️ Property Law", "💼 Labour Law"] for t in topics: st.markdown(f'

{t}

', unsafe_allow_html=True) st.markdown('

', unsafe_allow_html=True) if st.session_state.chat_history: msg_count = len([m for m in st.session_state.chat_history if m["role"] == "user"]) st.markdown(f'

💬 {msg_count} question{"s" if msg_count!=1 else ""} asked

', unsafe_allow_html=True) st.markdown("") if st.button("🗑 Clear Chat", use_container_width=True): st.session_state.chat_history = [] st.rerun() st.markdown("""

Designed & Developed by
Yedeedya Injeti
Innomatics Research Labs

""", unsafe_allow_html=True) # ── HERO ────────────────────────────────────────────────────────── st.markdown("""

AI Lawyer

Your intelligent legal assistant for Indian law • IPC • CrPC • Constitution & more

""", unsafe_allow_html=True) # ── QUICK SUGGESTIONS (shown only when chat is empty) ───────────── suggestions = [ "Punishment for murder under IPC?", "What is Article 21 of the Constitution?", "Define culpable homicide.", "What are bailable offences?", "Right to self-defence in India?", "What is the limitation period for filing a civil suit?", ] if not st.session_state.chat_history: st.markdown('

Suggested Questions

', unsafe_allow_html=True) col1, col2, col3 = st.columns(3) chosen = None for i, sug in enumerate(suggestions): col = [col1, col2, col3][i % 3] with col: if st.button(sug, key=f"sug_{i}", use_container_width=True): chosen = sug st.markdown("") # ── CHAT WINDOW ─────────────────────────────────────────────────── if not st.session_state.chat_history: st.markdown("""

⚖️

How can I assist you today?

Ask me anything about Indian law — I'll provide accurate answers
with relevant section references and case principles.

""", unsafe_allow_html=True) else: bubbles_html = '

' for msg in st.session_state.chat_history: if msg["role"] == "user": bubbles_html += f"""

{msg["content"]}

""" else: # ✅ Use markdown_to_html instead of plain .replace("\n", "
") content = markdown_to_html(msg["content"]) bubbles_html += f"""

⚖️

{content}

""" bubbles_html += "

" st.markdown(bubbles_html, unsafe_allow_html=True) # ── INPUT BAR ───────────────────────────────────────────────────── col_inp, col_send = st.columns([6, 1]) with col_inp: question = st.text_input( "input", label_visibility="collapsed", placeholder="Ask your legal question…", key="chat_input", ) with col_send: send_btn = st.button("Send ⚖️", use_container_width=True) # ── PROCESS ─────────────────────────────────────────────────────── final_q = None if "chosen" in dir() and chosen: final_q = chosen elif send_btn and question.strip(): final_q = question.strip() if final_q: st.session_state.chat_history.append({"role": "user", "content": final_q}) with st.spinner(""): answer = ask_llm(st.session_state.chat_history) st.session_state.chat_history.append({"role": "assistant", "content": answer}) st.rerun() # ── FOOTER ──────────────────────────────────────────────────────── st.markdown(""" """, unsafe_allow_html=True)