app.py

app.py

@@ -15,6 +15,7 @@ st.caption("Chat with an AI Assistant on your LOR Chat Report")
 with st.sidebar:
     OPENAI_API_KEY = st.text_input("Enter your C2 Group of Technologies Access Key", type="password")
 
+# Check for valid API key
 if OPENAI_API_KEY:
     client = OpenAI(api_key=OPENAI_API_KEY)
 else:
@@ -47,7 +48,7 @@ if uploaded_file:
     st.write(df)
 
     # Process user input
-    prompt = f"Please analyze the data in the following format: {df.head()}"
+    prompt = f"Please analyze the following data: {df.head()}"
     st.session_state.messages.append({"role": "user", "content": prompt})
     st.chat_message("user").write(prompt)
 
@@ -91,17 +92,21 @@ if uploaded_file:
         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)
+        # Define the save path - Ensure it's accessible
+        try:
+            word_filename = "/mnt/data/AI_Report.docx"
+            doc.save(word_filename)
+
+            # Provide download link
+            with open(word_filename, "rb") as file:
+                st.download_button(
+                    label="Download the Report",
+                    data=file.read(),
+                    file_name="AI_Report.docx",
+                    mime="application/vnd.openxmlformats-officedocument.wordprocessingml.document"
+                )
+        except Exception as e:
+            st.error(f"Error saving the document: {str(e)}")
 
-        # 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)}")
+        st.error(f"Error processing the chat: {str(e)}")