remove latex tags

app.py

@@ -1,6 +1,7 @@
 from gradio_client import Client
 import gradio as gr
 import os
+import random
 
 def check_password(username, password):
     if password == os.environ["ACCESS"]:
@@ -8,11 +9,25 @@ def check_password(username, password):
     else:
         return False
 
+def remove_latex_tags(input_string):
+    if input_string.startswith("```latex") and input_string.endswith("```"):
+        # Remove the ```latex at the beginning and ``` at the end
+        result_string = input_string[len("```latex"):-len("```")]
+        return result_string
+    else:
+        # If the input string doesn't have the expected tags, return it unchanged
+        return input_string
+
 def func(file, number_of_pages, secret):
     if secret != os.environ["ACCESS"]:
         return "Wrong password, please try again"
-    else:
-        return client.predict(file, number_of_pages)
+    text = client.predict(file, number_of_pages)
+    text = remove_latex_tags(text)
+    # generate a random sequence of numbers
+    # s = "".join([str(random.randint(0, 9)) for i in range(10)])
+    # with open(f"{s}.tex", "w", encoding="utf-8") as f:
+    #     f.write(text)
+    return text
 
 def upload_file(file):
     return file.name
@@ -24,9 +39,11 @@ description = r"""
 ❗️❗️❗️[<b>Important</b>] Instructions:<br>
 1️⃣ <b>Upload the PDF document</b>: Select the PDF file you want to convert into slides.<br>
 2️⃣ <b>Specify the number of pages</b>: Indicate the range of pages you'd like to include in the slide generation. <b>Set it to 0</b> if you want to include all pages. <br>
+3️⃣ <b>Enter the password provided in the invite email.</b>
 3️⃣ <b>Click the Generate button</b>: Initiate the slide generation process by clicking the designated "Generate" button.<br>
-4️⃣ <b>Wait for approximately 200 seconds</b>: Allow a brief moment for the system to generate the slides. The process may take up to 200 seconds.<br>
+4️⃣ <b>Be patient 🙂</b>: Generating the slides could take between 1 minute and 5 minutes.<br>
 
+ 
 📝 If you get an error message, you can send me email with the PDF file attached to this email address: <b>nkoisheke [at] ethz [dot] ch</b>, and I will generate the slides for you. If there are any other issues or questions, please do not hesitate to contact me 🤗 <br>
 """