Update app.py

app.py

@@ -64,7 +64,7 @@ if compute_dtype == torch.float16 and use_4bit:
 
 model = AutoModelForCausalLM.from_pretrained(
     model_name, quantization_config=bnb_config)
-stop_list = [" \n\nAnswer:", " \n", " \n\n"]
+stop_list = [" \n\nAnswer:",  " \n\n"]
 stop_token_ids = [tokenizer(x,  return_tensors='pt', add_special_tokens=False)['input_ids'] for x in stop_list]
 stop_token_ids = [torch.LongTensor(x).to("cuda") for x in stop_token_ids]
     
@@ -132,11 +132,11 @@ rag_chain_with_source = RunnableParallel(
 
 
 def format_result(result):
-    unique_videos = set((doc.metadata['act_number'], doc.metadata['act_url']) for doc in result['context'])
+    unique_videos = set((doc.metadata['title'], doc.metadata['act_url']) for doc in result['context'])
 
     # Create a plain text string where each title is followed by its URL
     titles_with_links = [
-        f"{act_number}: {act_url}" for act_number, act_url in unique_videos
+        f"{title}: {act_url}" for title, act_url in unique_videos
     ]
 
     # Join these entries with line breaks to form a clear list
@@ -220,7 +220,7 @@ def generate_answer(message, history, checkbox, subject_input, rubric, date_beg,
 
     result['answer'] =result['answer'].split("[/INST]")[-1].strip()
     formatted_results = format_result(result)
-    return formatted_results
+    return result
 
 
 
@@ -257,7 +257,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo :
                             chatbot=gr.Chatbot(height=400, render = False),
                             textbox = gr.Textbox(placeholder="Ввести питання", container=False, scale=7, render = False),
                             title="Законодавчий Помічник",
-                            description="Спитай мене про будь-які регуляції в чинних законах України.",
+                            description="Спитай мене про будь-які регуляції в чинних законах України, прийнятих VIII та IX скликанням ВРУ.",
                           #  # examples=["мобілізація", "земельна реформа", "екологія"],
 
                             cache_examples=False,