Update pages/artifacts.py

pages/artifacts.py

@@ -1,10 +1,35 @@
+
 """Artifacts tab: Summary, Podcast, and Quiz generation."""
 
+import os
+import tempfile
 from datetime import datetime
 from state import UserData, get_active_notebook, get_all_artifacts, get_latest_artifact
 from mock_data import generate_mock_artifact
 
 
+def _save_artifact_to_temp_file(artifact) -> str | None:
+    """Save artifact content to a temporary file. Returns the file path."""
+    try:
+        if artifact.type == "podcast" and artifact.audio_path:
+            if os.path.exists(artifact.audio_path):
+                return artifact.audio_path
+        else:
+            safe_title = artifact.title.replace(" ", "_")[:50]
+            # Save quizzes as HTML (they contain styled HTML), others as markdown
+            suffix = '.html' if artifact.type == "quiz" else '.md'
+            with tempfile.NamedTemporaryFile(
+                mode='w', suffix=suffix, prefix=f'{safe_title}_',
+                delete=False, encoding='utf-8'
+            ) as f:
+                f.write(artifact.content)
+                return f.name
+    except Exception as e:
+        import logging
+        logging.error(f"Failed to save artifact: {e}")
+    return None
+
+
 def _format_time(iso_str: str) -> str:
     try:
         dt = datetime.fromisoformat(iso_str)
@@ -36,7 +61,8 @@ def _render_artifact_content(artifact) -> str:
                 f'border-radius:10px;">'
                 f'<div style="font-size:0.82rem; color:#8090d0; margin-bottom:8px;">🎧 Podcast Audio</div>'
                 f'<audio controls style="width:100%; border-radius:6px;">'
-                f'<source src="/file={artifact.audio_path}" type="audio/wav">'
+                f'<source src="/gradio_api/file={artifact.audio_path}" type="audio/mpeg">'
+                f'<source src="/file={artifact.audio_path}" type="audio/mpeg">'
                 f'Your browser does not support the audio element.'
                 f'</audio>'
                 f'</div>'
@@ -47,10 +73,10 @@ def _render_artifact_content(artifact) -> str:
                 'background:rgba(102,126,234,0.06); border:1px solid rgba(102,126,234,0.15); '
                 'border-radius:10px; margin-top:8px;">'
                 '<span style="font-size:1.3rem;">🔇</span>'
-                '<span style="font-size:0.85rem; color:#8888aa;">Audio generation failed or Bark is not installed. '
-                'Install <code>bark</code> and <code>scipy</code> to enable TTS.</span>'
+                '<span style="font-size:0.85rem; color:#8888aa;">Audio generation failed. Verify <code>OPENAI_API_KEY</code> is set in Space secrets and your account has credits.</span>'
                 '</div>'
             )
+    
     return html
 
 
@@ -96,17 +122,20 @@ def render_conv_summary_section(state: UserData) -> str:
     if not nb:
         return ""
     if not nb.messages:
-        return "*No conversation yet. Start chatting in the **Chat** tab first.*"
+        return (
+            '<div style="text-align:center; padding:30px 20px; color:#606078; '
+            'border:1px dashed rgba(255,255,255,0.08); border-radius:14px;">'
+            '<p style="margin:0;">No conversation yet. Start chatting in the <strong>Chat</strong> tab first.</p>'
+            '</div>'
+        )
     summaries = get_all_artifacts(nb, "conversation_summary")
     if not summaries:
-        return "*Click Generate to create a conversation summary.*"
-    latest = summaries[0]
-    time_str = _format_time(latest.created_at)
-    md = "*Generated " + time_str + "*\n\n---\n\n" + latest.content
-    if len(summaries) > 1:
-        prev = ", ".join(_format_time(a.created_at) for a in summaries[1:])
-        md += "\n\n---\n\n**Previous summaries:** " + prev
-    return md
+        return '<p style="color:#606078; text-align:center; padding:20px;">Click "Generate" to create a conversation summary.</p>'
+    html = _render_artifact_content(summaries[0])
+    html += _render_history(summaries, "conversation summaries")
+    return html
+
+
 def handle_gen_conv_summary(style: str, state: UserData) -> tuple[UserData, str]:
     nb = get_active_notebook(state)
     if not nb or not nb.messages:
@@ -125,14 +154,12 @@ def render_doc_summary_section(state: UserData) -> str:
         return ""
     summaries = get_all_artifacts(nb, "document_summary")
     if not summaries:
-        return "*Click Generate to create a document summary.*"
-    latest = summaries[0]
-    time_str = _format_time(latest.created_at)
-    md = "*Generated " + time_str + "*\n\n---\n\n" + latest.content
-    if len(summaries) > 1:
-        prev = ", ".join(_format_time(a.created_at) for a in summaries[1:])
-        md += "\n\n---\n\n**Previous summaries:** " + prev
-    return md
+        return '<p style="color:#606078; text-align:center; padding:20px;">Click "Generate" to create a document summary.</p>'
+    html = _render_artifact_content(summaries[0])
+    html += _render_history(summaries, "document summaries")
+    return html
+
+
 def handle_gen_doc_summary(style: str, state: UserData) -> tuple[UserData, str]:
     nb = get_active_notebook(state)
     if not nb:
@@ -263,4 +290,27 @@ def handle_gen_quiz(num_questions: int, state: UserData) -> tuple[UserData, str]
     num_q = int(num_questions) if num_questions else 5
     artifact = generate_quiz(nb, num_q)
     nb.artifacts.append(artifact)
-    return state, render_quiz_section(state)
+    return state, render_quiz_section(state)
+
+
+# ── Download Handlers ────────────────────────────────────────────────────────
+
+def _find_artifact_by_id(state: UserData, artifact_id: str):
+    """Find an artifact by ID across all notebooks (current one only for now)."""
+    nb = get_active_notebook(state)
+    if not nb:
+        return None
+    for artifact in nb.artifacts:
+        if artifact.id == artifact_id:
+            return artifact
+    return None
+
+
+def download_artifact(artifact_id: str, state: UserData) -> str | None:
+    """Download an artifact and return the file path for Gradio download."""
+    artifact = _find_artifact_by_id(state, artifact_id)
+    if not artifact:
+        return None
+    
+    file_path = _save_artifact_to_temp_file(artifact)
+    return file_path