Update app.py

app.py

@@ -1,8 +1,6 @@
 import streamlit as st
 from openai import OpenAI
 import os
-
-# Import our new modules
 import prompts
 import file_processing
 
@@ -17,7 +15,7 @@ with st.sidebar:
     st.header("⚙️ Controls")
     
     # 1. Mode Selection (Pulled from prompts.py)
-    selected_mode = st.radio("Editing Mode", list(prompts.MODE_INSTRUCTIONS.keys()))
+    selected_mode = st.radio("Editing Mode", list(prompts.MODE_CONFIG.keys()))
     
     st.divider()
     
@@ -67,7 +65,16 @@ if user_input := st.chat_input("Type instructions or paste text..."):
         full_response = ""
         
         # Construct Prompt: Base + Specific Mode
-        system_instruction = prompts.BASE_SYSTEM_PROMPT + "\n\nCURRENT MODE: " + prompts.MODE_INSTRUCTIONS[selected_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