Update app.py

app.py

@@ -8,6 +8,68 @@ import shutil
 
 app = FastAPI()
 
+
+EPUB_CSS = """
+body { 
+    font-family: serif; 
+    line-height: 1.5; 
+    margin: 5%; 
+    color: #e0e0e0; 
+    background-color: #1a1a1a; 
+}
+h1 { 
+    text-align: center; 
+    color: #f4a261; 
+    text-transform: uppercase; 
+    margin-bottom: 0.2em; 
+}
+h3 { 
+    border-bottom: 1px solid #333; 
+    padding-bottom: 5px; 
+    margin-top: 30px; 
+    color: #e76f51; 
+}
+blockquote { 
+    font-style: italic; 
+    border-left: 3px solid #e76f51; 
+    padding-left: 15px; 
+    color: #b0b0b0; 
+    margin: 1.5em 10px; 
+}
+li { 
+    margin-bottom: 8px; 
+}
+table {
+    width: 100%;
+    border-collapse: separate;
+    border-spacing: 0;
+    margin: 20px 0;
+    border: 1px solid #333;
+    border-radius: 8px;
+    overflow: hidden;
+}
+th {
+    background-color: #2d2d2d;
+    color: #f4a261;
+    font-weight: bold;
+    text-align: left;
+    padding: 12px;
+    border-bottom: 2px solid #3d3d3d;
+}
+td {
+    padding: 10px 12px;
+    border-bottom: 1px solid #2d2d2d;
+    vertical-align: top;
+    font-size: 0.95em;
+}
+tr:last-child td {
+    border-bottom: none;
+}
+tr:nth-child(even) {
+    background-color: #222222;
+}
+"""
+
 @app.post("/extract")
 async def extract(file: UploadFile = File(...)):
     pdf_bytes = await file.read()
@@ -64,6 +126,57 @@ async def convert_to_markdown(file: UploadFile = File(...)):
         if os.path.exists(temp_path):
             os.remove(temp_path)
 
+@app.post("/export-epub")
+async def export_epub(file: UploadFile = File(...)):
+    base_name = os.path.splitext(file.filename)[0]
+    temp_input = f"temp_{file.filename}"
+    temp_css = "style.css"
+    output_epub = f"{base_name}.epub"
+    
+    try:
+        # 1. Save the upload
+        with open(temp_input, "wb") as buffer:
+            content = await file.read()
+            buffer.write(content)
+        
+        # 2. Use MarkItDown to get the Markdown content
+        result = md.convert(temp_input)
+        markdown_content = result.text_content
+        
+        # 3. Create a temporary CSS file for Pandoc
+        with open(temp_css, "w") as f:
+            f.write(EPUB_CSS)
+        
+        # 4. Call Pandoc to convert Markdown string to EPUB
+        # We pipe the markdown_content directly into pandoc's stdin
+        process = subprocess.Popen(
+            ['pandoc', '--from=markdown', '--to=epub', '--css', temp_css, 
+             '--metadata', f'title={base_name}', '-o', output_epub],
+            stdin=subprocess.PIPE,
+            stderr=subprocess.PIPE,
+            text=True
+        )
+        stdout, stderr = process.communicate(input=markdown_content)
+        
+        if process.returncode != 0:
+            raise Exception(f"Pandoc Error: {stderr}")
+
+        # 5. Return the generated EPUB file
+        return FileResponse(
+            path=output_epub, 
+            filename=output_epub, 
+            media_type='application/epub+zip'
+        )
+
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=str(e))
+    
+    finally:
+        # Cleanup
+        for path in [temp_input, temp_css]:
+            if os.path.exists(path):
+                os.remove(path)
+
 @app.get("/health")
 async def health():
     return {