#!/usr/bin/python
import sys,string
import lsa
from optparse import OptionParser

# Set up the option parser

parser = OptionParser()
parser.add_option("-v", "--vector", action="store_true", help="Get vector associated with argument.", default=False)

(options, args) = parser.parse_args()

space = lsa.lsaSpace()
if options.vector:
  for i in xrange(len(args)):
    print space.getDocVector(args[i])
  sys.exit()

if len(args) != 2:
	print "lsa takes two required arguments."
	sys.exit(1)
firstArg = string.split(args[0])
secondArg = string.split(args[1])

if len(firstArg) == 1 and len(secondArg) == 1:
	print(space.cosTermTerm(args[0], args[1]))
elif len(firstArg) == 1:
	print(space.cosTermDoc(args[0], args[1]))
elif len(secondArg) == 1:
	print(space.cosTermDoc(args[1], args[0]))
else:
	print(space.cosDocDoc(args[0], args[1]))