Update app.py

app.py

@@ -10,16 +10,20 @@ from langchain.embeddings import HuggingFaceEmbeddings
 import fitz  # PyMuPDF
 import base64
 
+# Load environment variables
 load_dotenv()
 client = groq.Client(api_key=os.getenv("GROQ_FINANCE_API_KEY"))
 embeddings = HuggingFaceEmbeddings(model_name="sentence-transformers/all-MiniLM-L6-v2")
 
+# Directory to store FAISS indexes
 FAISS_INDEX_DIR = "faiss_indexes_finance"
 if not os.path.exists(FAISS_INDEX_DIR):
     os.makedirs(FAISS_INDEX_DIR)
 
+# Dictionary to store user-specific vectorstores
 user_vectorstores = {}
 
+# Custom CSS for styling
 custom_css = """
 :root {
     --primary-green: #10B981;
@@ -47,6 +51,7 @@ body { background-color: var(--light-grey); font-family: 'Inter', sans-serif; }
 .stats-box { background-color: var(--light-green); padding: 10px; border-radius: 8px; margin-top: 10px; }
 """
 
+# Function to process PDF files
 def process_pdf(pdf_file):
     if pdf_file is None:
         return None, "No file uploaded", {"page_images": [], "total_pages": 0, "total_words": 0}
@@ -87,6 +92,7 @@ def process_pdf(pdf_file):
             os.unlink(pdf_path)
         return None, f"Error processing PDF: {str(e)}", {"page_images": [], "total_pages": 0, "total_words": 0}
 
+# Function to generate chatbot responses
 def generate_response(message, session_id, model_name, history):
     if not message:
         return history
@@ -116,16 +122,19 @@ def generate_response(message, session_id, model_name, history):
         history.append((message, f"Error generating response: {str(e)}"))
         return history
 
+# Function to update the PDF viewer with the first page
 def update_pdf_viewer(pdf_state):
     if not pdf_state["total_pages"]:
         return 0, None, "No PDF uploaded yet"
     return pdf_state["total_pages"], f"data:image/png;base64,{pdf_state['page_images'][0]}", f"**Total Pages:** {pdf_state['total_pages']}\n**Total Words:** {pdf_state['total_words']}"
 
+# Function to update the displayed PDF page based on the slider value
 def update_image(page_num, pdf_state):
     if not pdf_state["total_pages"] or page_num < 1 or page_num > pdf_state["total_pages"]:
         return None
     return f"data:image/png;base64,{pdf_state['page_images'][page_num - 1]}"
 
+# Gradio interface
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     current_session_id = gr.State(None)
     pdf_state = gr.State({"page_images": [], "total_pages": 0, "total_words": 0})
@@ -155,7 +164,7 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
                     clear_btn = gr.Button("Clear Conversation")
                 with gr.TabItem("PDF Viewer"):
                     with gr.Column(elem_classes="pdf-viewer-container"):
-                        page_slider = gr.Slider(minimum=1, maximum=0, step=1, label="Page Number", value=1)
+                        page_slider = gr.Slider(minimum=1, maximum=1, step=1, label="Page Number", value=1)
                         pdf_image = gr.Image(label="PDF Page", type="pil", elem_classes="pdf-viewer-image")
                         stats_display = gr.Markdown("No PDF uploaded yet", elem_classes="stats-box")
     
@@ -194,5 +203,6 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
         outputs=[pdf_image]
     )
 
+# Launch the app
 if __name__ == "__main__":
     demo.launch()