1 files changed, 7 insertions, 1 deletions

@@ -10,6 +10,7 @@
 #include <twitcurl.h>
 #include <unistd.h>
 #include <yaml-cpp/yaml.h>
+#include "freevars.h"
 
 using namespace::std;
 
@@ -28,6 +29,11 @@ int main(int argc, char** args)
 
     cout << "Preprocessing corpus..." << endl;
         kgramstats* stats = new kgramstats(corpus, 5);
+    
+    cout << "Preprocessing freevars..." << endl;
+    freevars* vars = new freevars();
+    vars->addVar("name", "names.txt");
+    vars->addVar("noun", "nouns.txt");
 
     cout << "Generating..." << endl;
         for (;;)
@@ -36,7 +42,7 @@ int main(int argc, char** args)
                 string hi;
                 for (vector<string>::iterator it = doc.begin(); it != doc.end(); ++it)
                 {
-                        hi += *it + " ";
+                        hi += vars->parse(*it) + " ";
                 }
         
                 hi = hi.substr(0,140);