Update src/streamlit_app.py

src/streamlit_app.py

@@ -19,34 +19,25 @@ def compile_latex(source: str) -> bytes:
     bytes
         The compiled PDF file contents.
     """
-    # Create a sandbox directory for all generated files
     with tempfile.TemporaryDirectory() as tmpdir:
-        # Initialize document with an arbitrary base name
-        doc = Document(
-            # This sets the base filename for output (without extension)
-            "document",
-            documentclass="article",
-            font_size="10pt",
-        )
+        # ドキュメントを初期化
+        doc = Document("document", documentclass="article", font_size="10pt")
         doc.append(source)
 
-        # Full path without extension
-        output_base = os.path.join(tmpdir, "document")
-        # Generate PDF; this writes 'document.pdf' into tmpdir
+        base_path = os.path.join(tmpdir, "document")
+        # compiler 引数で pdflatex を指定。latexmk があれば省略可。
         _ = doc.generate_pdf(
-            filepath=output_base,
-            clean=True,        # remove aux files like .aux, .log
-            clean_tex=True,    # also remove the .tex file
-            silent=True        # suppress compiler output
+            filepath=base_path,
+            compiler="pdflatex",
+            clean=True,
+            clean_tex=True,
+            silent=True,
         )
 
-        # Read the compiled PDF back into memory
-        pdf_path = Path(output_base + ".pdf")
-        return pdf_path.read_bytes()
+        return Path(base_path + ".pdf").read_bytes()
 
 def main():
-    """Streamlit entry point."""
-    st.title("LaTeX Compiler on Hugging Face Space")
+    st.title("LaTeX Compiler")
     prompt = st.text_area(
         "Enter full LaTeX source",
         value=(
@@ -60,8 +51,8 @@ def main():
 
     if st.button("Compile to PDF"):
         with st.spinner("Compiling…"):
-            pdf_bytes = compile_latex(source=prompt)
-        st.success("Done")
+            pdf_bytes = compile_latex(prompt)
+        st.success("Compilation succeeded")
         st.download_button(
             "Download PDF",
             data=pdf_bytes,