Update app.py

app.py

@@ -10,7 +10,6 @@ from transformers import (
 from huggingface_hub import login
 import gradio as gr
 import torch
-import markdown
 
 login(os.getenv("HF_TOKEN", None))
 
@@ -77,17 +76,17 @@ def bot(input_message: str, db_info="", temperature=0.1, top_p=0.9, top_k=0, rep
         partial_text += new_text
 
     # Split the text by "|", and get the last element in the list which should be the final query
-    final_query = partial_text.split("|")[-1].strip()
+    final_query = partial_text.split("|")[1].strip()
     # Convert SQL to markdown (not required, but just to show how to use the markdown module)
-    final_query_markdown = f'```sql\n{final_query}\n```'
-    return markdown.markdown(final_query_markdown)
+    final_query_markdown = "```sql\n{final_query}\n```"
+    return final_query_markdown
 
 
 gradio_interface = gr.Interface(
   fn=bot,
   inputs=[
-      gr.Textbox(lines=20, placeholder='Input text here...', label='Input Text'),
-      gr.Textbox(lines=20, placeholder='(Recommended) Example: | stadium : stadium_id , location , name , capacity , highest , lowest , average | singer : singer_id , name , country , song_name , song_release_year , age , is_male | concert : concert_id , concert_name , theme , stadium_id , year | singer_in_concert : concert_id , singer_id | concert.stadium_id = stadium.stadium_id | singer_in_concert.singer_id = singer.singer_id | singer_in_concert.concert_id = concert.concert_id | ', label='Databse Info'),
+      gr.Textbox(lines=4, placeholder='Input text here...', label='Input Text'),
+      gr.Textbox(lines=8, placeholder='(Recommended) Example: | stadium : stadium_id , location , name , capacity , highest , lowest , average | singer : singer_id , name , country , song_name , song_release_year , age , is_male | concert : concert_id , concert_name , theme , stadium_id , year | singer_in_concert : concert_id , singer_id | concert.stadium_id = stadium.stadium_id | singer_in_concert.singer_id = singer.singer_id | singer_in_concert.concert_id = concert.concert_id | ', label='Databse Info'),
       gr.Slider(label="Temperature", minimum=0.0, maximum=1.0, value=0.1, step=0.1),
       gr.Slider(label="Top-p (nucleus sampling)", minimum=0.0, maximum=1.0, value=0.9, step=0.01),
       gr.Slider(label="Top-k", minimum=0, maximum=200, value=0, step=1),