Upd PDF downloader + code snippet copier

app.py

@@ -430,11 +430,11 @@ async def upload_files(
     preloaded_files = []
     for uf in files:
         raw = await uf.read()
-        if len(raw) > max_mb * 1024 * 1024:
-            raise HTTPException(400, detail=f"{uf.filename} exceeds {max_mb} MB limit")
-        # Apply rename if present
-        eff_name = rename_dict.get(uf.filename, uf.filename)
-        preloaded_files.append((eff_name, raw))
+    if len(raw) > max_mb * 1024 * 1024:
+        raise HTTPException(400, detail=f"{uf.filename} exceeds {max_mb} MB limit")
+    # Apply rename if present
+    eff_name = rename_dict.get(uf.filename, uf.filename)
+    preloaded_files.append((eff_name, raw))
 
     # Initialize job status
     app.state.jobs[job_id] = {
@@ -473,7 +473,7 @@ async def upload_files(
                                 cap = captioner.caption_image(im)
                                 caps.append(cap)
                             except Exception as e:
-                                logger.warning(f"[{job_id}] Caption error in {fname}: {e}")
+                                    logger.warning(f"[{job_id}] Caption error in {fname}: {e}")
                         captions.append(caps)
                 else:
                     captions = [[] for _ in pages]
@@ -713,6 +713,36 @@ async def generate_report(
     return ReportResponse(filename=eff_name, report_markdown=report_md, sources=sources_meta)
 
 
+@app.post("/report/pdf")
+async def generate_report_pdf(
+    user_id: str = Form(...),
+    project_id: str = Form(...),
+    report_content: str = Form(...)
+):
+    """
+    Generate a PDF from report content using the PDF utility module
+    """
+    from utils.pdf import generate_report_pdf as generate_pdf
+    from fastapi.responses import Response
+    
+    try:
+        pdf_content = await generate_pdf(report_content, user_id, project_id)
+        
+        # Return PDF as response
+        return Response(
+            content=pdf_content,
+            media_type="application/pdf",
+            headers={"Content-Disposition": f"attachment; filename=report-{datetime.now().strftime('%Y-%m-%d')}.pdf"}
+        )
+        
+    except HTTPException:
+        # Re-raise HTTP exceptions as-is
+        raise
+    except Exception as e:
+        logger.error(f"[PDF] Unexpected error in PDF endpoint: {e}")
+        raise HTTPException(500, detail=f"Failed to generate PDF: {str(e)}")
+
+
 @app.post("/chat", response_model=ChatAnswerResponse)
 async def chat(
     user_id: str = Form(...), 
@@ -775,7 +805,7 @@ async def _chat_impl(
             match = next((f["filename"] for f in files_ci if f.get("filename", "").lower() == fn.lower()), None)
             if match:
                 doc = rag.get_file_summary(user_id=user_id, project_id=project_id, filename=match)
-                if doc:
+        if doc:
                     return ChatAnswerResponse(
                         answer=doc.get("summary", ""),
                         sources=[{"filename": match, "file_summary": True}]
@@ -935,7 +965,7 @@ async def _chat_impl(
         fsum_map = {f["filename"]: f.get("summary","") for f in files_list}
         lines = [f"[{fn}] {fsum_map.get(fn, '')}" for fn in relevant_files]
         file_summary_block = "\n".join(lines)
-        
+
     # Guardrail instruction to avoid hallucination
     system_prompt = (
         "You are a careful study assistant. Answer strictly using the given CONTEXT.\n"

requirements.txt

@@ -10,4 +10,6 @@ transformers==4.44.2
 torch==2.2.2
 sentence-transformers==3.1.1
 sumy==0.11.0
-numpy==1.26.4
+numpy==1.26.4
+weasyprint==62.3
+markdown==3.6

static/script.js

@@ -514,7 +514,7 @@
         const data = await response.json();
         if (response.ok) {
           thinkingMsg.remove();
-          appendMessage('assistant', data.report_markdown || 'No report');
+          appendMessage('assistant', data.report_markdown || 'No report', true); // isReport = true
           if (data.sources && data.sources.length) appendSources(data.sources);
           // Save assistant report to chat history for persistence
           try { await saveChatMessage(user.user_id, currentProject.project_id, 'assistant', data.report_markdown || 'No report'); } catch {}
@@ -656,7 +656,7 @@
     }
   }
   
-  function appendMessage(role, text) {
+  function appendMessage(role, text, isReport = false) {
     const messageDiv = document.createElement('div');
     messageDiv.className = `msg ${role}`;
     
@@ -666,6 +666,14 @@
         // Use marked library to convert Markdown to HTML
         const htmlContent = marked.parse(text);
         messageDiv.innerHTML = htmlContent;
+        
+        // Add copy buttons to code blocks
+        addCopyButtonsToCodeBlocks(messageDiv);
+        
+        // Add download PDF button for reports
+        if (isReport) {
+          addDownloadPdfButton(messageDiv, text);
+        }
       } catch (e) {
         // Fallback to plain text if Markdown parsing fails
         messageDiv.textContent = text;
@@ -683,6 +691,187 @@
     
     return messageDiv;
   }
+
+  function addCopyButtonsToCodeBlocks(messageDiv) {
+    const codeBlocks = messageDiv.querySelectorAll('pre code');
+    codeBlocks.forEach((codeBlock, index) => {
+      const pre = codeBlock.parentElement;
+      const language = codeBlock.className.match(/language-(\w+)/)?.[1] || 'code';
+      
+      // Create wrapper
+      const wrapper = document.createElement('div');
+      wrapper.className = 'code-block-wrapper';
+      
+      // Create header with language and copy button
+      const header = document.createElement('div');
+      header.className = 'code-block-header';
+      header.innerHTML = `
+        <span class="code-block-language">${language}</span>
+        <button class="copy-code-btn" data-code-index="${index}">
+          <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+            <rect x="9" y="9" width="13" height="13" rx="2" ry="2"></rect>
+            <path d="M5 15H4a2 2 0 0 1-2-2V4a2 2 0 0 1 2-2h9a2 2 0 0 1 2 2v1"></path>
+          </svg>
+          Copy
+        </button>
+      `;
+      
+      // Create content wrapper
+      const content = document.createElement('div');
+      content.className = 'code-block-content';
+      content.appendChild(codeBlock.cloneNode(true));
+      
+      // Assemble wrapper
+      wrapper.appendChild(header);
+      wrapper.appendChild(content);
+      
+      // Replace original pre with wrapper
+      pre.parentNode.replaceChild(wrapper, pre);
+      
+      // Add click handler for copy button
+      const copyBtn = wrapper.querySelector('.copy-code-btn');
+      copyBtn.addEventListener('click', () => copyCodeToClipboard(codeBlock.textContent, copyBtn));
+    });
+  }
+
+  function copyCodeToClipboard(code, button) {
+    navigator.clipboard.writeText(code).then(() => {
+      const originalText = button.innerHTML;
+      button.innerHTML = `
+        <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+          <polyline points="20,6 9,17 4,12"></polyline>
+        </svg>
+        Copied!
+      `;
+      button.classList.add('copied');
+      
+      setTimeout(() => {
+        button.innerHTML = originalText;
+        button.classList.remove('copied');
+      }, 2000);
+    }).catch(err => {
+      console.error('Failed to copy code:', err);
+      // Fallback for older browsers
+      const textArea = document.createElement('textarea');
+      textArea.value = code;
+      document.body.appendChild(textArea);
+      textArea.select();
+      try {
+        document.execCommand('copy');
+        button.innerHTML = `
+          <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+            <polyline points="20,6 9,17 4,12"></polyline>
+          </svg>
+          Copied!
+        `;
+        button.classList.add('copied');
+        setTimeout(() => {
+          button.innerHTML = `
+            <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+              <rect x="9" y="9" width="13" height="13" rx="2" ry="2"></rect>
+              <path d="M5 15H4a2 2 0 0 1-2-2V4a2 2 0 0 1 2-2h9a2 2 0 0 1 2 2v1"></path>
+            </svg>
+            Copy
+          `;
+          button.classList.remove('copied');
+        }, 2000);
+      } catch (fallbackErr) {
+        console.error('Fallback copy failed:', fallbackErr);
+      }
+      document.body.removeChild(textArea);
+    });
+  }
+
+  function addDownloadPdfButton(messageDiv, reportContent) {
+    const downloadBtn = document.createElement('button');
+    downloadBtn.className = 'download-pdf-btn';
+    downloadBtn.innerHTML = `
+      <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+        <path d="M21 15v4a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2v-4"></path>
+        <polyline points="7,10 12,15 17,10"></polyline>
+        <line x1="12" y1="15" x2="12" y2="3"></line>
+      </svg>
+      Download PDF
+    `;
+    
+    downloadBtn.addEventListener('click', () => downloadReportAsPdf(reportContent, downloadBtn));
+    messageDiv.appendChild(downloadBtn);
+  }
+
+  async function downloadReportAsPdf(reportContent, button) {
+    const user = window.__sb_get_user();
+    const currentProject = window.__sb_get_current_project && window.__sb_get_current_project();
+    
+    if (!user || !currentProject) {
+      alert('Please sign in and select a project');
+      return;
+    }
+
+    button.disabled = true;
+    button.innerHTML = `
+      <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+        <circle cx="12" cy="12" r="10"></circle>
+        <polyline points="12,6 12,12 16,14"></polyline>
+      </svg>
+      Generating PDF...
+    `;
+
+    try {
+      const formData = new FormData();
+      formData.append('user_id', user.user_id);
+      formData.append('project_id', currentProject.project_id);
+      formData.append('report_content', reportContent);
+
+      const response = await fetch('/report/pdf', {
+        method: 'POST',
+        body: formData
+      });
+
+      if (response.ok) {
+        const blob = await response.blob();
+        const url = window.URL.createObjectURL(blob);
+        const a = document.createElement('a');
+        a.href = url;
+        a.download = `report-${new Date().toISOString().split('T')[0]}.pdf`;
+        document.body.appendChild(a);
+        a.click();
+        window.URL.revokeObjectURL(url);
+        document.body.removeChild(a);
+        
+        button.innerHTML = `
+          <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+            <polyline points="20,6 9,17 4,12"></polyline>
+          </svg>
+          Downloaded!
+        `;
+        setTimeout(() => {
+          button.innerHTML = `
+            <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+              <path d="M21 15v4a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2v-4"></path>
+              <polyline points="7,10 12,15 17,10"></polyline>
+              <line x1="12" y1="15" x2="12" y2="3"></line>
+            </svg>
+            Download PDF
+          `;
+          button.disabled = false;
+        }, 2000);
+      } else {
+        throw new Error('Failed to generate PDF');
+      }
+    } catch (error) {
+      console.error('PDF generation failed:', error);
+      alert('Failed to generate PDF. Please try again.');
+      button.innerHTML = `
+        <svg viewBox="0 0 24 24" fill="none" stroke="currentColor" stroke-width="2">
+          <path d="M21 15v4a2 2 0 0 1-2 2H5a2 2 0 0 1-2-2v-4"></path>
+          <polyline points="7,10 12,15 17,10"></polyline>
+          <line x1="12" y1="15" x2="12" y2="3"></line>
+        </svg>
+        Download PDF
+      `;
+      button.disabled = false;
+    }
+  }
   
   function appendSources(sources) {
     const sourcesDiv = document.createElement('div');

static/styles.css

@@ -1018,6 +1018,105 @@
   overflow: auto;
 }
 
+/* Code blocks with copy button */
+.code-block-wrapper {
+  position: relative;
+  margin: 16px 0;
+  border-radius: var(--radius);
+  overflow: hidden;
+  border: 1px solid var(--border);
+}
+
+.code-block-header {
+  display: flex;
+  justify-content: space-between;
+  align-items: center;
+  padding: 8px 12px;
+  background: var(--bg-secondary);
+  border-bottom: 1px solid var(--border);
+  font-size: 12px;
+  color: var(--muted);
+}
+
+.code-block-language {
+  font-weight: 600;
+  text-transform: uppercase;
+}
+
+.copy-code-btn {
+  display: flex;
+  align-items: center;
+  gap: 4px;
+  padding: 4px 8px;
+  background: var(--card);
+  border: 1px solid var(--border);
+  border-radius: var(--radius-sm);
+  color: var(--text-secondary);
+  font-size: 11px;
+  cursor: pointer;
+  transition: all 0.2s ease;
+}
+
+.copy-code-btn:hover {
+  background: var(--card-hover);
+  border-color: var(--border-light);
+  color: var(--text);
+}
+
+.copy-code-btn.copied {
+  background: var(--success);
+  color: white;
+  border-color: var(--success);
+}
+
+.copy-code-btn svg {
+  width: 12px;
+  height: 12px;
+}
+
+.code-block-content {
+  background: var(--code-bg, #1e1e1e);
+  color: var(--code-text, #d4d4d4);
+  padding: 16px;
+  overflow-x: auto;
+  font-family: 'Monaco', 'Menlo', 'Ubuntu Mono', monospace;
+  font-size: 14px;
+  line-height: 1.5;
+}
+
+/* Download PDF button */
+.download-pdf-btn {
+  display: inline-flex;
+  align-items: center;
+  gap: 8px;
+  margin-top: 16px;
+  padding: 12px 20px;
+  background: var(--gradient-accent);
+  color: white;
+  border: none;
+  border-radius: var(--radius);
+  font-weight: 600;
+  cursor: pointer;
+  transition: all 0.2s ease;
+  box-shadow: var(--shadow);
+}
+
+.download-pdf-btn:hover {
+  transform: translateY(-1px);
+  box-shadow: var(--shadow-lg);
+}
+
+.download-pdf-btn:disabled {
+  opacity: 0.6;
+  cursor: not-allowed;
+  transform: none;
+}
+
+.download-pdf-btn svg {
+  width: 16px;
+  height: 16px;
+}
+
 /* Modal */
 .modal {
   position: fixed;

utils/__init__.py

@@ -0,0 +1 @@
+# Utils package for StudyBuddy

utils/pdf.py

@@ -0,0 +1,171 @@
+"""
+PDF generation utilities for StudyBuddy
+"""
+import os
+import tempfile
+import markdown
+from datetime import datetime
+from fastapi import HTTPException
+from utils.logger import get_logger
+
+logger = get_logger("PDF", __name__)
+
+
+async def generate_report_pdf(report_content: str, user_id: str, project_id: str) -> bytes:
+    """
+    Generate a PDF from report content using weasyprint
+    
+    Args:
+        report_content: Markdown content of the report
+        user_id: User ID for logging
+        project_id: Project ID for logging
+        
+    Returns:
+        PDF content as bytes
+        
+    Raises:
+        HTTPException: If PDF generation fails
+    """
+    try:
+        from weasyprint import HTML, CSS
+        from weasyprint.text.fonts import FontConfiguration
+        
+        logger.info(f"[PDF] Generating PDF for user {user_id}, project {project_id}")
+        
+        # Convert markdown to HTML
+        html_content = markdown.markdown(report_content, extensions=['codehilite', 'fenced_code'])
+        
+        # Create a complete HTML document with styling
+        full_html = f"""
+        <!DOCTYPE html>
+        <html>
+        <head>
+            <meta charset="utf-8">
+            <title>Study Report</title>
+            <style>
+                body {{
+                    font-family: -apple-system, BlinkMacSystemFont, 'Segoe UI', Roboto, sans-serif;
+                    line-height: 1.6;
+                    color: #333;
+                    max-width: 800px;
+                    margin: 0 auto;
+                    padding: 40px 20px;
+                }}
+                h1, h2, h3, h4, h5, h6 {{
+                    color: #2c3e50;
+                    margin-top: 2em;
+                    margin-bottom: 1em;
+                }}
+                h1 {{
+                    border-bottom: 2px solid #3498db;
+                    padding-bottom: 10px;
+                }}
+                h2 {{
+                    border-bottom: 1px solid #bdc3c7;
+                    padding-bottom: 5px;
+                }}
+                code {{
+                    background: #f8f9fa;
+                    padding: 2px 6px;
+                    border-radius: 3px;
+                    font-family: 'Monaco', 'Menlo', 'Ubuntu Mono', monospace;
+                    font-size: 0.9em;
+                }}
+                pre {{
+                    background: #2c3e50;
+                    color: #ecf0f1;
+                    padding: 20px;
+                    border-radius: 8px;
+                    overflow-x: auto;
+                    margin: 1em 0;
+                }}
+                pre code {{
+                    background: none;
+                    padding: 0;
+                    color: inherit;
+                }}
+                blockquote {{
+                    border-left: 4px solid #3498db;
+                    margin: 1em 0;
+                    padding-left: 20px;
+                    color: #7f8c8d;
+                }}
+                table {{
+                    border-collapse: collapse;
+                    width: 100%;
+                    margin: 1em 0;
+                }}
+                th, td {{
+                    border: 1px solid #bdc3c7;
+                    padding: 12px;
+                    text-align: left;
+                }}
+                th {{
+                    background: #ecf0f1;
+                    font-weight: 600;
+                }}
+                ul, ol {{
+                    padding-left: 2em;
+                }}
+                li {{
+                    margin: 0.5em 0;
+                }}
+                p {{
+                    margin: 1em 0;
+                }}
+                .page-break {{
+                    page-break-before: always;
+                }}
+                @page {{
+                    margin: 2cm;
+                    @bottom-center {{
+                        content: "Page " counter(page);
+                        font-size: 10px;
+                        color: #7f8c8d;
+                    }}
+                }}
+            </style>
+        </head>
+        <body>
+            <h1>Study Report</h1>
+            <p><em>Generated on {datetime.now().strftime('%B %d, %Y at %I:%M %p')}</em></p>
+            <hr>
+            {html_content}
+        </body>
+        </html>
+        """
+        
+        # Generate PDF
+        font_config = FontConfiguration()
+        html_doc = HTML(string=full_html)
+        css = CSS(string='', font_config=font_config)
+        
+        # Create temporary file for PDF
+        with tempfile.NamedTemporaryFile(delete=False, suffix='.pdf') as tmp_file:
+            pdf_path = tmp_file.name
+        
+        try:
+            html_doc.write_pdf(pdf_path, stylesheets=[css], font_config=font_config)
+            
+            # Read the generated PDF
+            with open(pdf_path, 'rb') as pdf_file:
+                pdf_content = pdf_file.read()
+            
+            # Clean up temporary file
+            os.unlink(pdf_path)
+            
+            logger.info(f"[PDF] Successfully generated PDF ({len(pdf_content)} bytes) for user {user_id}, project {project_id}")
+            return pdf_content
+            
+        except Exception as e:
+            # Clean up temporary file on error
+            if os.path.exists(pdf_path):
+                os.unlink(pdf_path)
+            raise e
+            
+    except ImportError:
+        logger.error("[PDF] weasyprint not installed. Install with: pip install weasyprint")
+        raise HTTPException(500, detail="PDF generation not available. Please install weasyprint.")
+    except Exception as e:
+        logger.error(f"[PDF] Failed to generate PDF: {e}")
+        raise HTTPException(500, detail=f"Failed to generate PDF: {str(e)}")