test

app.py

@@ -10,8 +10,7 @@ import gradio as gr
 openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 # sample data
-markdown_table_sample = """4
-
+markdown_table_sample = """
 |    | 1 | 2 | 3 | 4 | 
 |----|---|---|---|---|
 | BD |   |   | x |   |
@@ -23,17 +22,16 @@ markdown_table_sample = """4
 | HT | x |   |   | x | 
 """
 
-markdown_table_sample2 = """8
-
-|    | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 |
-|----|---|---|---|---|---|---|---|---|
-| BD |   |   | x |   |   |   | x |   |
-| SD |   |   |   | x |   |   |   | x |
-| CH | x |   | x |   | x |   | x |   |
-| OH |   |   |   | x |   |   | x |   |
-| LT |   |   |   |   |   | x |   |   |
-| MT |   | x |   |   | x |   |   |   |
-| HT | x |   |   | x |   |   |   |   |
+markdown_table_sample2 = """
+|    | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10| 11| 12| 13| 14| 15| 16|
+|----|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
+| BD | x |   | x |   |   |   | x |   | x |   | x |   | x |   | x |   |
+| SD |   |   |   | x |   |   |   | x |   |   | x |   |   |   | x |   |
+| CH | x |   | x |   | x |   | x |   | x |   | x |   | x |   | x |   |
+| OH |   |   |   | x |   |   | x |   |   |   |   | x |   |   | x |   |
+| LT |   |   |   |   |   | x |   |   |   |   |   |   |   | x |   |   |
+| MT |   | x |   |   | x |   |   |   |   | x |   |   | x |   |   |   |
+| HT | x |   |   | x |   |   |   |   | x |   |   | x |   |   |   |   |
 """
 
 MIDI_NOTENUM = {
@@ -91,8 +89,8 @@ def get_answer(question):
     response = openai.ChatCompletion.create(
         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. You need to write the time resolution first."},
+            {"role": "system", "content": "You are a rhythm generator. Time resolution is the 16th note. "},
+            {"role": "user", "content": "Please generate a rhythm pattern in a Markdown table. 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": "assistant", "content": markdown_table_sample2},
             # {"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},
@@ -111,12 +109,13 @@ 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_text = text_output.split('|')[0]
+    # try:
+    #     resolution_text = re.findall(r'\d+', resolution_text)[0]
+    #     resolution = int(resolution_text)   
+    # except:
+    #     resolution = 8 # default 
+    resolution = 16 # default 
 
     # Extract rhythm table
     table = "|" + "|".join(text_output.split('|')[1:-1]) +  "|"