fix: streamlit expander

app.py

@@ -329,14 +329,14 @@ def main():
             # Generate answer
             st.write("Analyzing research papers...")
             answer = model_handler.generate_answer(query, context)
+            status.write("I've got it!")
             
-            # Display sources
-            with st.expander("📚 View source papers"):
-                for paper in papers:
-                    st.markdown(f"- [{paper.title}]({paper.url}) ({paper.published})")
-            
-            st.success("Research analysis complete!")
-            st.markdown(answer)
+        with st.expander("📚 View source papers"):
+            for paper in papers:
+                st.markdown(f"- [{paper.title}]({paper.url}) ({paper.published})")
+        
+        st.success("Research analysis complete!")
+        st.markdown(answer)
 
 if __name__ == "__main__":
     main()