Update app.py

app.py

@@ -284,7 +284,7 @@ class ResearchWorkflow:
         
         response = self.processor.process_query(validation_prompt)
         return {
-            "messages": [AIMessage(content=analysis + f"\n\nValidation: {response.get('choices', [{}])[0].get('message', {}).get('content', '')}")]
+            "messages": [AIMessage(content=analysis + f"\n\nValidation: {response.get('choices', [{}])[0].get('message', {}).get('content', '')}"]
         }
 
     def refine_results(self, state: AgentState) -> Dict:
@@ -398,41 +398,55 @@ class ResearchInterface:
             self._execute_analysis(query)
 
     def _execute_analysis(self, query: str):
-        with st.status("Processing Quantum Analysis...", expanded=True) as status:
-            try:
+        try:
+            with st.spinner("Initializing Quantum Analysis..."):
                 results = self.workflow.app.stream(
                     {"messages": [HumanMessage(content=query)], "context": {}, "metadata": {}}
                 )
                 
-                container = st.container()
                 for event in results:
-                    self._render_event(event, container)
+                    self._render_event(event)
                 
-                status.update(label="Analysis Complete", state="complete")
-            except Exception as e:
-                st.error(f"""**Analysis Failed**
-                {str(e)}
-                Potential issues:
-                - API connectivity
-                - Query complexity
-                - System overload""")
+                st.success("✅ Analysis Completed Successfully")
+        except Exception as e:
+            st.error(f"""**Analysis Failed**
+            {str(e)}
+            Potential issues:
+            - Complex query structure
+            - Document correlation failure
+            - Temporal processing constraints""")
 
-    def _render_event(self, event: Dict, container):
+    def _render_event(self, event: Dict):
         if 'ingest' in event:
-            with container:
+            with st.container():
                 st.success("✅ Query Ingested")
+                
         elif 'retrieve' in event:
-            with container:
-                st.info(f"📚 Retrieved {len(event['retrieve']['context']['documents'])} documents")
+            with st.container():
+                docs = event['retrieve']['context']['documents']
+                st.info(f"📚 Retrieved {len(docs)} documents")
+                with st.expander("View Retrieved Documents", expanded=False):
+                    for i, doc in enumerate(docs, 1):
+                        st.markdown(f"**Document {i}**")
+                        st.code(doc.page_content, language='text')
+                        
         elif 'analyze' in event:
-            with container.expander("Technical Analysis", expanded=True):
-                st.markdown(event['analyze']['messages'][0].content)
+            with st.container():
+                content = event['analyze']['messages'][0].content
+                with st.expander("Technical Analysis Report", expanded=True):
+                    st.markdown(content)
+                    
         elif 'validate' in event:
-            with container:
-                if "VALID" in event['validate']['messages'][0].content:
+            with st.container():
+                content = event['validate']['messages'][0].content
+                if "VALID" in content:
                     st.success("✅ Validation Passed")
+                    with st.expander("View Validated Analysis", expanded=True):
+                        st.markdown(content.split("Validation:")[0])
                 else:
                     st.warning("⚠️ Validation Issues Detected")
+                    with st.expander("View Validation Details", expanded=True):
+                        st.markdown(content)
 
 if __name__ == "__main__":
     ResearchInterface()