Update main.py

main.py

@@ -66,13 +66,17 @@ css = Style("""
         margin-top: 20px;
     }
 
+    .download-links {
+        margin: 20px 0;
+    }
+
     .download-link {
         display: inline-block;
         padding: 10px 20px;
         background-color: #2c3e50;
         color: white;
         border-radius: 3px;
-        margin: 10px 0;
+        margin: 0 10px 10px 0;
         font-family: Georgia, Times, serif;
     }
 
@@ -90,75 +94,110 @@ app = FastHTML(hdrs=(css, MarkdownJS(),
 # Start the scheduler when the app starts
 @app.on_event("startup")
 async def start_scheduler():
-  scheduler.start()
+    scheduler.start()
 
 
 # Shut down the scheduler when the app stops
 @app.on_event("shutdown")
 async def shutdown_scheduler():
-  scheduler.shutdown()
+    scheduler.shutdown()
 
 
 def get_newsletter_list():
-  # Fetch the list of newsletters from the Hugging Face repository
-  files = api.list_repo_files(repo_id=DATASET_NAME, repo_type="dataset")
-  newsletters = [f for f in files if f.endswith('newsletter.json')]
-  return sorted(newsletters, reverse=True)
+    # Fetch the list of newsletters from the Hugging Face repository
+    files = api.list_repo_files(repo_id=DATASET_NAME, repo_type="dataset")
+    newsletters = [f for f in files if f.endswith('newsletter.json')]
+    return sorted(newsletters, reverse=True)
 
 
 def get_newsletter_content(path):
-  # Download and parse the newsletter content
-  content = api.hf_hub_download(repo_id=DATASET_NAME,
+    # Download and parse the newsletter content
+    content = api.hf_hub_download(repo_id=DATASET_NAME,
                                 filename=path,
                                 repo_type="dataset")
-  with open(content, 'r') as f:
-    return json.load(f)
+    with open(content, 'r') as f:
+        return json.load(f)
+
+
+def check_format_exists(date: str, format: str) -> bool:
+    """Check if a specific format exists for a given date"""
+    try:
+        api.hf_hub_download(
+            repo_id=DATASET_NAME,
+            filename=f"{date}/newsletter.{format}",
+            repo_type="dataset"
+        )
+        return True
+    except Exception:
+        return False
 
 
 @app.get("/")
 def index():
-  newsletters = get_newsletter_list()
-  links = [
-      Li(
-          A(datetime.strptime(n.split('/')[0], '%Y%m%d').strftime('%B %d, %Y'),
-            href=f"/newsletter/{n.split('/')[0]}")) for n in newsletters
-  ]
-  return Titled("This Week in Rheumatology", H2("Available Newsletters"),
-                Ul(*links))
+    newsletters = get_newsletter_list()
+    links = [
+        Li(
+            A(datetime.strptime(n.split('/')[0], '%Y%m%d').strftime('%B %d, %Y'),
+              href=f"/newsletter/{n.split('/')[0]}")) for n in newsletters
+    ]
+    return Titled("This Week in Rheumatology", H2("Available Newsletters"),
+                  Ul(*links))
 
 
 @app.get("/newsletter/{date}")
 def newsletter(date: str):
-  path = f"{date}/newsletter.json"
-  pdf_path = f"{date}/newsletter.pdf"
-  try:
-    content = get_newsletter_content(path)
-    return Titled(
-        f"This Week in Rheumatology - {content['date']}",
-        A("Back to Index", href="/"),
-        Div(
-        A("Download PDF", href=f"/download/{date}", cls="download-link")
-        ),
-        Div(content['content'], cls="marked"))
-  except Exception as e:
-    return Titled("Error", H2("Newsletter not found"),
-                  P(f"Unable to load newsletter for date: {date}"),
-                  A("Back to Index", href="/"))
-
-
-@app.get("/download/{date}")
-def download_pdf(date: str):
-  try:
-    pdf_path = f"{date}/newsletter.pdf"
-    content = api.hf_hub_download(repo_id=DATASET_NAME,
-                                  filename=pdf_path,
-                                  repo_type="dataset")
-    return FileResponse(content,
-                        media_type="application/pdf",
-                        filename=f"newsletter_{date}.pdf")
-  except Exception as e:
-    return Titled("Error", H2("PDF not found"),
-                  P(f"Unable to load PDF for date: {date}"),
-                  A("Back to Index", href="/"))
+    path = f"{date}/newsletter.json"
+    try:
+        content = get_newsletter_content(path)
+        
+        # Create download links div
+        download_links = []
+        
+        # Check for PDF
+        if check_format_exists(date, "pdf"):
+            download_links.append(
+                A("Download PDF", href=f"/download/{date}/pdf", cls="download-link")
+            )
+        
+        # Check for EPUB
+        if check_format_exists(date, "epub"):
+            download_links.append(
+                A("Download EPUB", href=f"/download/{date}/epub", cls="download-link")
+            )
+        
+        return Titled(
+            f"This Week in Rheumatology - {content['date']}",
+            A("Back to Index", href="/"),
+            Div(*download_links, cls="download-links"),
+            Div(content['content'], cls="marked"))
+    except Exception as e:
+        return Titled("Error", H2("Newsletter not found"),
+                    P(f"Unable to load newsletter for date: {date}"),
+                    A("Back to Index", href="/"))
+
+
+@app.get("/download/{date}/{format}")
+def download_file(date: str, format: str):
+    try:
+        file_path = f"{date}/newsletter.{format}"
+        content = api.hf_hub_download(repo_id=DATASET_NAME,
+                                    filename=file_path,
+                                    repo_type="dataset")
+        
+        # Set appropriate media type and filename
+        if format == "pdf":
+            media_type = "application/pdf"
+        elif format == "epub":
+            media_type = "application/epub+zip"
+        else:
+            raise ValueError(f"Unsupported format: {format}")
+            
+        return FileResponse(content,
+                          media_type=media_type,
+                          filename=f"newsletter_{date}.{format}")
+    except Exception as e:
+        return Titled("Error", H2(f"{format.upper()} not found"),
+                    P(f"Unable to load {format.upper()} for date: {date}"),
+                    A("Back to Index", href="/"))
 
 serve()