Add cardifying additional content

app.py

@@ -1,4 +1,6 @@
 import json
+import os
+import requests
 
 import gradio as gr
 
@@ -6,16 +8,46 @@ from autocards.autocards import Autocards
 
 AUTO = Autocards()
 
-def cardify(content):
+def cardify(content, mode):
     global AUTO
     
     AUTO.clear_qa()
-    AUTO.consume_var(content)
+    if mode != "raw" and mode != "url":
+        r = requests.get(content)
+        if mode == "pdf":
+            with open("temp.pdf", 'wb') as f:
+                f.write(r.content) 
+            AUTO.consume_pdf(content)
+            os.remove("temp.pdf")
+        elif mode == "epub":
+            with open("temp.epub", 'wb') as f:
+                f.write(r.content)
+            AUTO.consume_epub(content)
+            os.remove("temp.epub")
+        elif mode == "textfile":
+            with open("temp.txt", 'w') as f:
+                f.write(r.text)
+            AUTO.consume_textfile(content)
+            os.remove("temp.txt")
+    else:
+        if mode == "raw":
+            AUTO.consume_var(content)
+        elif mode == "url":
+            AUTO.consume_web(content)
+
     AUTO.to_json("output.json", prefix="")
     with open("output_basic.json") as f:
         res = json.load(f)
+    
+    # delete the output files
+    os.remove("output_basic.json")
+    os.remove("output_cloze.json")
 
     return json.dumps(res)
 
-iface = gr.Interface(fn=cardify, inputs="text", outputs="text")
+iface = gr.Interface(
+    fn=cardify,
+    inputs=["text", gr.Radio(["pdf", "epub", "raw", "textfile", "url"])],
+    outputs="text"
+)
 iface.launch()