Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,38 +5,40 @@ import io
 import re
 
 # ==========================================
-# 1. GLOBAL INITIALIZATION & UI SETUP
+# 1. UI CONFIGURATION & SECURE SETUP
 # ==========================================
 st.set_page_config(page_title="CX Architect Auto-Mapper", layout="wide", page_icon="🏗️")
 st.title("🏗️ CX Architect: Omni-Channel Generator")
 st.markdown("Upload your PRD, VOC data, or an existing spreadsheet. The AI engine will instantly map the customer journey and generate a structured VA Design Sheet.")
 
-# Initialize variables at the very top to prevent any scope or NameError crashes
-api_key = ""
-demo_mode = False
-
-# Configuration Sidebar Panel
+# Sidebar Configuration Panel
 with st.sidebar:
     st.header("⚙️ Configuration")
     
-    # Fail-safe Toggle for Presentation Stability
+    # 1. Toggle for Demo Simulation Mode
     demo_mode = st.toggle("🚀 Activate Demo Simulation Mode", value=False, help="Turn this ON if your Google API Key is unavailable or restricted.")
     
-    if not demo_mode:
-        api_key = st.text_input("Enter Gemini API Key", type="password")
-        if api_key.startswith("AQ."):
-            st.warning("⚠️ Restricted 'AQ.' token detected. If it fails, switch to Demo Simulation Mode.")
-    else:
+    # 2. Key Input - ALWAYS evaluated so it NEVER throws a NameError, but disabled if Demo Mode is active
+    api_key = st.text_input(
+        "Enter Gemini API Key", 
+        type="password", 
+        disabled=demo_mode,
+        placeholder="Paste your AIza... key here"
+    )
+    
+    if demo_mode:
         st.success("🟢 Simulation Mode Active (No Key Required)")
+    elif api_key.startswith("AQ."):
+        st.warning("⚠️ Restricted 'AQ.' token detected. If it fails, switch to Demo Simulation Mode.")
         
     st.markdown("---")
     st.markdown("**Framework: LAMF Process Automation Engine**")
 
-# File Selection Support Box
+# Multi-Format File Selection Box
 uploaded_file = st.file_uploader("Drop your Lending Document here (PDF, TXT, or CSV)", type=["pdf", "txt", "csv"])
 
 # ==========================================
-# 2. CORE STORAGE FOR SIMULATED RUNS
+# 2. DATA PROCESSING & SIMULATION LAYOUTS
 # ==========================================
 def get_simulation_assets():
     """Returns high-fidelity layout data modeled directly from real lending datasets"""
@@ -68,8 +70,23 @@ def get_simulation_assets():
     ]
     return flow_text, pd.DataFrame(rows, columns=headers)
 
+def safe_read_file(file):
+    name = file.name
+    try:
+        if name.endswith('.pdf'):
+            import PyPDF2
+            pdf_reader = PyPDF2.PdfReader(io.BytesIO(file.getvalue()))
+            return "".join([page.extract_text() or "" for page in pdf_reader.pages])
+        elif name.endswith('.csv'):
+            df = pd.read_csv(io.BytesIO(file.getvalue()), errors='ignore')
+            return df.to_string()
+        else:
+            return file.getvalue().decode("utf-8", errors="ignore")
+    except Exception as e:
+        return f"Error: {str(e)}"
+
 # ==========================================
-# 3. INTERFACE EXECUTION RUNTIME
+# 3. RUNTIME CONTROLLER
 # ==========================================
 if st.button("🚀 Generate Design Sheets & Flow"):
     if not demo_mode and not api_key:
@@ -78,12 +95,10 @@ if st.button("🚀 Generate Design Sheets & Flow"):
         st.error("⚠️ Data Source Missing! Please upload a PDF, TXT, or CSV file to analyze.")
     else:
         with st.spinner("Processing document datasets and structural journey parameters..."):
-            
-            # Setup container objects to capture outputs securely
-            flow_output = ""
-            dataframe_output = None
-            
             try:
+                flow_output = ""
+                dataframe_output = None
+                
                 # SCENARIO A: Direct Presentation Simulation Mode
                 if demo_mode:
                     flow_output, dataframe_output = get_simulation_assets()
