Create app.py

app.py

@@ -0,0 +1,102 @@
+import os
+import gradio as gr
+from openai import OpenAI
+from add_embeddings import LegalDocumentProcessor
+import json
+from typing import Dict, List
+import markdown
+
+# Initialize clients and models
+client = OpenAI(
+    api_key=os.environ.get("MISTRAL_API_KEY"),
+    base_url="https://api.mistral.ai/v1"
+)
+
+# Initialize document processor
+doc_processor = LegalDocumentProcessor()
+
+def get_mistral_response(query: str, context: str) -> str:
+    """Get response from Mistral AI"""
+    system_prompt = """You are a helpful legal assistant for the Indian justice system. 
+    Your role is to assist the public in understanding and navigating the new criminal laws (BNS, BNSS, and BSA).
+    When providing information:
+    1. Be precise and cite specific sections when possible
+    2. Explain legal concepts in simple terms
+    3. Suggest relevant documents or next steps
+    4. For case-specific queries, provide general guidance while noting you cannot provide legal advice
+    5. If asked about document drafting, explain the required documents and their purpose
+    
+    Base your responses on the provided context and maintain a professional yet approachable tone."""
+    
+    try:
+        response = client.chat.completions.create(
+            model="mistral-medium",
+            messages=[
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": f"Context: {context}\n\nQuestion: {query}"}
+            ]
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+def format_sources(metadatas: List[Dict]) -> str:
+    """Format source information for display"""
+    sources = []
+    for metadata in metadatas:
+        law_code = metadata["law_code"]
+        source = metadata["source"]
+        sources.append(f"- {law_code} ({source})")
+    return "\n".join(sources)
+
+def chat_interface(message: str, history: List[List[str]]) -> str:
+    """Main chat interface function"""
+    # Search for relevant context
+    results = doc_processor.search_documents(message)
+    context = "\n".join(results["documents"])
+    
+    # Get response from Mistral
+    response = get_mistral_response(message, context)
+    
+    # Add source information
+    sources = format_sources(results["metadatas"])
+    full_response = f"{response}\n\n**Sources:**\n{sources}"
+    
+    # Format response with markdown
+    formatted_response = markdown.markdown(full_response)
+    
+    return formatted_response
+
+# Initialize ChromaDB with legal documents
+doc_processor.process_and_store_documents()
+
+# Create Gradio interface
+iface = gr.ChatInterface(
+    fn=chat_interface,
+    title="Indian Legal Assistant",
+    description="""Welcome to the Indian Legal Assistant! I can help you understand:
+    - Bharatiya Nyaya Sanhita (BNS)
+    - Bharatiya Nagarik Suraksha Sanhita (BNSS)
+    - Bharatiya Sakshya Adhiniyam (BSA)
+    
+    I can:
+    1. Explain specific sections and provisions
+    2. Help understand your legal rights
+    3. Suggest relevant documents for your case
+    4. Guide you through legal procedures
+    5. Assist with document requirements
+    
+    Ask me any questions about these new criminal laws.""",
+    theme="soft",
+    examples=[
+        "What are the key changes in the new criminal laws?",
+        "How does BNS handle digital crimes?",
+        "What documents do I need to file an FIR under BNSS?",
+        "How is electronic evidence handled under BSA?",
+        "What are the provisions for mob lynching in BNS?"
+    ]
+)
+
+# Launch the interface
+if __name__ == "__main__":
+    iface.launch()