resolution

app.py

@@ -10,7 +10,8 @@ import gradio as gr
 openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 # sample data
-markdown_table_sample = """
+markdown_table_sample = """8th
+
 |    | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 |
 |----|---|---|---|---|---|---|---|---|
 | BD | x |   | x |   |   |   | x |   | 
@@ -22,7 +23,8 @@ markdown_table_sample = """
 | HT | x |   |   | x |   |   |   |   | 
 """
 
-markdown_table_sample2 = """
+markdown_table_sample2 = """16th
+
 |    | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10| 11| 12| 13| 14| 15| 16|
 |----|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
 | BD | x |   | x |   |   |   | x |   | x |   | x |   | x |   | x |   |
@@ -90,7 +92,7 @@ def get_answer(question):
         model="gpt-3.5-turbo",
         messages=[
             {"role": "system", "content": "You are a rhythm generator. "},
-            {"role": "user", "content": "Please generate a rhythm pattern in a Markdown table.  Time resolution is the 8th note. You use the following drums. Kick drum:BD, Snare drum:SD, Closed-hihat:CH, Open-hihat:OH, Low-tom:LT, Mid-tom:MT, High-tom:HT. use 'x' for an accented beat, 'o' for a weak beat."},
+            {"role": "user", "content": "Please generate a rhythm pattern in a Markdown table.  Time resolution is the 8th note. You use the following drums. Kick drum:BD, Snare drum:SD, Closed-hihat:CH, Open-hihat:OH, Low-tom:LT, Mid-tom:MT, High-tom:HT. use 'x' for an accented beat, 'o' for a weak beat. You need to write the time resolution first."},
             {"role": "assistant", "content": markdown_table_sample},
             # {"role": "user", "content": "Please generate a rhythm pattern. The resolution is the fourth note. You use the following drums. Kick drum:BD, Snare drum:SD, Closed-hihat:CH, Open-hihat:OH, Low-tom:LT, Mid-tom:MT, High-tom:HT. use 'x' for an accented beat, 'o' for a weak beat. You need to write the time resolution first."},
             # {"role": "assistant", "content": markdown_table_sample},
@@ -109,13 +111,12 @@ def generate_rhythm(query, state):
     text_output = get_answer(query)
     
     # Try to use the first row as time resolution
-    # resolution_text = text_output.split('|')[0]
-    # try:
-    #     resolution_text = re.findall(r'\d+', resolution_text)[0]
-    #     resolution = int(resolution_text)   
-    # except:
-    #     resolution = 8 # default 
-    resolution = 8 # default 
+    resolution_text = text_output.split('|')[0]
+    try:
+        resolution_text = re.findall(r'\d+', resolution_text)[0]
+        resolution = int(resolution_text)   
+    except:
+        resolution = 8 # default 
 
     # Extract rhythm table
     table = "|" + "|".join(text_output.split('|')[1:-1]) +  "|"