Update app.py

app.py

@@ -1,6 +1,8 @@
 import time
 import os
 import streamlit as st
+from reportlab.lib.pagesizes import letter
+from reportlab.pdfgen import canvas
 from langchain_community.vectorstores import FAISS
 from langchain_community.embeddings import HuggingFaceEmbeddings
 from langchain.prompts import PromptTemplate
@@ -13,10 +15,23 @@ from footer import footer
 # Set the Streamlit page configuration and theme
 st.set_page_config(page_title="In-Legal-IPC", layout="centered")
 
-# Display the logo image
+# Display the logo image with blur shadow
 col1, col2, col3 = st.columns([1, 30, 1])
 with col2:
-    st.image("https://raw.githubusercontent.com/shiv4321/Images/refs/heads/main/Banner.png", use_container_width=True)
+    st.markdown(
+        """
+        <style>
+        .blur-shadow {
+            box-shadow: 0px 4px 20px rgba(0, 0, 0, 0.5);
+            border-radius: 10px;
+        }
+        </style>
+        <div class="blur-shadow">
+            <img src="https://raw.githubusercontent.com/shiv4321/Images/refs/heads/main/Banner.png" alt="Banner" width="100%">
+        </div>
+        """,
+        unsafe_allow_html=True
+    )
 
 def hide_hamburger_menu():
     st.markdown("""
@@ -44,6 +59,7 @@ embeddings = load_embeddings()
 db = FAISS.load_local("ipc_embed_db", embeddings, allow_dangerous_deserialization=True)
 db_retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 3})
 
+# Define the prompt template
 prompt_template = """
 <s>[INST]
 As a legal chatbot specializing in the Indian Penal Code, you are tasked with providing highly accurate and contextually appropriate responses. Ensure your answers meet these criteria:
@@ -86,14 +102,56 @@ def reset_conversation():
     st.session_state.messages = []
     st.session_state.memory.clear()
 
-# Add the PDF button just above the chat input
+# Function to create a PDF
+def create_pdf(content):
+    pdf_filename = "legal_letter.pdf"
+    c = canvas.Canvas(pdf_filename, pagesize=letter)
+    width, height = letter
+    c.drawString(100, height - 100, 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)
+    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)
+
+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)
+    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)
+        
+
+# Add CSS for the button styling
 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)
+    <style>
+    .stButton button {
+        background-color: #ADD8E6;
+        color: black;
+        font-size: 16px;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+
 
 # Display previous messages
 for message in st.session_state.messages:
@@ -123,6 +181,13 @@ if input_prompt:
 
         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()