Update docs

docs/_sources/chat.rst.txt

@@ -19,14 +19,51 @@ call the `chat` method to interact with the data and get insights from it via Na
 
    from wordview.text_analysis import TextStatsPlots
    imdb_df = pd.read_csv("data/IMDB_Dataset_sample_5k.csv")
-   with open("wordview/chat/secrets/openai_api_key.json", "r") as f:
+   with open("your_secrets_dir/openai_api_key.json", "r") as f:
       credentials = json.load(f)
 
    tsp = TextStatsPlots(df=imdb_df, text_column="review")
    tsp.chat(api_key=credentials.get("openai_api_key"))
 
 The chat UI is available under http://127.0.0.1:5000/
 
-|chat|
+|chat_stats|
 
-.. |chat| image:: ../figs/chat.png
+Chat with MWEs
+~~~~~~~~~~~~~~
+
+After allowing Wordview to extract MWEs, you can call the `chat` method to get insights from this extraction through Natural Language.
+
+.. code:: python
+
+   import json
+
+   import pandas as pd
+
+   from wordview.mwe_extraction import MWEs
+   from wordview.preprocessing import NgramExtractor
+
+   imdb_df = pd.read_csv("data/IMDB_Dataset_sample_5k.csv")
+   with open("your_secrets_dir/openai_api_key.json", "r") as f:
+      credentials = json.load(f)
+
+   extractor = NgramExtractor(imdb_df, "review")
+   extractor.extract_ngrams()
+   extractor.get_ngram_counts(ngram_count_file_path="ngram_counts.json")
+
+   mwe_obj = MWE(imdb_df, 'review',
+                ngram_count_file_path='ngram_counts.json',
+                language='EN',
+                custom_patterns="NP: {<DT>?<JJ>*<NN>}",
+                only_custom_patterns=False,
+                )
+    mwe_obj.extract_mwes(sort=True, top_n=10)
+    mwe_obj.chat(api_key=credentials.get("openai_api_key"))
+
+The chat UI for MWEs is available under http://127.0.0.1:5001/
+
+|chat_mwe|
+
+.. |chat_stats| image:: ../figs/chat_stats.png
+
+.. |chat_mwe| image:: ../figs/chat_mwe.png

docs/_sources/mwes.rst.txt

@@ -32,16 +32,20 @@ the documentation.
                   custom_patterns="NP: {<DT>?<JJ>*<NN>}",
                   only_custom_patterns=False,
                   )
-    mwes = mwe_obj.extract_mwes(sort=True, top_n=10)
-    json.dump(mwes, open('data/mwes.json', 'w'), indent=4)
-    
+    mwe_obj.extract_mwes(sort=True, top_n=10)
+    json.dump(mwe_obj.mwes, open('data/mwes.json', 'w'), indent=4)
 
-The above returns the results in a dictionary, that in this example we stored in `mwes.json` file.
+The above returns the results in a dictionary, that in this example we stored in a json file called `data/mwes.json`.
 You can also return the result in a table:
 
 .. code-block:: python
 
     mwe_obj.print_mwe_table()
+
+Which will return a table like this:
+
+.. code-block:: text
+
     ╔═════════════════════════╦═══════════════╗
     ║ LVC                     ║   Association ║
     ╠═════════════════════════╬═══════════════╣

docs/chat.html

@@ -60,6 +60,7 @@
 <li class="toctree-l1"><a class="reference internal" href="clustering.html">Cluster Analysis</a></li>
 <li class="toctree-l1 current"><a class="current reference internal" href="#">Chat with Wordview</a><ul>
 <li class="toctree-l2"><a class="reference internal" href="#chat-with-textstatsplots">Chat with TextStatsPlots</a></li>
+<li class="toctree-l2"><a class="reference internal" href="#chat-with-mwes">Chat with MWEs</a></li>
 </ul>
 </li>
 </ul>
@@ -115,15 +116,46 @@ <h2>Chat with TextStatsPlots<a class="headerlink" href="#chat-with-textstatsplot
 
 <span class="kn">from</span> <span class="nn">wordview.text_analysis</span> <span class="kn">import</span> <span class="n">TextStatsPlots</span>
 <span class="n">imdb_df</span> <span class="o">=</span> <span class="n">pd</span><span class="o">.</span><span class="n">read_csv</span><span class="p">(</span><span class="s2">&quot;data/IMDB_Dataset_sample_5k.csv&quot;</span><span class="p">)</span>
