Spaces:
Sleeping
Sleeping
File size: 2,320 Bytes
fb4d9a7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
# components/notebook_ui.py
import streamlit as st
from typing import Dict, List, Optional
import json
class NotebookUI:
def __init__(self):
if 'notebook_cells' not in st.session_state:
st.session_state.notebook_cells = []
def add_cell(self, cell_type: str, content: Dict, metadata: Optional[Dict] = None):
"""Add a new cell to the notebook"""
cell = {
'type': cell_type,
'content': content,
'metadata': metadata or {},
'id': len(st.session_state.notebook_cells)
}
st.session_state.notebook_cells.append(cell)
def render_cells(self):
"""Render all notebook cells"""
for cell in st.session_state.notebook_cells:
self._render_cell(cell)
def _render_cell(self, cell: Dict):
"""Render a single cell"""
with st.container():
with st.expander(f"Cell {cell['id'] + 1}", expanded=True):
if cell['type'] == 'query':
self._render_query_cell(cell)
elif cell['type'] == 'analysis':
self._render_analysis_cell(cell)
elif cell['type'] == 'document':
self._render_document_cell(cell)
def _render_query_cell(self, cell: Dict):
st.markdown("**Query:**")
st.write(cell['content']['query'])
st.markdown("**Response:**")
st.write(cell['content']['response'])
col1, col2 = st.columns([1, 4])
with col1:
if st.button("♻️ Regenerate", key=f"regen_{cell['id']}"):
pass # Implement regeneration logic
if st.button("📋 Copy", key=f"copy_{cell['id']}"):
st.toast("Copied to clipboard!")
def _render_analysis_cell(self, cell: Dict):
st.markdown(f"**{cell['metadata'].get('template', 'Analysis')}**")
for section in cell['content']:
st.markdown(f"### {section['title']}")
st.write(section['content'])
def _render_document_cell(self, cell: Dict):
st.markdown(f"**Document: {cell['metadata'].get('filename', 'Unknown')}**")
with st.expander("Document Content"):
st.write(cell['content']['text'])
st.markdown("### Metadata")
st.json(cell['metadata']) |