Update app.py

app.py

@@ -17,16 +17,14 @@ st.set_page_config(page_title="Enterprise document search + chat", layout="wide"
 # Streamlit app header
 st.title("Enterprise document helpdesk")
 
-# Sidebar for API Key input
+# Sidebar
 with st.sidebar:
     st.header("Configuration")
     api_key = st.text_input("Enter your OpenAI API Key:", type="password")
     if api_key:
         os.environ["OPENAI_API_KEY"] = api_key
 
-# Main app logic
-if "OPENAI_API_KEY" in os.environ:
-    st.header('Multiple File Upload')
+    st.header('Document Upload and Processing')
     uploaded_files = st.file_uploader('Upload your files', accept_multiple_files=True, type=['txt', 'pdf', 'csv', 'ppt', 'doc', 'xls', 'pptx', 'xlsx'])
 
     def load_file(file):
@@ -55,6 +53,20 @@ if "OPENAI_API_KEY" in os.environ:
         os.unlink(temp_file_path)
         return documents
 
+    def summarize_documents(documents):
+        chat = ChatOpenAI(model="gpt-3.5-turbo-1106", temperature=0.2)
+        
+        combined_text = " ".join([doc.page_content for doc in documents])
+        
+        prompt = f"""Summarize the following document in a concise manner, highlighting the key points:
+
+        {combined_text}
+
+        Summary:"""
+        
+        response = chat.invoke(prompt)
+        return response.content
+
     # Process uploaded files
     if uploaded_files:
         if st.button("Process Documents"):
@@ -69,7 +81,21 @@ if "OPENAI_API_KEY" in os.environ:
                 # Store processed documents in session state
                 st.session_state.processed_documents = all_splits
             st.success("Documents processed successfully!")
+            
+            # Add a button for summarization
+            if st.button("Generate Summary"):
+                with st.spinner("Generating summary..."):
+                    summary = summarize_documents(st.session_state.processed_documents)
+                    st.session_state.document_summary = summary
+                st.success("Summary generated successfully!")
+
+    # Display the summary if it exists
+    if 'document_summary' in st.session_state:
+        st.subheader("Document Summary")
+        st.write(st.session_state.document_summary)
 
+# Main app logic
+if "OPENAI_API_KEY" in os.environ:
     # Initialize components
     @st.cache_resource
     def initialize_components():
@@ -79,8 +105,7 @@ if "OPENAI_API_KEY" in os.environ:
         return chat, embeddings
 
     # Load components
-    with st.spinner("Initializing Assistant..."):
-        chat, embeddings = initialize_components()
+    chat, embeddings = initialize_components()
 
     # Create vectorstore and retriever only if documents are processed
     if 'processed_documents' in st.session_state: