Update src/streamlit_app.py

src/streamlit_app.py

@@ -10,22 +10,30 @@ from reportlab.lib.styles import getSampleStyleSheet, ParagraphStyle
 from reportlab.lib.units import inch
 from reportlab.lib import colors
 
-config: dict = {'api-url': 'https://chat.voxio.in/agents'}
+config : dict = {'api-url' : 'https://chat.voxio.in/agents'}
 API_KEY = 'vk_JZgdwJHVmqsWeNvO9b2QvcThfUxl16moNr3wxoDyf2QER97rpi5w3XJCL0H9NRKE'
 
-st.set_page_config(layout='wide')
+st.set_page_config(layout = 'wide')
 
-for key, default in [
-    ('messages', []),
-    ('session_id', None),
-    ('waiting_for_var', None),
-    ('actions', []),
-    ('end_diagnosis_reached', False),
-    ('summary', ''),
-    ('llm_response', ''),
-]:
-    if key not in st.session_state:
-        st.session_state[key] = default
+if 'messages' not in st.session_state :
+    st.session_state.messages = []
+
+if 'session_id' not in st.session_state :
+    st.session_state.session_id = None
+
+if 'waiting_for_var' not in st.session_state :
+    st.session_state.waiting_for_var = None
+
+# ── NEW: only these two additions ─────────────────────────────────────────────
+if 'actions' not in st.session_state :
+    st.session_state.actions = []
+
+if 'llm_response' not in st.session_state :
+    st.session_state.llm_response = None
+
+if 'summary' not in st.session_state :
+    st.session_state.summary = None
+# ─────────────────────────────────────────────────────────────────────────────
 
 
 def generate_pdf(summary: str, feedback: str) -> BytesIO:
@@ -34,202 +42,201 @@ def generate_pdf(summary: str, feedback: str) -> BytesIO:
                             topMargin=0.75*inch, bottomMargin=0.75*inch,
                             leftMargin=inch, rightMargin=inch)
     styles = getSampleStyleSheet()
-
-    title_style = ParagraphStyle('Title', parent=styles['Heading1'],
-                                 fontSize=22, textColor=colors.HexColor('#1a56db'),
-                                 spaceAfter=6, alignment=1)
-    meta_style = ParagraphStyle('Meta', parent=styles['Normal'],
-                                fontSize=9, textColor=colors.grey, alignment=1,
-                                spaceAfter=20)
-    h2_style = ParagraphStyle('H2', parent=styles['Heading2'],
-                              fontSize=14, textColor=colors.HexColor('#1a56db'),
-                              spaceBefore=16, spaceAfter=8)
-    body_style = ParagraphStyle('Body', parent=styles['Normal'],
-                                fontSize=11, leading=17)
-
-    story = [
-        Paragraph("Session Evaluation Report", title_style),
-        Paragraph(f"Generated: {datetime.now().strftime('%d %B %Y, %H:%M')}", meta_style),
-        Paragraph("Summary", h2_style),
-    ]
-    for line in (summary or "No summary available.").split('\n'):
+    h1 = ParagraphStyle('H1', parent=styles['Heading1'], fontSize=20,
+                         textColor=colors.HexColor('#1a56db'), spaceAfter=6)
+    body = ParagraphStyle('Body', parent=styles['Normal'], fontSize=11, leading=17)
+
+    story = []
+    story.append(Paragraph("# Summary", h1))
+    story.append(Spacer(1, 6))
+    for line in (summary or '').split('\n'):
         if line.strip():
-            story.append(Paragraph(line.strip(), body_style))
-            story.append(Spacer(1, 4))
-
-    story.append(Spacer(1, 8))
-    story.append(Paragraph("Feedback", h2_style))
-    for line in (feedback or "No feedback available.").split('\n'):
+            story.append(Paragraph(line.strip(), body))
+    story.append(Spacer(1, 20))
+    story.append(Paragraph("# Feedback", h1))
+    story.append(Spacer(1, 6))
+    for line in (feedback or '').split('\n'):
         if line.strip():
-            story.append(Paragraph(line.strip(), body_style))
-            story.append(Spacer(1, 4))
+            story.append(Paragraph(line.strip(), body))
 
     doc.build(story)
     buf.seek(0)
     return buf
 
 
-def process_stream(response):
-    for line in response.iter_lines():
-        if not line:
-            continue
-        try:
-            data: dict = json.loads(line)
-        except json.JSONDecodeError:
-            continue
-
-        # Handle errors first — these are the only early exits
-        if 'error' in data:
-            st.error(f'Server Error: {data["error"]}')
-            st.session_state.messages.append({'role': 'error', 'content': data['error']})
-            continue
-        if 'detail' in data:
-            st.error(f'API Error: {data["detail"]}')
-            st.session_state.messages.append({'role': 'error', 'content': str(data['detail'])})
-            continue
-
-        # ── Process every field independently — NO early continue ──────────────
-
-        # 1. Spoken NPC content — always capture this first
-        if data.get('speak'):
-            st.session_state.messages.append({
-                'role': 'assistant',
-                'content': data['speak']
-            })
-
-        # 2. Input gate — agent is waiting for trainee response
-        if data.get('waiting_for_input'):
-            st.session_state.waiting_for_var = data.get('required_variable')
-
-        # 3. Actions (accumulate across turns)
-        if data.get('actions'):
-            for a in data['actions']:
-                if a and a not in st.session_state.actions:
-                    st.session_state.actions.append(a)
-
-        # 4. Session end flag
-        if str(data.get('end_diagnosis', '')).lower() in ['yes', 'true', '1']:
-            st.session_state.end_diagnosis_reached = True
-
-        # 5. Summary from API
-        if data.get('summary'):
-            st.session_state.summary = data['summary']
-
-        # 6. LLM evaluation block
-        if data.get('llm_response'):
-            st.session_state.llm_response = data['llm_response']
-
-
-def reset_session():
-    for key in ('session_id', 'waiting_for_var', 'summary', 'llm_response'):
-        st.session_state[key] = None if key in ('session_id', 'waiting_for_var') else ''
-    st.session_state.messages = []
-    st.session_state.actions = []
-    st.session_state.end_diagnosis_reached = False
+def process_stream(response) :
 
+    for line in response.iter_lines() :
 
-# ── Sidebar ────────────────────────────────────────────────────────────────────
-with st.sidebar:
-    st.title("⚙️ Session Settings")
+        if line :
 
-    if st.session_state.session_id:
-        st.info('🔗 Session active')
+            try :
 
-        if st.session_state.actions:
-            st.divider()
-            st.caption("ACTIONS TRIGGERED")
-            for action in st.session_state.actions:
-                st.success(f"✅ `{action}`")
+                data : dict = json.loads(line)
 
-        if st.session_state.end_diagnosis_reached:
-            st.warning("⏹️ Session ended by agent")
+                if 'error' in data :
 
-        st.divider()
-        if st.button('End Session', type='primary'):
-            reset_session()
+                    st.error(f'Server Error: {data["error"]}')
+                    st.session_state.messages.append({'role' : 'error' , 'content' : data['error']})
+
+                elif 'detail' in data :
+
+                    st.error(f'API Error: {data["detail"]}')
+                    st.session_state.messages.append({'role' : 'error' , 'content' : str(data['detail'])})
+
+                elif data.get('waiting_for_input') :
+
+                    req_var = data.get('required_variable')
+                    st.session_state.waiting_for_var = req_var
+
+                else :
+
+                    content = data.get('out' , data)
+                    st.session_state.messages.append({'role' : 'assistant' , 'content' : str(content)})
+
+                    # ── NEW: capture actions, summary, llm_response ────────
+                    if data.get('actions') :
+                        for a in data['actions'] :
+                            if a and a not in st.session_state.actions :
+                                st.session_state.actions.append(a)
+
+                    if data.get('summary') :
+                        st.session_state.summary = data['summary']
+
+                    if data.get('llm_response') :
+                        st.session_state.llm_response = data['llm_response']
+                    # ─────────────────────────────────────────────────────────
+
+            except json.JSONDecodeError :
+                st.error(f'Failed to decode JSON line: {line}')
+
+
+with st.sidebar :
+
+    st.title("Settings")
+
+    if st.session_state.session_id :
+
+        st.info(f'Active Session: `{st.session_state.session_id}`')
+
+        # ── NEW: show actions ──────────────────────────────────────────────
+        if st.session_state.actions :
+            st.divider()
+            for a in st.session_state.actions :
+                st.warning(f'⚡ Action: `{a}`')
+        # ──────────────────────────────────────────────────────────────────
+
+        if st.button('End Session' , type = 'primary') :
+
+            st.session_state.session_id = None
+            st.session_state.messages = []
+            st.session_state.waiting_for_var = None
+            st.session_state.actions = []
+            st.session_state.llm_response = None
+            st.session_state.summary = None
             st.rerun()
-    else:
+
+    else :
         st.write("No active session.")
 
-    # PDF report section — appears once end_diagnosis fires
-    if st.session_state.end_diagnosis_reached and (
-        st.session_state.summary or st.session_state.llm_response
-    ):
+    # ── NEW: PDF download once llm_response arrives ────────────────────────
+    if st.session_state.llm_response :
         st.divider()
