mime: Support plaintext format as well

1 files changed, 7 insertions, 2 deletions

diff --git a/browset.py b/browset.py
index 5fb365e..3e65fff 100755
--- a/browset.py
+++ b/browset.py
@@ -4,7 +4,8 @@ from textual.widgets import Input, Button, Static, Footer
 from textual.containers import Container
 from rich.markdown import Markdown
 from textual.binding import Binding
-from gemtext import Gemtext
+from mime.gemtext import Gemtext
+from mime.plaintext import Plaintext
 from protocol.gemini import GeminiProtocol
 
 class Browset(App):
@@ -69,7 +70,11 @@ class Browset(App):
         self.url = url
         (mime, fp) = GeminiProtocol.get(url)
         self.query_one("#content").remove()
-        self.mount(Gemtext(fp=fp, id="content"))
+        if "text/gemini" in mime:
+          content = Gemtext(fp=fp, id="content")
+        else:
+          content = Plaintext(fp=fp, id="content")
+        self.mount(content)
 
 if __name__ == "__main__":
     app = Browset()