Update app.py

app.py

@@ -132,11 +132,11 @@ rag_chain_with_source = RunnableParallel(
 
 
 def format_result(result):
-    unique_videos = set((doc.metadata['title'], doc.metadata['act_url']) for doc in result['context'])
+    unique_videos = set((doc.metadata['act_number'], 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"{title}: {act_url}" for title, act_url in unique_videos
+        f"{act_number}: {act_url}" for act_number, act_url in unique_videos
     ]
 
     # Join these entries with line breaks to form a clear list
@@ -198,29 +198,29 @@ def summarize_act(message, act_number):
     return rag_chain_summarize.invoke("")
 
 
-def generate_answer(message, history, checkbox,subject_input, rubric, date_beg, date_end, act_number):
-  result = ""
-  if checkbox:
-    if act_number=="":
-      return "Будь ласка, введіть номер акту для отримання основної інформації з нього, або зніміть відповідний прапорець."
-    result = summarize_act(message, act_number)
-  else:
-      if subject_input is None and rubric is None and date_beg == "2010-01-01" and date_end == "2025-01-01":
-        result = rag_chain_with_source.invoke(message)
-
-  else:
-          if subject_input is None or subject_input ==[]:
-             subject_input = ["Президент України", "Кабінет міністрів України", "Народний депутат України"]
-          if rubric is None or rubric== []:
-            rubric = ['Двосторонні міжнародні угоди', 'Багатосторонні міжнародні угоди',
-       'Галузевий розвиток', 'Економічна політика',
-       'Державне будівництво', 'Соціальна політика', 'Правова політика',
-       'Безпека і оборона', 'Гуманітарна політика']
-          result = generate_with_filters(message)
-
-  result['answer'] =result['answer'].split("[/INST]")[-1].strip()
-  formatted_results = format_result(result)
-  return formatted_results
+def generate_answer(message, history, checkbox, subject_input, rubric, date_beg, date_end, act_number):
+    result = ""
+    if checkbox:
+        if act_number == "":
+            return "Будь ласка, введіть номер акту для отримання основної інформації з нього, або зніміть відповідний прапорець."
+        result = summarize_act(message, act_number)
+    else:
+        if subject_input is None and rubric is None and date_beg == "2010-01-01" and date_end == "2025-01-01":
+            result = rag_chain_with_source.invoke(message)
+        else:
+            if subject_input is None or subject_input == []:
+                subject_input = ["Президент України", "Кабінет міністрів України", "Народний депутат України"]
+            if rubric is None or rubric == []:
+                rubric = ['Двосторонні міжнародні угоди', 'Багатосторонні міжнародні угоди',
+                          'Галузевий розвиток', 'Економічна політика',
+                          'Державне будівництво', 'Соціальна політика', 'Правова політика',
+                          'Безпека і оборона', 'Гуманітарна політика']
+            result = generate_with_filters(message)
+
+
+    result['answer'] =result['answer'].split("[/INST]")[-1].strip()
+    formatted_results = format_result(result)
+    return formatted_results