Spaces:
Sleeping
Sleeping
Update pages/artifacts.py
Browse files- pages/artifacts.py +4 -2
pages/artifacts.py
CHANGED
|
@@ -114,7 +114,8 @@ def handle_gen_conv_summary(style: str, state: UserData) -> tuple[UserData, str]
|
|
| 114 |
nb = get_active_notebook(state)
|
| 115 |
if not nb or not nb.messages:
|
| 116 |
return state, render_conv_summary_section(state)
|
| 117 |
-
|
|
|
|
| 118 |
nb.artifacts.append(artifact)
|
| 119 |
return state, render_conv_summary_section(state)
|
| 120 |
|
|
@@ -137,7 +138,8 @@ def handle_gen_doc_summary(style: str, state: UserData) -> tuple[UserData, str]:
|
|
| 137 |
nb = get_active_notebook(state)
|
| 138 |
if not nb:
|
| 139 |
return state, render_doc_summary_section(state)
|
| 140 |
-
|
|
|
|
| 141 |
nb.artifacts.append(artifact)
|
| 142 |
return state, render_doc_summary_section(state)
|
| 143 |
|
|
|
|
| 114 |
nb = get_active_notebook(state)
|
| 115 |
if not nb or not nb.messages:
|
| 116 |
return state, render_conv_summary_section(state)
|
| 117 |
+
from services.summary_service import generate_conversation_summary
|
| 118 |
+
artifact = generate_conversation_summary(nb, style or "detailed")
|
| 119 |
nb.artifacts.append(artifact)
|
| 120 |
return state, render_conv_summary_section(state)
|
| 121 |
|
|
|
|
| 138 |
nb = get_active_notebook(state)
|
| 139 |
if not nb:
|
| 140 |
return state, render_doc_summary_section(state)
|
| 141 |
+
from services.summary_service import generate_document_summary
|
| 142 |
+
artifact = generate_document_summary(nb, style or "detailed")
|
| 143 |
nb.artifacts.append(artifact)
|
| 144 |
return state, render_doc_summary_section(state)
|
| 145 |
|