Update app.py

app.py

@@ -0,0 +1,118 @@
+import subprocess
+import os
+from PIL import Image
+##################################
+def write_txt(file_name,text):
+    with open(file_name, "w", encoding="utf-8") as file:
+        file.write(text)
+##################################
+def load_txt(file_name):
+    with open(file_name, "r", encoding="utf-8") as file:
+        text = file.read()
+        return text
+####
+os.system("apt-get install -y --no-install-recommends texlive-latex-base texlive-latex-recommended texlive-pictures texlive-science texlive-latex-extra  poppler-utils")
+print("all latex installed")
+####
+
+def extract_document_content(latex_code):
+    """
+    Extract the content between \begin{document} and \end{document} in LaTeX code.
+    
+    Args:
+        latex_code (str): The LaTeX code to extract from
+    
+    Returns:
+        str: The extracted content, or None if no document environment is found
+    """
+    start_tag = "\\begin{document}"
+    end_tag = "\\end{document}"
+    
+    start_index = latex_code.find(start_tag)
+    end_index = latex_code.find(end_tag)
+    
+    if start_index == -1 or end_index == -1:
+        return None
+    
+    # Add the length of the start tag to get the content right after it
+    start_content_index = start_index + len(start_tag)
+    
+    # Extract the content between the tags
+    document_content = latex_code[start_content_index:end_index]
+    
+    return document_content.strip()
+##################################
+from google import genai
+
+GEMINI_API_KEY = "AIzaSyB8pCfu9a3Bx7L6v1JyTdMMNf4W55B5D08"
+
+client = genai.Client(api_key=GEMINI_API_KEY)
+
+from google.genai import types
+def generate_tex(file_path):
+    print(file_path)
+    myfile = client.files.upload(file=file_path)
+
+    result = client.models.generate_content(
+        model="gemini-2.0-flash",#"gemini-1.5-pro",#
+        contents=[
+            myfile,
+            "\n\n",
+            "Act as an expert in tikzcd. Make the tikcd code of the image, include packages for special characters.\n"+
+            "The document class is always standalone, and include: \\usetikzlibrary{decorations.pathmorphing}, \\usepackage{stmaryrd}, \\usepackage{amssymb} after the tikcd library."+
+            "Do not make any suggestion nor commentaries."
+        ],
+        config=types.GenerateContentConfig(temperature=0.1)
+    )
+
+
+    #tex_code = load_txt("latex_code.tex")
+    return result.text
+
+def generate_pdf_png(latex_code):
+    ## output latex_code.pdf
+    # remove
+    latex_code = latex_code.replace("```tex","")
+    latex_code = latex_code.replace("```latex","")
+    latex_code = latex_code.replace("```","")
+    write_txt("latex_code.tex",latex_code)
+    print("file saved")
+    # Compile with pdflatex
+    #os.system("pdflatex latex_code.tex")
+    import subprocess
+    try:
+        # Set a timeout (in seconds)
+        result = subprocess.run(['pdflatex'],input=b'latex_code.tex\n', timeout=20, 
+                            stdout=subprocess.PIPE, 
+                            stderr=subprocess.PIPE)
+        print(result.stdout.decode())
+    except subprocess.TimeoutExpired:
+        print("Command timed out")
+    # Convert PDF to image using pdf2image (needs poppler installed)
+    pdf_file = "latex_code.pdf"
+    print("already have a pdf")
+    # save as pdf
+    from pdf2image import convert_from_path
+    images = convert_from_path(pdf_file)
+    image = images[0]
+    image.save("latex_code.png")
+    print("Image saved as latex_code.png")
+
+def generate_results(file_path):
+    tikcd_code = generate_tex(file_path) # filepath of the image, as input for the generative model
+    generate_pdf_png(tikcd_code)
+    return ("./latex_code.png",extract_document_content(tikcd_code))
+#
+print("all was initialized\n")
+#write_txt("aux.tex","Hello world")
+#
+import gradio as gr
+app = gr.Interface(
+    fn= generate_results, 
+    inputs=gr.Image(type="filepath"), 
+    outputs=[gr.Image(label="Imagen latex"),gr.TextArea(label="Código latex",lines=30)]
+)
+
+####
+#port = int(os.environ.get("PORT", 8080))
+app.launch(share=True)#server_name="0.0.0.0", server_port=port,debug=True)