diff --git a/lm/__main__.py b/lm/__main__.py
--- a/lm/__main__.py
+++ b/lm/__main__.py
@@ -1,4 +1,4 @@
-from .lm.spellbook import spellbook
+from .lm import spellbook, openai, gradio
 from .log.jsonfiles import jsonfiles
 from .log.arweaveditems import arweaveditems
 
@@ -9,6 +9,12 @@ import sys
 async def amain():
     timestamp = datetime.datetime.now().isoformat()
     parser = argparse.ArgumentParser()
+    parser.add_argument('--lm', choices=['spellbook', 'openai', 'gradio'], default='spellbook', help='Language model to use')
+    parser.add_argument('--log', choices=['jsonfiles', 'arweaveditems'], default='jsonfiles', help='Logging method to use')
     parser.add_argument('-o', '--output', type=argparse.FileType('w'), default=sys.stdout, help='Output file')
     parser.add_argument('input_data', nargs='*', help='Input data or filename')
     args = parser.parse_args()
@@ -36,8 +42,12 @@ async def amain():
     else:
         input_data = sys.stdin.read()
 
-    api = spellbook()
-    log = jsonfiles(prefix = 'gpt4-')
+    api = {
+        'spellbook': spellbook,
+        'openai': openai,
+        'gradio': gradio,
+    }[args.lm]()
+    log = {'jsonfiles': jsonfiles, 'arweaveditems': arweaveditems}[args.log](prefix='gpt4-')
     log2 = arweaveditems()
     
     async with api, log, log2: