Balance backticks and add title

app.py

@@ -41,6 +41,20 @@ def add_text(history, text):
     history = history + [(text, None)]
     return history, gr.Textbox(value="", interactive=False)
 
+def has_balanced_backticks(markdown_str):
+    in_code_block = False
+    lines = markdown_str.split('\n')
+    
+    for line in lines:
+        stripped_line = line.strip()
+        
+        # Check if the line contains triple backticks
+        if stripped_line.startswith("```"):
+            # Toggle the in_code_block flag
+            in_code_block = not in_code_block
+    
+    # If in_code_block is False at the end, all backticks are balanced
+    return not in_code_block
 
 def bot(history, api_kind,
             num_docs, model_kind, sub_vector_size, chunk_size, splitter_type, all_at_once):
@@ -79,7 +93,10 @@ def bot(history, api_kind,
             prompt_html = template_html.render(documents=documents, query=query)
             
             hist_chunk = ""
-            prev_hist = history[-1][1] + f"\nmodel {model_name}, splitter {doc}, size {size}, sub vector {sub_vector}\n"
+            prev_hist = history[-1][1] 
+            if not has_balanced_backticks(prev_hist):
+                prev_hist += "\n```\n"
+            prev_hist += f"\n\n## model {model_name}, splitter {doc}, size {size}, sub vector {sub_vector}\n\n"
             for character in generate_fn(prompt_i, history[:-1]):
                 hist_chunk = character
                 history[-1][1] = prev_hist + hist_chunk