Update app2.py

app2.py

@@ -1888,141 +1888,6 @@ def policy_analysis_ui():
         else:
             st.warning("Please enter policy text or upload a document to analyze.")
 
-def generate_web_search_query(user_input: str) -> str:
-    """Generates a web search query based on user input using Falcon."""
-    prompt = f"""
-    Generate a concise and effective web search query that can be used 
-    to find additional information about the following legal situation 
-    related to US law:
-
-    ```
-    {user_input}
-    ``` 
-    """
-
-    try:
-        response = ai71.chat.completions.create(
-            model="tiiuae/falcon-180b-chat",
-            messages=[{"role": "user", "content": prompt}],
-            stream=False,
-        )
-        return response.choices[0].message.content.strip()
-    except Exception as e:
-        print(f"Error generating web search query: {str(e)}")
-        return user_input
-
-def provide_legal_advice(user_input: str) -> Dict[str, Any]:
-    """Provides legal advice, suggestions, and generates a web search query."""
-    chunks = split_text(user_input)
-    full_advice = ""
-
-    for i, chunk in enumerate(chunks):
-        advice_prompt = f"""
-        Provide legal advice based on US law for the following situation or question (part {i+1}/{len(chunks)}):
-
-        User Input (Part {i+1}/{len(chunks)}):
-        ```
-        {chunk}
-        ```
-
-        Please include:
-        1. A summary of the legal issue or question
-        2. Relevant US laws or regulations that apply
-        3. Possible legal implications or consequences
-        4. General advice or next steps (without constituting specific legal counsel)
-        5. Any important disclaimers or limitations of this advice
-        Remember to maintain a professional and objective tone throughout your response.
-
-        Suggestions: Provide 2-3 actionable suggestions that the user could consider based on the legal situation.
-        """
-
-        try:
-            chunk_advice = get_ai_response(advice_prompt)
-            full_advice += chunk_advice + "\n\n"
-        except Exception as e:
-            return {"error": f"Error providing legal advice (part {i+1}): {str(e)}"}
-
-    return {"advice": full_advice} 
-
-def legal_consultant_ui():
-    st.subheader("Legal Consultant")
-    
-    if 'consultant_chat_history' not in st.session_state:
-        st.session_state.consultant_chat_history = []
-    
-    if 'uploaded_document' not in st.session_state:
-        st.session_state.uploaded_document = None
-    
-    if 'consultant_chat_mode' not in st.session_state:  # New variable for consultant chat mode
-        st.session_state.consultant_chat_mode = "normal"
-    
-    st.write('''
-    Describe your legal situation or ask your legal question related to US law. 
-    LexAI will provide information and guidance based on its understanding of the US legal system. 
-    Please remember that this is not a substitute for real legal advice from a qualified attorney.
-    ''')
-
-    st.warning("Please do not upload files larger than 5MB as it may cause issues and consume all available tokens.")
-
-    uploaded_file = st.file_uploader("Upload a legal document (PDF, DOCX, or TXT)", type=["pdf", "docx", "txt"])
-    
-    if uploaded_file:
-        st.session_state.uploaded_document = extract_text_from_document(uploaded_file)
-        st.success("Document uploaded successfully!")
-    
-    # Chat mode toggle for Legal Consultant
-    if st.session_state.uploaded_document:
-        if st.button("Switch Chat Mode"):
-            st.session_state.consultant_chat_mode = "document" if st.session_state.consultant_chat_mode == "normal" else "normal"
-        
-        st.write(f"Current mode: {'Document-based' if st.session_state.consultant_chat_mode == 'document' else 'Normal'} chat")
-
-    # Display chat history (similar to Legal Chatbot)
-    for message in st.session_state.consultant_chat_history:
-        if isinstance(message, tuple):
-            user_msg, bot_msg = message
-            st.info(f"**You:** {user_msg}")
-            st.success(f"**Lex AI:** {bot_msg}")
-        elif isinstance(message, dict):
-            if message.get('type') == 'web_search':
-                web_results_msg = "Web Search Results:\n"
-                for result in message.get('results', []):
-                    web_results_msg += f"[{result.get('title', 'No title')}]({result.get('link', '#')})\n{result.get('snippet', 'No snippet available.')}\n\n"
-                st.success(f"**Lex AI:** {web_results_msg}")
-
-    user_input = st.text_input("Your legal question:")
-    
-    if user_input and st.button("Send"):
-        with st.spinner("Processing your question..."):
-            if st.session_state.consultant_chat_mode == "document" and st.session_state.uploaded_document:
-                ai_response = get_document_based_response(user_input, st.session_state.uploaded_document)  # Assuming you have this function
-                st.session_state.consultant_chat_history.append((user_input, ai_response))
-            else:
-                ai_response = provide_legal_advice(user_input)
-                st.session_state.consultant_chat_history.append((user_input, ai_response))
-                
-                # Perform web search (same as Legal Chatbot)
-                web_results = search_web_duckduckgo(user_input)  # Using user_input directly here
-                st.session_state.consultant_chat_history.append({
-                    'type': 'web_search',
-                    'results': web_results
-                })
-        
-        st.rerun()
-
-def display_chat_history_legal_advise():
-    for message in st.session_state.chat_history:
-        if isinstance(message, tuple):
-            user_msg, bot_msg = message
-            st.info(f"**You:** {user_msg}")
-            st.success(f"**Lex AI:** {bot_msg}")
-        elif isinstance(message, dict):
-            if message.get('type') == 'web_search':
-                web_results_msg = "Web Search Results:\n"
-                for result in message.get('results', []):
-                    web_results_msg += f"[{result.get('title', 'No title')}]({result.get('link', '#')})\n{result.get('snippet', 'No snippet available.')}\n\n"
-                st.success(f"**Lex AI:** {web_results_msg}") 
-
 def draft_contract(contract_details: str) -> Dict[str, Any]:
     chunks = split_text(contract_details)
     full_draft = ""
@@ -2238,7 +2103,7 @@ with st.sidebar:
     
     feature = st.selectbox(
         "Select a feature",
-        ["Legal Chatbot", "Document Analysis", "Case Precedent Finder", "Legal Cost Estimator", "Contract Analysis", "Case Trend Visualizer", "Case Information Retrieval", "Automated Legal Brief Generation", "Find the Lawyers", "Policy Analysis & Impact", "Legal Consultant", "Contract Drafting Assistant", "Predictive Case Analysis"]
+        ["Legal Chatbot", "Document Analysis", "Case Precedent Finder", "Legal Cost Estimator", "Contract Analysis", "Case Trend Visualizer", "Case Information Retrieval", "Automated Legal Brief Generation", "Find the Lawyers", "Policy Analysis & Impact", "Contract Drafting Assistant", "Predictive Case Analysis"]
     )
 if feature == "Legal Chatbot":
     st.subheader("Legal Chatbot")
@@ -2408,11 +2273,9 @@ elif feature == "Automated Legal Brief Generation":
 
 elif feature == "Find the Lawyers":
     lawyer_finder_ui()
+
 elif feature == "Policy Analysis & Impact":
     policy_analysis_ui()
-
-elif feature == "Legal Consultant":
-    legal_consultant_ui()
     
 elif feature == "Contract Drafting Assistant":
     contract_drafting_ui()