@@ -92,20 +107,11 @@ if st.button("🚀 Generate Design Sheets & Flow"):
                 elif uploaded_file.name.endswith('.csv'):
                     raw_df = pd.read_csv(io.BytesIO(uploaded_file.getvalue()), errors='ignore')
                     dataframe_output = raw_df
-                    flow_output = f"📋 Native Document File Detected: '{uploaded_file.name}'\nSuccessfully loaded {len(raw_df)} workflow configuration rows directly into the dashboard matrix view."
+                    flow_output = f"📋 Native Document File Detected: '{uploaded_file.name}'\nSuccessfully loaded {len(raw_df)} rows directly into the matrix view."
                     
                 # SCENARIO C: Live AI Extraction Mode
                 else:
-                    # Read the file bytes safely
-                    file_name = uploaded_file.name
-                    if file_name.endswith('.pdf'):
-                        import PyPDF2
-                        pdf_reader = PyPDF2.PdfReader(io.BytesIO(uploaded_file.getvalue()))
-                        doc_text = "".join([page.extract_text() or "" for page in pdf_reader.pages])
-                    else:
-                        doc_text = uploaded_file.getvalue().decode("utf-8", errors="ignore")
-                    
-                    # Fire API Engine
+                    doc_text = safe_read_file(uploaded_file)
                     genai.configure(api_key=api_key)
                     model = genai.GenerativeModel('gemini-1.5-flash')
                     
@@ -127,20 +133,17 @@ if st.button("🚀 Generate Design Sheets & Flow"):
                     """
                     ai_response = model.generate_content(prompt).text
                     
-                    # Parse out string chunks safely
                     flow_match = re.search(r"<FLOW>(.*?)</FLOW>", ai_response, re.DOTALL)
                     table_match = re.search(r"<TABLE>(.*?)</TABLE>", ai_response, re.DOTALL)
                     
-                    flow_output = flow_match.group(1).strip().replace("```", "") if flow_match else "Journey breakdown completed."
-                    raw_table_str = table_match.group(1).strip().replace("```", "") if table_match else ""
+                    flow_output = flow_match.group(1).strip() if flow_match else "Journey breakdown completed."
+                    raw_table_str = table_match.group(1).strip() if table_match else ""
                     
-                    # Construct dataframe cleanly
                     table_lines = [l.strip() for l in raw_table_str.split('\n') if l.strip() and '---' not in l]
                     if table_lines:
                         headers = [h.strip() for h in table_lines[0].split('|')]
                         data_rows = [[cell.strip() for cell in r.split('|')] for r in table_lines[1:] if '|' in r]
                         
-                        # Adjust widths to keep columns uniform
                         clean_rows = []
                         for row in data_rows:
                             if len(row) < len(headers): row += ["NA"] * (len(headers) - len(row))
@@ -149,7 +152,7 @@ if st.button("🚀 Generate Design Sheets & Flow"):
                         dataframe_output = pd.DataFrame(clean_rows, columns=headers)
             
                 # ==========================================
-                # 4. RENDER PRESENTATION DELIVERABLES
+                # 4. RENDER OUTPUT INTERFACE
                 # ==========================================
                 st.success("🎯 Assets Computed Successfully!")
                 tab1, tab2 = st.tabs(["🗺️ Visual Journey Path", "📊 VA Design Sheet (Excel)"])
@@ -162,10 +165,8 @@ if st.button("🚀 Generate Design Sheets & Flow"):
                 with tab2:
                     st.subheader("Automated Routing Matrix (GTIS Mapping)")
                     if dataframe_output is not None:
-                        # Render interactive grid layout
                         st.dataframe(dataframe_output, use_container_width=True)
                         
-                        # Direct file download stream
                         csv_stream = dataframe_output.to_csv(index=False).encode('utf-8')
                         st.download_button(
                             label="⬇️ Download Sheet as CSV/Excel",
@@ -177,5 +178,4 @@ if st.button("🚀 Generate Design Sheets & Flow"):
                         st.warning("Data rows could not be arranged into a standard matrix framework layout.")
                         
             except Exception as runtime_err:
-                st.error(f"Execution Interrupted: {str(runtime_err)}")
-                st.info("💡 Presentation Hack: Simply toggle 'Activate Demo Simulation Mode' in the sidebar to bypass any technical blocks instantly.")
+                st.error(f"Execution Interrupted: {str(runtime_err)}")