update examples

app.py

@@ -9,6 +9,8 @@ import pprint
 from huggingface_hub import Repository
 from text_generation import Client
 
+from requests.exceptions import ReadTimeout
+
 PORT = 7860
 
 # TODO: implement maximum length (currently, each iteration is limited by the slider-specified max length, but this can be iterated, or long code entered into the editor, to get really long documents
@@ -109,6 +111,9 @@ async def generate_maybe(info: str):
         else:
             message = ''
         return {'result': 'success', 'type': 'generate', 'prompt': prompt, 'text': generation['text'], 'message': message}
+    except ReadTimeout as e:
+        print(e)
+        return {'result': 'error', 'type': 'generate', 'prompt': prompt, 'message': f'Request timed out.'}
     except Exception as e:
         traceback.print_exception(*sys.exc_info())
         return {'result': 'error', 'type': 'generate', 'prompt': prompt, 'message': f'Error: {e}.'}
@@ -134,6 +139,9 @@ async def infill_maybe(info: str):
         else:
             generation['message'] = ''
         return generation
+    except ReadTimeout as e:
+        print(e)
+        return {'result': 'error', 'type': 'generate', 'prompt': prompt, 'message': f'Request timed out.'}
     except Exception as e:
         traceback.print_exception(*sys.exc_info())
         return {'result': 'error', 'type': 'infill', 'message': f'Error: {e}.'}

static/index.html

@@ -135,17 +135,15 @@ label {
         <div id="examples-extend">
             <span class="softspan">Extend Examples:</span>
             <br>
-            <span class="softspan"><a href='javascript:select_example("hello-world");'>Hello World</a></span>
-            <span class="softspan"><a href='javascript:select_example("fibonacci");'>Fibonacci</a></span>
-            <span class="softspan"><a href='javascript:select_example("typing");'>Typing</a></span>
-            <span class="softspan"><a href='javascript:select_example("complex-numbers");'>Complex Numbers</a></span>
+            <span class="softspan"><a href='javascript:select_example("logistic-regression");'>Logistic Regression (Python)</a></span>
+            <span class="softspan"><a href='javascript:select_example("array");'>Arrays (JS)</a></span>
+            <span class="softspan"><a href='javascript:select_example("metadata");'>Metadata-Conditioning (JS)</a></span>
         </div>
         <div id="examples-infill">
             <span class="softspan">Infill Examples:</span>
             <br>
             <span class="softspan"><a href='javascript:select_example("type-pred");'>Type Prediction</a></span>
-            <span class="softspan"><a href='javascript:select_example("docstring-1");'>Docstring Generation 1</a></span>
-            <span class="softspan"><a href='javascript:select_example("docstring-2");'>Docstring Generation 2</a></span>
+            <span class="softspan"><a href='javascript:select_example("docstring");'>Docstring Generation</a></span>
         </div>
     </div>
     <div class="card" id="controls">
@@ -226,8 +224,7 @@ var Range = require("ace/range").Range;
 // examples for the user
 var EXAMPLES = {
     "type-pred": {
-        "prompt": 
-`def count_words(filename: str) -> <infill>
+        "prompt": `def count_words(filename: str) -> <infill>
     """Count the number of occurrences of each word in the file."""
     with open(filename, 'r') as f:
         word_counts = {}
@@ -243,9 +240,8 @@ var EXAMPLES = {
         "temperature": 0.2,
         "mode": "python"
     },
-    "docstring-1": {
-        "prompt":
-`def _minimize_in_graph(build_loss_fn, num_steps=200, optimizer=None):
+    "docstring": {
+        "prompt": `def _minimize_in_graph(build_loss_fn, num_steps=200, optimizer=None):
   """
   <infill>
   """
@@ -263,61 +259,34 @@ var EXAMPLES = {
       loop_vars=[tf.constant(0)],
       return_same_structure=True)[0] 
   return minimize_op`,
-  "length": 64,
-  "temperature": 0.3,
+  "length": 128,
+  "temperature": 0.2,
   "mode": "python",
     },
-    "docstring-2": {
-        "prompt": 
-`def count_words(filename: str) -> Dict[str, int]:
-    """<infill>
-    """
-    with open(filename, 'r') as f:
-        word_counts = {}
-        for line in f:
-            for word in line.split():
-                if word in word_counts:
-                    word_counts[word]  = 1
-                else:
-                    word_counts[word] = 1
-    return word_counts
-`,
-        "length": 32,
-        "temperature": 0.2,
-        "mode": "python"
-    },
   // extend examples
-    "hello-world": {
-        "prompt": "def print_hello_world():",
+    "logistic-regression": {
+        "prompt": `X_train, y_train, X_test, y_test = train_test_split(X, y, test_size=0.1)
+
+# Train a logistic regression model, predict the labels on the test set and compute the accuracy score`,
         "length": 64,
         "temperature": 0.2,
         "mode": "python"
     },
-    "fibonacci": {
+    "array": {
       "prompt": 
-`def fibonacci(n: int) -> int:
-    """ Compute the n-th Fibonacci number. """`,
+`// Returns every other value in the array as a new array.
+function everyOther(arr) {`,
         "temperature": 0.2,
         "length": 64,
-        "mode": "python"
+        "mode": "javascript"
     },
-  "typing": {
-    "prompt":
-`from typing import List, Tuple
-
-def sum_and_product(numbers: List[int]) -> Tuple[int, int]:
-    """ Return the sum and the product of the integers in the list as a tuple. Here is the answer of the exercise"""`,
-        "temperature": 0.2,
-        "length": 64,
-        "mode": "python"
-  },
-  "complex-numbers": {
-    "prompt":
-`class ComplexNumbers:`,
-        "temperature": 0.2,
+  "metadata": {
+    "prompt": `<filename>start_server.js<gh_stars>100-1000
+`,
+        "temperature": 0.6,
         "length": 128,
-        "mode": "python"
-  }
+        "mode": "javascript"
+  },
 };
 
 var editor = ace.edit("editor");
@@ -516,6 +485,8 @@ function make_generate_listener(url) {
                 }
             } else {
                 console.log("error");
+                console.log("data:");
+                console.log(receive_data);
                 set_text(receive_data["text"])
                 $("#error").text(receive_data["message"]);
             }