Update app2.py

app2.py

@@ -96,7 +96,7 @@ def get_document_based_response(prompt, document_content):
 def get_ai_response(prompt: str) -> str:
     """Gets the AI response based on the given prompt."""
     messages = [
-        {"role": "system", "content": "You are a helpful legal assistant with advanced capabilities."},
+        {"role": "system", "content": "You are a helpful legal assistant LexAI which has all the legal information in the world and is the the best assitand for lawyers, lawfirms and a common citizen, answer the question based on the US law but if the question lies out of the context of us law then answer it too saying i am LexAI and advanced legal assistant but this is what i know about the topic you are asking"},
         {"role": "user", "content": prompt}
     ]
     try:
@@ -1503,6 +1503,13 @@ def generate_legal_brief(case_info):
         6. An analysis of why the winning party won
         7. A review of how the losing party could have potentially won
 
+        And if the user has just provided the case information or details of an case which has not been presented in the court or its outcome(result) is yet to come then generate:
+        1. A summary of the facts and the case details.
+        2. How the user can win this case based on the provided information.
+        3. Key areas where user should be carefull and could potentially loose this case.
+        4. Relevant Arguments for this case to be provided in the court.
+        5. predict wheter user can win this case or not.
+
         Case Information (Part {i+1}/{len(chunks)}):
         {chunk}
 
@@ -1523,7 +1530,7 @@ def generate_legal_brief(case_info):
 
 def automated_legal_brief_generation_ui():
     st.title("Automated Legal Brief Generation")
-
+    st.subheader("enter the caase details to generate a legal brief or to generate legal brief or predict the case outcome and how the case should proceed")
     if 'legal_brief' not in st.session_state:
         st.session_state.legal_brief = ""