Upload main.py

main.py

@@ -15,6 +15,8 @@ import logging
 from langchain_core.exceptions import OutputParserException
 import os
 
+from sympy.codegen.ast import continue_
+
 # Constants
 PERSIST_DIRECTORY = "chroma_store"
 K_VALUE = 5
@@ -88,43 +90,55 @@ def llm_response(query):
     titles, links, res_titles, res_links = [], [], [], []
     try:
         result = chain.invoke({"input": query})
-        if not result['answer']['cited_answer']:
+        if not result['answer']['cited_answer'][0]["answer"]:
             return "There are no articles relevant to your inquiry..."
         answer = result['answer']['cited_answer'][0]["answer"]
-        if result['answer']['cited_answer']:
+        if not result['answer']['cited_answer'][1]['citations']:
+            answer_with_citations = f"{answer}"
+        else:
             citations = result['answer']['cited_answer'][1]['citations']
             for citation in citations:
-                edited_item = citation['citation'][1]["source"].replace("Articles/", "")
-                title, link = get_article_info(df, edited_item)
-                if title not in titles:
-                    titles.append(title)
-                if link not in links:
-                    links.append(link)
+                try:
+                    edited_item = citation['citation'][1]["source"].replace("Articles/", "")
+                    title, link = get_article_info(df, edited_item)
+                    if title not in titles:
+                        titles.append(title)
+                    if link not in links:
+                        links.append(link)
+                except (TypeError, KeyError, IndexError):
+                    # Handle the error or simply pass if citation does not have the expected keys
+                    continue
             question_search = retriever.invoke(query)
             for res_item in question_search:
                 edited_item = res_item.metadata["source"].replace("Articles/", "")
                 res_title, res_link = get_article_info(df, edited_item)
-                if res_title not in res_titles:
+                if res_title not in res_titles and res_title not in titles:
                     res_titles.append(res_title)
-                if res_link not in res_links:
+                if res_link not in res_links and res_link not in links:
                     res_links.append(res_link)
             # Build the answer with superscript citations
             answer_with_citations = f"{answer}"
             for i, (title, link) in enumerate(zip(titles, links), start=1):
                 answer_with_citations += f" <sup>[[{i}]({link})]</sup> "
-        else:
-            answer_with_citations = f"{answer}"
 
         # Build the references section with clickable links
-        citations_section = "\n\nCitations:\n" + "\n".join(
-            [f"[{i}]: [{title}]({link})" for i, (title, link) in enumerate(zip(titles, links), start=1)]
-        )
+        if not links:
+            markdown_list = f"{answer_with_citations}"
+        else:
+            citations_section = "\n\nCitations:\n" + "\n".join(
+                [f"[{i}]: [{title}]({link})" for i, (title, link) in enumerate(zip(titles, links), start=1)]
+            )
+            markdown_list = f"{answer_with_citations}{citations_section}"
 
         # Combine answer and citations for final markdown output
-        markdown_list = f"{answer_with_citations}{citations_section}"
-        markdown_list += f"\n\n\nHere is a list of articles that can provide more information about your inquiry:\n"
-        markdown_list += "\n".join(
-            [f"- [{res_title}]({res_link})" for res_title, res_link in zip(res_titles, res_links)])
+
+		if not res_links:
+            return markdown_list
+        else:
+            markdown_list += f"\n\n\nHere is a list of articles that can provide more information about your inquiry:\n"
+            markdown_list += "\n".join([f"- [{res_title}]({res_link})" for res_title, res_link in zip(res_titles, res_links)])
+
+        
     except OutputParserException:
         markdown_list = "There are no articles relevant to your inquiry..."
     return markdown_list