-        st.subheader("📄 Evaluation Report")
-        pdf_bytes = generate_pdf(st.session_state.summary, st.session_state.llm_response)
+        st.subheader("📄 Report Ready")
+        pdf_bytes = generate_pdf(st.session_state.summary or '', st.session_state.llm_response)
         st.download_button(
             label="📥 Download PDF",
             data=pdf_bytes,
             file_name=f"report_{datetime.now().strftime('%Y%m%d_%H%M')}.pdf",
             mime="application/pdf",
         )
-        with st.expander("👁️ Preview Report"):
-            if st.session_state.summary:
-                st.markdown("### Summary")
+        with st.expander("👁️ Preview") :
+            if st.session_state.summary :
+                st.markdown("## Summary")
                 st.markdown(st.session_state.summary)
-            if st.session_state.llm_response:
-                st.markdown("### Feedback")
-                st.markdown(st.session_state.llm_response)
+            st.markdown("## Feedback")
+            st.markdown(st.session_state.llm_response)
+    # ──────────────────────────────────────────────────────────────────────
 
 
-# ── Start Session ──────────────────────────────────────────────────────────────
-if not st.session_state.session_id:
-    if st.button("🚀 Start Session", type="primary"):
-        new_id = str(uuid4())
-        payload = {"session_id": new_id, "api_key": API_KEY}
-        st.session_state.session_id = new_id
+if not st.session_state.session_id :
+
+    if st.button("🚀 Start Session" , type = "primary") :
+
+        payload = {
+            "session_id" : str(uuid4()) ,
+            "api_key" : API_KEY
+        }
+
+        st.session_state.session_id = payload['session_id']
+
+        with st.spinner('Initializing Workflow...') :
+
+            try :
+
+                with requests.post(f'{config["api-url"]}/start_session', json=payload, stream=True) as r:
 
-        with st.spinner('Initializing…'):
-            try:
-                with requests.post(
-                    f'{config["api-url"]}/start_session',
-                    json=payload, stream=True
-                ) as r:
                     if r.status_code == 200:
                         process_stream(r)
                         st.rerun()
-                    else:
+
+                    else :
                         st.error(f"Error {r.status_code}: {r.text}")
                         st.session_state.session_id = None
-            except Exception as e:
+
+            except Exception as e :
+
                 st.error(f"Connection Error: {e}")
                 st.session_state.session_id = None
 
+chat_container = st.container()
 
-# ── Chat Display ───────────────────────────────────────────────────────────────
-for msg in st.session_state.messages:
-    with st.chat_message(msg['role']):
-        st.write(msg['content'])
+with chat_container :
 
+    for msg in st.session_state.messages :
 
-# ── Input ──────────────────────────────────────────────────────────────────────
-if st.session_state.session_id:
-    if st.session_state.waiting_for_var:
-        with st.chat_message("assistant"):
-            st.write(f"Waiting for your response…")
+        with st.chat_message(msg['role']) :
 
-    if user_text := st.chat_input("Enter response..."):
-        st.session_state.messages.append({'role': 'user', 'content': user_text})
+            if msg['role'] == 'assistant' :
+                st.write(msg['content'])
+
+            else :
+                st.write(msg['content'])
+
+if st.session_state.session_id and st.session_state.waiting_for_var :
+
+    with st.chat_message("assistant") :
+
+        st.write(f"I need a value for: **{st.session_state.waiting_for_var}**")
+
+    if user_text := st.chat_input("Enter response...") :
+
+        st.session_state.messages.append({'role' : 'user' , 'content' : user_text})
 
         payload = {
-            'session_id': st.session_state.session_id,
-            'user_input': {st.session_state.waiting_for_var: user_text}
-            if st.session_state.waiting_for_var
-            else user_text
+            'session_id' : st.session_state.session_id ,
+            'user_input' : {st.session_state.waiting_for_var : user_text}
         }
+
         st.session_state.waiting_for_var = None
 
-        with st.spinner('Processing…'):
-            try:
+        with st.spinner('Processing...') :
+
+            try :
+
                 with requests.post(
-                    f'{config["api-url"]}/process_input',
-                    json=payload, stream=True
-                ) as r:
-                    if r.status_code == 200:
+                    f'{config["api-url"]}/process_input' ,
+                    json = payload ,
+                    stream = True
+                ) as r :
+
+                    if r.status_code == 200 :
+
                         process_stream(r)
                         st.rerun()
-                    else:
+
+                    else :
                         st.error("Server Error processing input.")
-            except Exception as e:
+
+            except Exception as e :
                 st.error(f"Connection Error: {e}")