richardr1126 commited on
Commit
b4efc08
1 Parent(s): 1185d5b
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -125,6 +125,7 @@ def format(text):
125
  return final_query_markdown
126
 
127
  def extract_db_code(text):
 
128
  pattern = r'```(?:\w+)?\s?(.*?)```'
129
  matches = re.findall(pattern, text, re.DOTALL)
130
  return [match.strip() for match in matches]
@@ -149,7 +150,6 @@ def generate_dummy_db(db_info, question):
149
  response_text = response['choices'][0]['message']['content']
150
 
151
  db_code = extract_db_code(response_text)
152
- print(db_code)
153
 
154
  return db_code
155
 
 
125
  return final_query_markdown
126
 
127
  def extract_db_code(text):
128
+ print(text)
129
  pattern = r'```(?:\w+)?\s?(.*?)```'
130
  matches = re.findall(pattern, text, re.DOTALL)
131
  return [match.strip() for match in matches]
 
150
  response_text = response['choices'][0]['message']['content']
151
 
152
  db_code = extract_db_code(response_text)
 
153
 
154
  return db_code
155