Update app.py

app.py

@@ -3,6 +3,8 @@ import os
 import streamlit as st
 from reportlab.lib.pagesizes import letter
 from reportlab.pdfgen import canvas
+from googleapiclient.discovery import build
+from google.oauth2.service_account import Credentials
 from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain.prompts import PromptTemplate
@@ -11,6 +13,48 @@ from langchain.chains import ConversationalRetrievalChain
 from langchain_together import Together
 
 from footer import footer
+ 
+# Google Drive API setup
+SCOPES = ["https://www.googleapis.com/auth/drive.readonly"]
+SERVICE_ACCOUNT_FILE = "data/credentials.json"  # Path to your Google API credentials file
+FOLDER_ID = "1LZIx-1tt_GormpU8nF_I2WL88Oxa9juU"  # Replace with your Google Drive folder ID
+
+def authenticate_drive():
+    """Authenticate and return the Google Drive API service."""
+    creds = Credentials.from_service_account_file(SERVICE_ACCOUNT_FILE, scopes=SCOPES)
+    return build("drive", "v3", credentials=creds)
+
+
+
+from fuzzywuzzy import process
+
+def search_drive_file(file_name):
+    """Search for a file by name in the specified Google Drive folder using fuzzy matching."""
+    service = authenticate_drive()
+    try:
+        # Get all files in the folder
+        query = f"'{FOLDER_ID}' in parents and trashed=false"
+        results = service.files().list(q=query, fields="files(id, name)").execute()
+        files = results.get("files", [])
+        
+        # Debug: Print all file names for inspection
+        st.write("Available files:", [f['name'] for f in files])
+        
+        # Perform fuzzy matching to find the best match
+        file_names = [f['name'] for f in files]
+        best_match, score = process.extractOne(file_name, file_names)
+
+        if score >= 75:  # Threshold for a match
+            matched_file = next(f for f in files if f['name'] == best_match)
+            st.write(f"Match found: {matched_file['name']} (Score: {score})")
+            return [matched_file]
+        else:
+            st.warning(f"No close matches found for '{file_name}'. Try rephrasing or checking the folder manually.")
+            return []
+
+    except Exception as e:
+        st.error(f"An error occurred: {e}")
+        return []
 
 # Set the Streamlit page configuration and theme
 st.set_page_config(page_title="In-Legal-IPC", layout="centered")
@@ -111,34 +155,21 @@ def create_pdf(content):
     c.save()
     return pdf_filename
 
-
 # Add links to multiple PDFs just above the chat input
-#st.markdown("""
-#    ### Useful PDFs
-#    - [📄 Commercial Court Rules and Forms](https://drive.google.com/file/d/1puzlPMT7fTt4utWJaGlFtOjW38CoFctc/view?usp=drive_link)
-#    - [📄 Bail-Bond](https://drive.google.com/file/d/1uRlT7Yo_2jemxs5aRyvHzoLgeS7S81Vn/view?usp=drive_link)
-#    - [📄 Inspection Form](https://drive.google.com/file/d/1Ib-RC4xPMqZVl7YgVES3Vb47Rb42W83s/view?usp=drive_link)
-#    - [📄 Additional PDF](https://drive.google.com/file/d/1Xkq64r4Id8qSyb5woVzdvArhzUxvKJk8/view?usp=drive_link)
-# """, unsafe_allow_html=True)
-
-
-
-# Add the PDF buttons
 st.markdown("<h3 class='underline'>Useful PDFs</h3>", unsafe_allow_html=True)
 
 col1, col2 = st.columns(2)  # Create two columns for better alignment
 with col1:
     if st.button("Commercial Court Rules and Forms 📄", key="ccrf", help="Open PDF", use_container_width=True):
-        st.markdown("[Open PDF](https://drive.google.com/file/d/1puzlPMT7fTt4utWJaGlFtOjW38CoFctc/view?usp=drive_link)", unsafe_allow_html=True)
+        st.markdown("[Open PDF](https://drive.google.com/file/d/198SC1mKipJ7WQXGN-5uc8qkNV5rLxVlT/view?usp=sharing)", unsafe_allow_html=True)
     if st.button("Bail-Bond 📄", key="bb", help="Open PDF", use_container_width=True):
-        st.markdown("[Open PDF](https://drive.google.com/file/d/1uRlT7Yo_2jemxs5aRyvHzoLgeS7S81Vn/view?usp=drive_link)", unsafe_allow_html=True)
+        st.markdown("[Open PDF](https://drive.google.com/file/d/1Eju14MgFFME3nUknjwlbU8C9nrQoeM1v/view?usp=drive_link)", unsafe_allow_html=True)
 
 with col2:
     if st.button("Inspection Form 📄", key="if", help="Open PDF", use_container_width=True):
-        st.markdown("[Open PDF](https://drive.google.com/file/d/1Ib-RC4xPMqZVl7YgVES3Vb47Rb42W83s/view?usp=drive_link)", unsafe_allow_html=True)
+        st.markdown("[Open PDF](https://drive.google.com/file/d/17FT5Pmgp4bgf31tFyQRMNVnoRuVlQ2zi/view?usp=sharing)", unsafe_allow_html=True)
     if st.button("Additional PDF 📄", key="apdf", help="Open PDF", use_container_width=True):
