Update src/streamlit_app.py

src/streamlit_app.py

@@ -595,27 +595,36 @@ if st.session_state.execute_batch_analysis and 'json_data_for_batch' in st.sessi
                         "analysis_result": response,
                         "context": retrieved_ctx
                     }
-                                        
-                    st.session_state.batch_results.append(item)
-                                        
+                    fieldnames = list(item[0].keys()) if data_list else []
+                    output_string_buffer = io.StringIO()
+                    writer = csv.DictWriter(
+                        output_string_buffer, 
+                        fieldnames=fieldnames, 
+                    )
+                    writer.writeheader()
+                    writer.writerows(item)
+                    csv_string = output_string_buffer.getvalue()
+                        
                     with results_container:
 
-                                                
                         # 呈現 LLM 分析結果
                         is_high = any(x in response.lower() for x in ['high-risk detected!'])
                         is_medium = any(x in response.lower() for x in ['medium-risk detected!'])
                         if is_high: 
-                            st.subheader(f"Log #{item['log_id']} (HIGH RISK DETECTED)")
+                            b64 = base64.b64encode(csv_string.encode('utf-8'))                       
+                            st.markdown(f'<a href="data:application/octet-stream;base64,{b64.decode()}" download="Log_{item['log_id']}.csv">Log #{item['log_id']} (HIGH RISK DETECTED)</a>', unsafe_allow_html=True)
+                            
                             with st.expander("序列內容 (JSON Format)"):
                                 st.code(item["sequence_analyzed"], language='json')
                                 st.error(item['analysis_result'])
                                 st.markdown("---")
                         elif is_medium:
-                                st.subheader(f"Log #{item['log_id']} (MEDIUM RISK DETECTED)")
-                                with st.expander("序列內容 (JSON Format)"):
-                                    st.code(item["sequence_analyzed"], language='json')
-                                    st.warning(item['analysis_result'])
-                                    st.markdown("---")
+                            b64 = base64.b64encode(csv_string.encode('utf-8'))                       
+                            st.markdown(f'<a href="data:application/octet-stream;base64,{b64.decode()}" download="Log_{item['log_id']}.csv">Log #{item['log_id']} (MEDIUM RISK DETECTED)</a>', unsafe_allow_html=True)
+                            with st.expander("序列內容 (JSON Format)"):
+                                st.code(item["sequence_analyzed"], language='json')
+                                st.warning(item['analysis_result'])
+                                st.markdown("---")
 
                         if item['context']:
                             with st.expander("參考 RAG 片段"): st.code(item['context'])