-<span class="k">with</span> <span class="nb">open</span><span class="p">(</span><span class="s2">&quot;wordview/chat/secrets/openai_api_key.json&quot;</span><span class="p">,</span> <span class="s2">&quot;r&quot;</span><span class="p">)</span> <span class="k">as</span> <span class="n">f</span><span class="p">:</span>
+<span class="k">with</span> <span class="nb">open</span><span class="p">(</span><span class="s2">&quot;your_secrets_dir/openai_api_key.json&quot;</span><span class="p">,</span> <span class="s2">&quot;r&quot;</span><span class="p">)</span> <span class="k">as</span> <span class="n">f</span><span class="p">:</span>
    <span class="n">credentials</span> <span class="o">=</span> <span class="n">json</span><span class="o">.</span><span class="n">load</span><span class="p">(</span><span class="n">f</span><span class="p">)</span>
 
 <span class="n">tsp</span> <span class="o">=</span> <span class="n">TextStatsPlots</span><span class="p">(</span><span class="n">df</span><span class="o">=</span><span class="n">imdb_df</span><span class="p">,</span> <span class="n">text_column</span><span class="o">=</span><span class="s2">&quot;review&quot;</span><span class="p">)</span>
 <span class="n">tsp</span><span class="o">.</span><span class="n">chat</span><span class="p">(</span><span class="n">api_key</span><span class="o">=</span><span class="n">credentials</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">&quot;openai_api_key&quot;</span><span class="p">))</span>
 </pre></div>
 </div>
 <p>The chat UI is available under <a class="reference external" href="http://127.0.0.1:5000/">http://127.0.0.1:5000/</a></p>
-<p><img alt="chat" src="_images/chat.png" /></p>
+<p><img alt="chat_stats" src="_images/chat_stats.png" /></p>
+</section>
+<section id="chat-with-mwes">
+<h2>Chat with MWEs<a class="headerlink" href="#chat-with-mwes" title="Permalink to this heading"></a></h2>
+<p>After allowing Wordview to extract MWEs, you can call the <cite>chat</cite> method to get insights from this extraction through Natural Language.</p>
+<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="kn">import</span> <span class="nn">json</span>
+
+<span class="kn">import</span> <span class="nn">pandas</span> <span class="k">as</span> <span class="nn">pd</span>
+
+<span class="kn">from</span> <span class="nn">wordview.mwe_extraction</span> <span class="kn">import</span> <span class="n">MWEs</span>
+<span class="kn">from</span> <span class="nn">wordview.preprocessing</span> <span class="kn">import</span> <span class="n">NgramExtractor</span>
+
+<span class="n">imdb_df</span> <span class="o">=</span> <span class="n">pd</span><span class="o">.</span><span class="n">read_csv</span><span class="p">(</span><span class="s2">&quot;data/IMDB_Dataset_sample_5k.csv&quot;</span><span class="p">)</span>
+<span class="k">with</span> <span class="nb">open</span><span class="p">(</span><span class="s2">&quot;your_secrets_dir/openai_api_key.json&quot;</span><span class="p">,</span> <span class="s2">&quot;r&quot;</span><span class="p">)</span> <span class="k">as</span> <span class="n">f</span><span class="p">:</span>
+   <span class="n">credentials</span> <span class="o">=</span> <span class="n">json</span><span class="o">.</span><span class="n">load</span><span class="p">(</span><span class="n">f</span><span class="p">)</span>
+
+<span class="n">extractor</span> <span class="o">=</span> <span class="n">NgramExtractor</span><span class="p">(</span><span class="n">imdb_df</span><span class="p">,</span> <span class="s2">&quot;review&quot;</span><span class="p">)</span>
+<span class="n">extractor</span><span class="o">.</span><span class="n">extract_ngrams</span><span class="p">()</span>
+<span class="n">extractor</span><span class="o">.</span><span class="n">get_ngram_counts</span><span class="p">(</span><span class="n">ngram_count_file_path</span><span class="o">=</span><span class="s2">&quot;ngram_counts.json&quot;</span><span class="p">)</span>
+
+<span class="n">mwe_obj</span> <span class="o">=</span> <span class="n">MWE</span><span class="p">(</span><span class="n">imdb_df</span><span class="p">,</span> <span class="s1">&#39;review&#39;</span><span class="p">,</span>
+             <span class="n">ngram_count_file_path</span><span class="o">=</span><span class="s1">&#39;ngram_counts.json&#39;</span><span class="p">,</span>
+             <span class="n">language</span><span class="o">=</span><span class="s1">&#39;EN&#39;</span><span class="p">,</span>
+             <span class="n">custom_patterns</span><span class="o">=</span><span class="s2">&quot;NP: {&lt;DT&gt;?&lt;JJ&gt;*&lt;NN&gt;}&quot;</span><span class="p">,</span>
+             <span class="n">only_custom_patterns</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
+             <span class="p">)</span>
+ <span class="n">mwe_obj</span><span class="o">.</span><span class="n">extract_mwes</span><span class="p">(</span><span class="n">sort</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">top_n</span><span class="o">=</span><span class="mi">10</span><span class="p">)</span>
+ <span class="n">mwe_obj</span><span class="o">.</span><span class="n">chat</span><span class="p">(</span><span class="n">api_key</span><span class="o">=</span><span class="n">credentials</span><span class="o">.</span><span class="n">get</span><span class="p">(</span><span class="s2">&quot;openai_api_key&quot;</span><span class="p">))</span>
+</pre></div>
+</div>
+<p>The chat UI for MWEs is available under <a class="reference external" href="http://127.0.0.1:5001/">http://127.0.0.1:5001/</a></p>
+<p><img alt="chat_mwe" src="_images/chat_mwe.png" /></p>
 </section>
 </section>
 

docs/genindex.html

@@ -153,8 +153,12 @@ <h2 id="B">B</h2>
 <h2 id="C">C</h2>
 <table style="width: 100%" class="indextable genindextable"><tr>
   <td style="width: 33%; vertical-align: top;"><ul>
-      <li><a href="api.html#wordview.text_analysis.TextStatsPlots.chat">chat() (wordview.text_analysis.TextStatsPlots method)</a>
+      <li><a href="api.html#wordview.mwes.MWE.chat">chat() (wordview.mwes.MWE method)</a>
+
+      <ul>
+        <li><a href="api.html#wordview.text_analysis.TextStatsPlots.chat">(wordview.text_analysis.TextStatsPlots method)</a>
 </li>
+      </ul></li>
   </ul></td>
   <td style="width: 33%; vertical-align: top;"><ul>
       <li><a href="api.html#wordview.clustering.cluster.Cluster">Cluster (class in wordview.clustering.cluster)</a>

docs/mwes.html

@@ -58,6 +58,7 @@
 <li class="toctree-l1"><a class="reference internal" href="bias.html">Bias Analysis</a></li>
 <li class="toctree-l1"><a class="reference internal" href="anomalies.html">Analysis of Anomalies &amp; Outliers</a></li>
 <li class="toctree-l1"><a class="reference internal" href="clustering.html">Cluster Analysis</a></li>
+<li class="toctree-l1"><a class="reference internal" href="chat.html">Chat with Wordview</a></li>
 </ul>
 <p class="caption" role="heading"><span class="caption-text">Utilities</span></p>
 <ul>
@@ -126,14 +127,17 @@ <h1>Analysis &amp; Extraction of Multiword Expressions (MWEs)<a class="headerlin
                <span class="n">custom_patterns</span><span class="o">=</span><span class="s2">&quot;NP: {&lt;DT&gt;?&lt;JJ&gt;*&lt;NN&gt;}&quot;</span><span class="p">,</span>
                <span class="n">only_custom_patterns</span><span class="o">=</span><span class="kc">False</span><span class="p">,</span>
                <span class="p">)</span>
- <span class="n">mwes</span> <span class="o">=</span> <span class="n">mwe_obj</span><span class="o">.</span><span class="n">extract_mwes</span><span class="p">(</span><span class="n">sort</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">top_n</span><span class="o">=</span><span class="mi">10</span><span class="p">)</span>
- <span class="n">json</span><span class="o">.</span><span class="n">dump</span><span class="p">(</span><span class="n">mwes</span><span class="p">,</span> <span class="nb">open</span><span class="p">(</span><span class="s1">&#39;data/mwes.json&#39;</span><span class="p">,</span> <span class="s1">&#39;w&#39;</span><span class="p">),</span> <span class="n">indent</span><span class="o">=</span><span class="mi">4</span><span class="p">)</span>
+ <span class="n">mwe_obj</span><span class="o">.</span><span class="n">extract_mwes</span><span class="p">(</span><span class="n">sort</span><span class="o">=</span><span class="kc">True</span><span class="p">,</span> <span class="n">top_n</span><span class="o">=</span><span class="mi">10</span><span class="p">)</span>
+ <span class="n">json</span><span class="o">.</span><span class="n">dump</span><span class="p">(</span><span class="n">mwe_obj</span><span class="o">.</span><span class="n">mwes</span><span class="p">,</span> <span class="nb">open</span><span class="p">(</span><span class="s1">&#39;data/mwes.json&#39;</span><span class="p">,</span> <span class="s1">&#39;w&#39;</span><span class="p">),</span> <span class="n">indent</span><span class="o">=</span><span class="mi">4</span><span class="p">)</span>
 </pre></div>
 </div>
-<p>The above returns the results in a dictionary, that in this example we stored in <cite>mwes.json</cite> file.
+<p>The above returns the results in a dictionary, that in this example we stored in a json file called <cite>data/mwes.json</cite>.
 You can also return the result in a table:</p>
-<div class="highlight-python notranslate"><div class="highlight"><pre><span></span>mwe_obj.print_mwe_table()
-╔═════════════════════════╦═══════════════╗
+<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="n">mwe_obj</span><span class="o">.</span><span class="n">print_mwe_table</span><span class="p">()</span>
+</pre></div>
+</div>
+<p>Which will return a table like this:</p>
+<div class="highlight-text notranslate"><div class="highlight"><pre><span></span>╔═════════════════════════╦═══════════════╗
 ║ LVC                     ║   Association ║
 ╠═════════════════════════╬═══════════════╣
 ║ SHOOT the binding       ║         26.02 ║