Update app.py

app.py

@@ -20,7 +20,10 @@ with st.sidebar:
     st.divider()
     
     # 2. FILE UPLOADER (The new feature)
-    uploaded_file = st.file_uploader("Upload Context", type=['pdf', 'docx', 'txt', 'csv', 'xlsx'])
+    uploaded_files = st.file_uploader(
+        "Upload Context", 
+        type=['pdf', 'docx', 'txt', 'csv', 'xlsx'],
+        accept_multiple_files=True)
     
     if st.button("Clear Chat"):
         st.session_state.messages = []
@@ -32,72 +35,123 @@ st.title("⚓ Executive Editor Pro")
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# --- LOGIC: HANDLE FILE UPLOAD ---
-# If a file is uploaded, we automatically extract it and put it in the chat as "User Context"
-if uploaded_file and "file_processed" not in st.session_state:
-    with st.spinner("Reading document..."):
-        file_text = file_processing.extract_text_from_file(uploaded_file)
+# --- MAIN LOGIC: HANDLE BATCH UPLOADS ---
+if uploaded_files and "file_processed" not in st.session_state:
+    st.session_state.file_processed = False
+
+# We check if files are uploaded AND if we haven't processed this specific batch yet
+if uploaded_files and not st.session_state.file_processed:
+    
+    master_text = ""
+    file_names = []
+    error_log = []
+    
+    with st.spinner(f"Processing {len(uploaded_files)} documents..."):
+        for file in uploaded_files:
+            text, error = file_processing.validate_and_extract(file)
+            
+            if error:
+                error_log.append(f"❌ {file.name}: {error}")
+            else:
+                # We wrap each file in XML tags so the AI knows where one ends and the next begins
+                master_text += f"\n<document name='{file.name}'>\n{text}\n</document>\n"
+                file_names.append(file.name)
+    
+    # Check if we successfully extracted anything
+    if master_text:
+        # Create the System Note
+        system_note = f"**[SYSTEM: User attached {len(file_names)} files]**\n\n{master_text}"
+        
+        # Create the UI Message (Hidden context, Clean display)
+        display_message = f"📂 **Batch Upload Processed:**\n"
+        for name in file_names:
+            display_message += f"- `{name}`\n"
+        
+        if error_log:
+            display_message += "\n**Errors:**\n" + "\n".join(error_log)
+
+        st.session_state.messages.append({
+            "role": "user", 
+            "content": system_note, 
+            "display_text": display_message
+        })
         
-        # We add a hidden system note or a user message with the file content
-        file_message = f"**[SYSTEM: Attached File Content from {uploaded_file.name}]**\n\n{file_text}"
+        st.session_state.file_processed = True
+        st.rerun()
         
-        # Append to history so the LLM sees it
-        st.session_state.messages.append({"role": "user", "content": file_message})
-        st.session_state.file_processed = True # Flag to prevent re-reading
-        st.success(f"Loaded {uploaded_file.name}")
+    elif error_log:
+        # If everything failed, just show errors
+        for e in error_log:
+            st.error(e)
 
 # --- DISPLAY CHAT ---
 for msg in st.session_state.messages:
     with st.chat_message(msg["role"]):
-        st.markdown(msg["content"])
+        # Check if there is specific 'display_text' (for files), otherwise use standard 'content'
+        display_content = msg.get("display_text", msg["content"])
+        st.markdown(display_content)
 
-# --- CHAT INPUT ---
+# --- CHAT INPUT & PROCESSING ---
 if user_input := st.chat_input("Type instructions or paste text..."):
     
-    # User step
+    # 1. User Step
     st.session_state.messages.append({"role": "user", "content": user_input})
     with st.chat_message("user"):
         st.markdown(user_input)
 
-    # AI step
+    # 2. AI Step
     with st.chat_message("assistant"):
         message_placeholder = st.empty()
         full_response = ""
         
-        # Construct Prompt: Base + Specific Mode
-        config = prompts.MODE_CONFIG[selected_mode]
-        system_message = f"""
-        {prompts.BASE_SYSTEM_PROMPT}
-        
-        CURRENT MODE: {selected_mode}
-        INSTRUCTION: {config['instruction']}
-        
-        EXAMPLES OF DESIRED OUTPUT:
-        {config['examples']}
-        """
-        
-        messages_payload = [{"role": "system", "content": system_instruction}] + st.session_state.messages
+        # --- FIX FOR NAME ERROR IS HERE ---
+        # We explicitly grab the config dictionary for the selected mode
+        mode_config = prompts.MODE_CONFIG[selected_mode]
         
-        stream = client.chat.completions.create(
-            model="gpt-4o",
-            messages=messages_payload,
-            temperature=0.3,
-            stream=True
-        )
+        # We construct the variable 'system_instruction' explicitly
+        system_instruction = f"""
+{prompts.BASE_SYSTEM_PROMPT}
+
+CURRENT MODE: {selected_mode}
+INSTRUCTION: {mode_config['instruction']}
+
+EXAMPLES OF DESIRED OUTPUT:
+{mode_config['examples']}
+"""
+        # ----------------------------------
+
+        # Build the payload
+        # Note: We filter the messages to only send 'role' and 'content' to OpenAI
+        # (OpenAI will crash if we send our custom 'display_text' field)
+        api_messages = [{"role": "system", "content": system_instruction}]
         
-        for chunk in stream:
-            if chunk.choices[0].delta.content:
-                full_response += chunk.choices[0].delta.content
-                message_placeholder.markdown(full_response + "▌")
+        for msg in st.session_state.messages:
+            api_messages.append({"role": msg["role"], "content": msg["content"]})
         
-        message_placeholder.markdown(full_response)
-        st.session_state.messages.append({"role": "assistant", "content": full_response})
+        try:
+            stream = client.chat.completions.create(
+                model="gpt-4o",
+                messages=api_messages,
+                temperature=0.3,
+                stream=True
+            )
+            
+            for chunk in stream:
+                if chunk.choices[0].delta.content:
+                    full_response += chunk.choices[0].delta.content
+                    message_placeholder.markdown(full_response + "▌")
+            
+            message_placeholder.markdown(full_response)
+            st.session_state.messages.append({"role": "assistant", "content": full_response})
 
-        # --- 3. DOWNLOAD BUTTON (The new feature) ---
-        # We place this immediately after the generation
-        st.download_button(
-            label="💾 Download Result",
-            data=full_response,
-            file_name="edited_text.md",
-            mime="text/markdown"
+            # Download Button
+            st.download_button(
+                label="💾 Download Result",
+                data=full_response,
+                file_name="edited_text.md",
+                mime="text/markdown"
+            )
+            
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
         )