Miguel Diaz commited on
Commit
45d3058
1 Parent(s): 6f38106

corrección, remover test

Browse files
Files changed (1) hide show
  1. main.py +2 -4
main.py CHANGED
@@ -81,12 +81,10 @@ async def chat(request: Request):
81
  )
82
 
83
  message = response.choices[0].message.content
84
- print(message)
85
  while "```" in message:
86
- start = message.find("```")+1
87
  end = message.find("\n", start)
88
- language = message[start:end]
89
- print("-"*5,language,"-"*5, sep="-")
90
  language = language.replace(" ", "-")
91
  if language in supp_langs:
92
  message = message.replace("```"+message[start:end], f'<pre><code class="language-{language}">', 1)
 
81
  )
82
 
83
  message = response.choices[0].message.content
 
84
  while "```" in message:
85
+ start = message.find("```")+3
86
  end = message.find("\n", start)
87
+ language = message[start:end].strip()
 
88
  language = language.replace(" ", "-")
89
  if language in supp_langs:
90
  message = message.replace("```"+message[start:end], f'<pre><code class="language-{language}">', 1)