test

app.py

@@ -11,15 +11,14 @@ openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 # sample data
 markdown_table_sample = """
-|    | 1 | 2 | 3 | 4 | 
-|----|---|---|---|---|
-| BD |   |   | x |   |
-| SD |   |   |   | x |
-| CH | x |   | x |   |
-| OH |   |   |   | x |  
-| LT |   |   |   |   | 
-| MT |   | x |   |   |
-| HT | x |   |   | x | 
+|    | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 |
+| BD | x |   | x |   |   |   | x |   | 
+| SD |   |   |   | x |   |   |   | x |
+| CH | x |   | x |   | x |   | x |   |
+| OH |   |   |   | x |   |   | x |   |
+| LT |   |   |   |   |   | x |   |   | 
+| MT |   | x |   |   | x |   |   |   | 
+| HT | x |   |   | x |   |   |   |   | 
 """
 
 markdown_table_sample2 = """
@@ -89,9 +88,9 @@ def get_answer(question):
     response = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
         messages=[
-            {"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": "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": "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},
             {"role": "user", "content": question}
@@ -115,7 +114,7 @@ def generate_rhythm(query, state):
     #     resolution = int(resolution_text)   
     # except:
     #     resolution = 8 # default 
-    resolution = 16 # default 
+    resolution = 8 # default 
 
     # Extract rhythm table
     table = "|" + "|".join(text_output.split('|')[1:-1]) +  "|"