Update app.py

app.py

@@ -17,7 +17,6 @@ client = OpenAI(
 )
 
 print("Starting Nyaya-Mitra initialization...")
-print(f"Current working directory: {os.getcwd()}")
 
 # Initialize document processor
 try:
@@ -52,16 +51,16 @@ def get_mistral_response(query: str, context: str, max_retries: int = 3) -> str:
         return cached_response
         
     system_prompt = """You are Nyaya-Mitra (न्याय-मित्र), 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
+    Provide concise, accurate responses based on the context provided. Focus on the most relevant information.
     
-    Base your responses on the provided context and maintain a professional yet approachable tone.
-    Always format your responses in a clear, readable manner using markdown."""
+    Guidelines:
+    1. Be precise and cite specific sections
+    2. Explain concepts simply
+    3. Suggest next steps if applicable
+    4. Note that you cannot provide legal advice
+    5. Keep responses focused and relevant
+    
+    Format responses clearly using markdown."""
     
     for attempt in range(max_retries):
         try:
@@ -79,17 +78,17 @@ def get_mistral_response(query: str, context: str, max_retries: int = 3) -> str:
         except Exception as e:
             if "rate limit" in str(e).lower():
                 if attempt < max_retries - 1:
-                    wait_time = (attempt + 1) * 2  # Exponential backoff
+                    wait_time = (attempt + 1) * 2
                     time.sleep(wait_time)
                     continue
             return f"""I apologize, but I'm currently experiencing high demand. Please try:
             1. Waiting a few moments before asking another question
-            2. Rephrasing your question
-            3. Breaking down complex questions into simpler ones
+            2. Making your question more specific
+            3. Breaking down your question into smaller parts
             
             Error details: {str(e)}"""
     
-    return "I apologize, but I'm currently unable to process your request due to high demand. Please try again in a few moments."
+    return "I apologize, but I'm currently unable to process your request. Please try again in a few moments."
 
 def format_sources(metadatas: List[Dict]) -> str:
     """Format source information for display"""
@@ -107,10 +106,7 @@ def chat_interface(message: str, history: List[List[str]]) -> str:
     try:
         if doc_processor is None:
             return """I apologize, but I'm having trouble accessing the legal documents. 
-            This might be due to initialization issues. Please try:
-            1. Refreshing the page
-            2. Checking if the PDF files are properly uploaded
-            3. Waiting a few moments and trying again"""
+            Please try refreshing the page or waiting a moment."""
             
         # Search for relevant context
         results = doc_processor.search_documents(message)
@@ -128,9 +124,12 @@ def chat_interface(message: str, history: List[List[str]]) -> str:
         
         return formatted_response
     except Exception as e:
-        return f"""I apologize, but I encountered an error: {str(e)}. 
-        Please try again or rephrase your question. If the problem persists, 
-        try refreshing the page."""
+        return f"""I apologize, but I encountered an error. Please try:
+        1. Making your question more specific
+        2. Waiting a moment and trying again
+        3. Refreshing the page
+        
+        Error: {str(e)}"""
 
 # Create Gradio interface
 iface = gr.ChatInterface(
@@ -143,32 +142,23 @@ Your AI Legal Assistant for India's New Criminal Laws:
 - 📋 Bharatiya Nagarik Suraksha Sanhita (BNSS)
 - ⚖️ Bharatiya Sakshya Adhiniyam (BSA)
 
-I can help you:
-1. 📖 Understand specific sections and provisions
-2. ⚡ Learn about your legal rights
-3. 📝 Get guidance on required documents
-4. 🔍 Navigate legal procedures
-5. 💡 Get explanations in simple terms
-
-Ask me any questions about these new criminal laws!
-
-*Note: This is an AI assistant meant for information purposes only. For legal advice, please consult qualified legal professionals.*
+**Tips for Best Results:**
+1. Ask specific, focused questions
+2. One topic at a time
+3. Wait briefly between questions
+4. Use simple, clear language
 
-**Usage Tips:**
-- Ask one question at a time
-- Be specific in your questions
-- Wait a few moments between questions
-- Refresh the page if you encounter any issues""",
+*Note: This is an AI assistant for information only. For legal advice, please consult qualified legal professionals.*""",
     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?"
+        "What are the main provisions for cybercrime in BNS?",
+        "How do I file a complaint under BNSS?",
+        "What is the process for electronic evidence under BSA?",
+        "What are the key changes in mob lynching laws?",
+        "How does BNS define organized crime?"
     ]
 )
 
 # Launch the interface
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()