Update app.py

app.py

@@ -1,17 +1,20 @@
 import streamlit as st
 from llama_index.core import VectorStoreIndex, Document
-from llama_index.llms import OpenAI
+from llama_index.llms.openai import OpenAI
 import os
 import pdfplumber
 from docx import Document as DocxDocument
 from dotenv import load_dotenv
 import json
+
+# Load environment variables from .env file
 load_dotenv()
 
-st.header("Chat")
+st.header("Chat with the uploaded docs 💬 📚")
 
+# Sidebar for OpenAI API Key
 if 'openai_api_key' not in st.session_state:
-  st.session_state.openai_api_key = ""
+    st.session_state.openai_api_key = ""
 
 # Input for OpenAI API Key
 st.session_state.openai_api_key = st.sidebar.text_input("Enter your OpenAI API Key:", type="password",
@@ -21,32 +24,29 @@ st.session_state.openai_api_key = st.sidebar.text_input("Enter your OpenAI API K
 if 'phone_number' not in st.session_state:
     st.session_state.phone_number = ""
 
-st.session_state.phone_number = st.sidebar.text_input("Enter your phone number: ")
-
-# Instructions
-if 'messages' not in st.session_state:
-  st.session_state.messages = [
-      {"role": "assistant", "content": "Ask me a question about the docs you uploaded!"}, 
-  ]
+st.session_state.phone_number = st.sidebar.text_input("Enter your phone number:")
 
+# Initialize session state for messages
+if "messages" not in st.session_state:
+    st.session_state.messages = [
+        {"role": "assistant", "content": "Ask me a question about the documents you uploaded!"}
+    ]
 
+# Function to read PDF files
 def read_pdf(file):
-  with pdfplumber.open(file) as pdf:
-    text = ''
-    for page in pdf.pages:
-      text += page.extract_text + '\n'
-  
-  return text 
-
+    with pdfplumber.open(file) as pdf:
+        text = ''
+        for page in pdf.pages:
+            text += page.extract_text() + '\n'
+    return text
 
+# Function to read DOCX files
 def read_docx(file):
-  doc = DocxDocument(file)
-  text = ''
-  
-  for paragraph in doc.paragraphs:
-    text += paragraph.text + '\n'
-
-  return text 
+    doc = DocxDocument(file)
+    text = ''
+    for paragraph in doc.paragraphs:
+        text += paragraph.text + '\n'
+    return text
 
 @st.cache_resource(show_spinner=False)
 def load_data(uploaded_files):
@@ -66,21 +66,18 @@ def load_data(uploaded_files):
         index = VectorStoreIndex.from_documents(docs, settings=llm)
         return index
 
+# Function to save the conversation
 def save_conversation():
-  if st.session_state.phone_number:
-
-    conversation_entry = {
-        "phone_number": st.session_state.phone_number,
-        "messages": st.session_state.messages,
-    }
-
-    with open("conversation.json", "a") as f:
-      json.dump(conversation_entry, f) # dump means adding 
-      f.write("\n")
-  
-  else:
-    st.warning("Please enter phone number before saving conversation")
-
+    if st.session_state.phone_number:
+        conversation_entry = {
+            "phone_number": st.session_state.phone_number,
+            "messages": st.session_state.messages
+        }
+        with open("conversations.json", "a") as f:
+            json.dump(conversation_entry, f)
+            f.write("\n")
+    else:
+        st.warning("Please enter a phone number before saving the conversation.")
 
 # Function to load previous conversations based on phone number
 def load_conversations(phone_number):
@@ -101,15 +98,14 @@ def load_conversations(phone_number):
             return conversations
     return []
 
-
+# Function to delete selected conversations
 def delete_selected_conversations(selected_indices, phone_number):
-  conversations = load_conversations(phone_number)
-  remaining_conversation = [conv for i, conv in enumerate(conversations) if i not in selected_indices] 
-  
-  with open("conversatinons.json", "w") as f:
-    for conv in remaining_conversations:
-      json.dump(conv, f)
-      f.write("\n")
+    conversations = load_conversations(phone_number)
+    remaining_conversations = [conv for i, conv in enumerate(conversations) if i not in selected_indices]
+    with open("conversations.json", "w") as f:
+        for conv in remaining_conversations:
+            json.dump(conv, f)
+            f.write("\n")
 
 # File uploader for multiple PDF and DOCX files
 uploaded_files = st.file_uploader("Upload PDF or DOCX files", type=["pdf", "docx"], accept_multiple_files=True)
@@ -188,4 +184,4 @@ if st.session_state.show_conversations and st.session_state.phone_number:
     else:
         st.sidebar.write("No previous conversations found for this phone number.")
 else:
-    st.sidebar.write("Previous conversations are hidden. Enter your phone number to view them.")
+    st.sidebar.write("Previous conversations are hidden. Enter your phone number to view them.")