app.py

app.py

@@ -1,11 +1,15 @@
 import streamlit as st
 from openai import OpenAI
 import time
+import pandas as pd
+from docx import Document
 
+# Set the page configuration
 st.set_page_config(page_title="LOR Chat ReportAi")
 
+# Title and caption
 st.title("LOR Chat Report Ai Assistant")
-st.caption("Chat with an Ai Assistant on your LOR Chat Report")
+st.caption("Chat with an AI Assistant on your LOR Chat Report")
 
 # Sidebar for API Key input
 with st.sidebar:
@@ -34,7 +38,16 @@ for message in st.session_state.messages:
     st.chat_message(role).write(content)
 
 # Process user input
-if prompt := st.chat_input():
+uploaded_file = st.file_uploader("Upload Excel file", type=["xlsx"])
+
+if uploaded_file:
+    # Read the Excel file
+    df = pd.read_excel(uploaded_file)
+    st.write("Data from uploaded Excel file:")
+    st.write(df)
+
+    # Process user input
+    prompt = f"Please analyze the data in the following format: {df.head()}"
     st.session_state.messages.append({"role": "user", "content": prompt})
     st.chat_message("user").write(prompt)
 
@@ -72,6 +85,23 @@ if prompt := st.chat_input():
 
         # Store in session state
         st.session_state.messages.append({"role": "assistant", "content": assistant_message})
+
+        # Generate Word (DOCX) file for download
+        doc = Document()
+        doc.add_heading('AI Assistant Report', 0)
+        doc.add_paragraph(assistant_message)
+
+        # Save the document
+        word_filename = "/mnt/data/AI_Report.docx"
+        doc.save(word_filename)
+
+        # Provide download link
+        st.download_button(
+            label="Download the Report",
+            data=open(word_filename, "rb").read(),
+            file_name="AI_Report.docx",
+            mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+        )
     
     except Exception as e:
         st.error(f"Error: {str(e)}")