added a small example application: find place names in input text

Author: mark-watson

OpenAiApiClient/GenText.hs

@@ -6,6 +6,7 @@ import Network.HTTP.Client.TLS
 import System.Environment (getEnv)
 import qualified Data.Text as T
 import Data.Maybe (fromMaybe)
+import Data.Text (splitOn)
 
 -- example derived from the openai-client library documentation
 
@@ -45,7 +46,20 @@ completionRequestToString prompt = do
                     return $ fromMaybe "No content" $ T.unpack <$> content
                 _ -> return "No choices returned"
 
+-- find place names
+findPlaces :: String -> IO [String]
+findPlaces text = do
+    let prompt = "Extract only the place names separated by commas from the following text:\n\n" ++ text
+    response <- completionRequestToString prompt 
+    -- Convert Text to String using T.unpack before filtering
+    let places = filter (not . null) $ map T.unpack $ splitOn "," (T.pack response) 
+    -- Strip leading and trailing whitespace from each place name
+    return $ map (T.unpack . T.strip . T.pack) places
+
 main :: IO ()
 main = do
     response <- completionRequestToString "Write a hello world program in Haskell"
-    putStrLn response
+    putStrLn response
+
+    places <- findPlaces "I visited London, Paris, and New York last year."
+    print places