fix: lock model

app.py

@@ -1,23 +1,120 @@
-from os import getenv
+from os import getenv, system
 from textwrap import dedent
 from torch import cuda
+from functools import partial
 
+from tempfile import NamedTemporaryFile
 import gradio as gr
-from src.automatikz.examples.webui.webui import build_ui, get_banner
+from automatikz.infer import TikzDocument
+import fitz
+#from src.automatikz.examples.webui.webui import get_banner
 
-PUBLIC_DEMO = getenv("SPACE_ID") == "nllg/AutomaTikZ"
+tikz=r"""\documentclass[tikz,border=3mm]{standalone}
+\usepackage{pgfplots}
+\pgfplotsset{compat=1.16}
+\usepgfplotslibrary{colormaps}
+\tikzset{pics/2d-grid/.style 2 args={code={%
+   \pgfmathtruncatemacro{\mymax}{max(#1,#2)}
+   \draw[blue,thick] (0,0) grid[xstep=0.2,ystep=0.2] (\mymax,0.1);
+   \foreach \x in {0.1,0.3,...,\mymax}
+     \node[below,font=\sffamily] at (\x,0) {\pgfmathprintnumber{\x}};
+   \foreach \y in {0.1,0.2,...,\mymax}
+     \node[left,font=\sffamily] at (0.2,\y) {\pgfmathprintnumber{\y}};
+   \foreach \sh in {0.05,0.1,...,#1}
+     \foreach \ch in {0.05,0.1,...,#2}   
+     \path[blue,fill=blue!30] (\sh,\ch) rectangle (\sh+0.05,\ch+0.05);
+}}}
+\begin{document}
+\begin{tikzpicture}[scale=1.5]
+ \begin{axis}[
+    view={35}{40},
+    grid=major,
+    colormap/violet,
+    zlabel=$x_1$,
+    ylabel=$x_2$,
+    xlabel=$x_3$,
+    zlabel at={(3.6,2.)},
+    ylabel at={(2.5,3.6)},
+    zlabel shift={(0pt, -12pt)},
+    set layers=standard,
+    axis on top,
+    legend style={font=\small,legend cell align=right,
+        legend plot pos=right,
+        legend style={draw=none,
+        /tikz/every even column/.append style={column sep=0.35cm}},
+    },
+ ]       
+  \begin{scope}[on background layer]
+    \pic{2d-grid={17}{7.2}};
+  \end{scope}
+  \begin{scope}[on background layer]
+    \pic{2d-grid={12}{1.5}};
+  \end{scope}
+  \begin{scope}[on background layer]
+    \pic{2d-grid={9}{0.3}};
+  \end{scope}
+  \begin{scope}[on background layer]
+    \pic{2d-grid={9}{0.3}};
+  \end{scope}
+  \begin{scope}[on background layer]
+    \pic{2d-grid={4.2}{0.4}};
+  \end{scope}
+  \begin{scope}[on background layer]
+    \pic{2d-grid={4.2}{0.4}};
+  \end{scope}
+  \begin{scope}[on background layer]
+    \pic{2d-grid={4.2}{0.4}};
+  \end{scope}
+  \begin{scope}[on background layer]
+    \pic{2d-grid={4.2}{0.4}};
+  \end{scope}
+  \begin{scope}[on background layer]
+    \pic{2d-grid={4.2}{0.4}};
+  \end{scope}
+    %    
+ \end{axis}
+\end{tikzpicture}
+\end{document}"""
 
-if PUBLIC_DEMO and not cuda.is_available():
-    with gr.Blocks(css=".gradio-container {text-align: center}", theme=gr.themes.Soft()) as demo:
-        badge = "https://huggingface.co/datasets/huggingface/badges/resolve/main/duplicate-this-space-xl.svg"
-        link = "https://huggingface.co/spaces/nllg/AutomaTikZ?duplicate=true"
-        html = f'<a style="display:inline-block" href="{link}"> <img src="{badge}" alt="Duplicate this Space"> </a>'
-        message = dedent("""\
-        The size of our models exceeds the resource constraints offered by the
-        free tier of Hugging Face Spaces. For full functionality, we recommend
-        duplicating this space on a paid private GPU runtime.
-        """)
-        gr.Markdown(f'{get_banner()}\n{message}\n{html}')
-        demo.launch()
-else:
-    build_ui().queue().launch()
+def convert_to_svg(pdf):
+    doc = fitz.open("pdf", pdf.raw) # type: ignore
+    return doc[0].get_svg_image()
+
+def tex_compile(
+    code: str,
+    timeout: int = 120,
+    rasterize: bool = False
+):
+    tikzdoc = TikzDocument(code, timeout=timeout)
+    if not tikzdoc.has_content:
+        if tikzdoc.compiled_with_errors:
+            raise gr.Error("TikZ code did not compile!") # type: ignore
+        else:
+            gr.Warning("TikZ code compiled to an empty image!") # type: ignore
+    elif tikzdoc.compiled_with_errors:
+        gr.Warning("TikZ code compiled with errors!") # type: ignore
+
+    if rasterize:
+        yield tikzdoc.rasterize()
+    else:
+        with NamedTemporaryFile(suffix=".svg", buffering=0) as tmpfile:
+            if pdf:=tikzdoc.pdf:
+                tmpfile.write(convert_to_svg(pdf).encode())
+            yield tmpfile.name if pdf else None
+
+with gr.Blocks(css=".gradio-container {text-align: center}", theme=gr.themes.Soft()) as demo:
+    with open("/etc/os-release") as f:
+        print(f.read())
+        system("latexmk --version")
+        #with open("bla.tex", "w") as f:
+        #    f.write(tikz)
+        #    system("latexmk -f -nobibtex -norc -file-line-error -interaction=nonstopmode bla.tex")
+        #doc = TikzDocument(tikz)
+        #print(doc.log, doc.has_content, doc.compiled_with_errors, doc.status)
+        image = gr.Image()
+        run_btn = gr.Button("Run", variant="primary")
+        run_btn.click(
+            partial(tex_compile, code=tikz),
+            outputs=image
+        )
+        demo.queue().launch()