Update app.py

app.py

@@ -2,109 +2,95 @@ import streamlit as st
 from openai import OpenAI
 import os
 
-# 1. Configuration
-st.set_page_config(page_title="Executive Editor", layout="wide", page_icon="⚓")
+# Import our new modules
+import prompts
+import file_processing
 
-# API Key Logic (Same as before)
-api_key = os.environ.get("OPENAI_API_KEY")
-if not api_key:
-    try:
-        api_key = st.secrets["OPENAI_API_KEY"]
-    except (FileNotFoundError, KeyError):
-        st.error("OpenAI API Key not found.")
-        st.stop()
+st.set_page_config(page_title="Executive Editor Pro", layout="wide", page_icon="⚓")
 
+# ... (API Key Logic remains the same) ...
+api_key = os.environ.get("OPENAI_API_KEY")
 client = OpenAI(api_key=api_key)
 
-# 2. Sidebar Controls (The "Executive Dashboard")
+# --- SIDEBAR ---
 with st.sidebar:
-    st.header("🖊️ Editing Parameters")
+    st.header("⚙️ Controls")
     
-    # Mode Selection
-    edit_mode = st.radio(
-        "Optimization Goal",
-        [
-            "Standard Polish (Grammar & Flow)",
-            "BLUF (Bottom Line Up Front)",
-            "Diplomatic / De-escalate",
-            "Executive Summary (Concise)",
-            "Technical Precision"
-        ]
-    )
-    
-    # Output Format
-    format_mode = st.selectbox("Output Format", ["Plain Text", "Email Structure", "Memo Format", "Bullet Points"])
+    # 1. Mode Selection (Pulled from prompts.py)
+    selected_mode = st.radio("Editing Mode", list(prompts.MODE_INSTRUCTIONS.keys()))
     
     st.divider()
     
-    if st.button("🗑️ Clear Context", use_container_width=True):
+    # 2. FILE UPLOADER (The new feature)
+    uploaded_file = st.file_uploader("Upload Context", type=['pdf', 'docx', 'txt', 'csv', 'xlsx'])
+    
+    if st.button("Clear Chat"):
         st.session_state.messages = []
         st.rerun()
 
-# 3. The "Command" System Prompt
-# Note the lower temperature and specific instructions for professional contexts.
-SYSTEM_PROMPT = f"""
-You are an expert Executive Editor and Technical Writer.
-Your goal is to rewrite the user's input to be clear, concise, and professional.
-
-CURRENT SETTINGS:
-- **Goal:** {edit_mode}
-- **Format:** {format_mode}
-
-OPERATIONAL RULES:
-1. **Direct Action:** Do not explain your changes unless asked. Just provide the rewritten text.
-2. **Preserve Intent:** Never change the core meaning or facts of the input.
-3. **No Fluff:** Remove adjectives and adverbs that do not add value. 
-4. **BLUF:** If the mode is "BLUF", ensure the main request or conclusion is the very first sentence.
-5. **Diplomacy:** If the mode is "Diplomatic", strip out aggression/emotion and replace with professional firmness.
-
-Start every response immediately with the rewritten text. 
-"""
+# --- MAIN APP ---
+st.title("⚓ Executive Editor Pro")
 
-# 4. Session State
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# 5. Main Interface
-st.title("⚓ Executive Editor")
-st.markdown(f"**Current Mode:** `{edit_mode}`")
+# --- 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)
+        
+        # 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}"
+        
+        # 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}")
 
-# Display History
-for message in st.session_state.messages:
-    with st.chat_message(message["role"]):
-        st.markdown(message["content"])
+# --- DISPLAY CHAT ---
+for msg in st.session_state.messages:
+    with st.chat_message(msg["role"]):
+        st.markdown(msg["content"])
 
-# Input Handling
-if user_input := st.chat_input("Paste text to edit, or type instructions..."):
+# --- CHAT INPUT ---
+if user_input := st.chat_input("Type instructions or paste text..."):
     
-    # User Message
+    # User step
     st.session_state.messages.append({"role": "user", "content": user_input})
     with st.chat_message("user"):
         st.markdown(user_input)
 
-    # AI Response
+    # AI step
     with st.chat_message("assistant"):
         message_placeholder = st.empty()
         full_response = ""
         
-        # We re-inject the system prompt every turn to capture Sidebar changes
-        messages_payload = [{"role": "system", "content": SYSTEM_PROMPT}] + st.session_state.messages
+        # Construct Prompt: Base + Specific Mode
+        system_instruction = prompts.BASE_SYSTEM_PROMPT + "\n\nCURRENT MODE: " + prompts.MODE_INSTRUCTIONS[selected_mode]
+        
+        messages_payload = [{"role": "system", "content": system_instruction}] + st.session_state.messages
         
-        try:
-            response = client.chat.completions.create(
-                model="gpt-4o",
-                messages=messages_payload,
-                temperature=0.3, # Low temp = High consistency/Precision
-                stream=True
-            )
-            
-            for chunk in response:
-                if chunk.choices[0].delta.content is not None:
-                    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})
-            
-        except Exception as e:
-            st.error(f"Error: {e}")
+        stream = client.chat.completions.create(
+            model="gpt-4o",
+            messages=messages_payload,
+            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"
+        )