-        st.markdown("[Open PDF](https://drive.google.com/file/d/1Xkq64r4Id8qSyb5woVzdvArhzUxvKJk8/view?usp=drive_link)", unsafe_allow_html=True)
-        
+        st.markdown("[Open PDF](https://drive.google.com/file/d/1LY1-R9chmd_I7Tf3iC4jNZ5dHRFFkjaV/view?usp=sharing)", unsafe_allow_html=True)
 
 # Add CSS for the button styling
 st.markdown("""
@@ -151,45 +182,79 @@ st.markdown("""
     </style>
     """, unsafe_allow_html=True)
 
-
-
 # Display previous messages
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.write(message["content"])
 
+# Initialize session state variables
+if "show_reset" not in st.session_state:
+    st.session_state.show_reset = False
+
 # Chat input area
 input_prompt = st.chat_input("Say something...")
 if input_prompt:
     with st.chat_message("user"):
         st.markdown(f"**You:** {input_prompt}")
 
-    st.session_state.messages.append({"role": "user", "content": input_prompt})
-    with st.chat_message("assistant"):
-        with st.spinner("Thinking 💡..."):
-            result = qa.invoke(input=input_prompt)
-            message_placeholder = st.empty()
-            answer = extract_answer(result["answer"])
-
-            # Initialize the response message
-            full_response = "⚠️ **_Gentle reminder: We generally ensure precise information, but do double-check._** \n\n\n"
-            for chunk in answer:
-                # Simulate typing by appending chunks of the response over time
-                full_response += chunk
-                time.sleep(0.02)  # Adjust the sleep time to control the "typing" speed
-                message_placeholder.markdown(full_response + " |", unsafe_allow_html=True)
-
-        st.session_state.messages.append({"role": "assistant", "content": answer})
-
-        # Generate PDF if the user indicates a personal legal problem
-        if "personal legal problem" in input_prompt.lower():
-            user_problem = input_prompt  # Capture the user's legal problem
-            pdf_content = f"Dear Client,\n\nYour legal problem is as follows:\n{user_problem}\n\nBest Regards,\nYour Legal Advisor"
-            pdf_file = create_pdf(pdf_content)
-            st.download_button(label="Download your legal letter", data=pdf_file, file_name="legal_letter.pdf")
-
-        if st.button('🗑️ Reset All Chat', on_click=reset_conversation):
-            st.experimental_rerun()
-
-# Define the CSS to style the footer
-footer()
+    # Enable the reset button after receiving input
+    st.session_state.show_reset = True
+
+    if "form" in input_prompt.lower() or "document" in input_prompt.lower():
+        with st.spinner("Searching Google Drive..."):
+            # Call the updated search function
+            search_results = search_drive_file(input_prompt)
+        
+        if search_results:
+            # Generate response for found files
+            response = "🔍 Document(s) found! Click below to view:"
+            for file in search_results:
+                response += f"\n- [{file['name']}](https://drive.google.com/file/d/{file['id']}/view)"
+            st.session_state.messages.append({"role": "assistant", "content": response})
+            st.write(response)
+        else:
+            # If no results, provide an alternative message
+            response = (
+                "⚠️ No matching documents found. "
+                "Please check the spelling or explore the folder directly: "
+                f"[Google Drive Folder](https://drive.google.com/drive/folders/{FOLDER_ID})"
+            )
+            st.session_state.messages.append({"role": "assistant", "content": response})
+            st.write(response)
+
+    else:
+        # Handle general questions
+        with st.chat_message("assistant"):
+            with st.spinner("Thinking 💡..."):
+                try:
+                    # Validate the input before invoking the QA chain
+                    if not input_prompt.strip():
+                        st.warning("⚠️ Input cannot be empty!")
+                    else:
+                        result = qa.invoke(input=input_prompt)
+                        answer = result["answer"].strip()
+
+                        # Simulate typing effect for the response
+                        message_placeholder = st.empty()
+                        full_response = (
+                            "⚠️ **_Gentle reminder: We strive for precision, but please double-check._**\n\n"
+                        )
+                        for chunk in answer.split():
+                            full_response += chunk + " "
+                            time.sleep(0.02)  # Simulating typing
+                            message_placeholder.markdown(full_response + " |", unsafe_allow_html=True)
+                        
+                        st.session_state.messages.append({"role": "assistant", "content": answer})
+
+                except Exception as e:
+                    # Handle unexpected errors during QA invocation
+                    error_message = f"⚠️ **_Error: An unexpected issue occurred: {str(e)}._**"
+                    st.error(error_message)
+                    st.session_state.messages.append({"role": "assistant", "content": error_message})
+
+# Reset button
+if st.session_state.show_reset:
+    if st.button('🗑️ Reset All Chat', on_click=reset_conversation):
+        st.rerun()  # Updated from st.experimental_rerun
+
+footer()