Upload app.py

app.py

@@ -147,7 +147,6 @@ def generate_unique_string():
 
 def llm_response(query, session_id):
     titles, links, res_titles, res_links = [], [], [], []
-    print(session_id)
     unique_id = session_id
     config = {"configurable": {"thread_id": unique_id }}
     try:
@@ -156,42 +155,57 @@ def llm_response(query, session_id):
                                          start_on="human", end_on=("human", "tool"), include_system=True,)
         modified_query = runnable.invoke({"input": query, "chat_history": filtered_history}).content
         result = chain.invoke({"input": modified_query}, config=config)
+        if not result['answer']['cited_answer'][0]["answer"]:
+            return "There is no direct information in our database relevant to your inquiry. Please contact [FORTIFIED customer support](https://fortifiedhome.org/contact/) for further assistance."
         answer = result['answer']['cited_answer'][0]["answer"]
         history = get_session_history(unique_id)
         history.add_user_message(modified_query)
         history.add_ai_message(answer)
-        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)
-        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:
-                res_titles.append(res_title)
-            if res_link not in res_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> "  # Append superscript citation numbers to the answer text
-
-        # 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)]
-        )
-
-        # 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 resources 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 result['answer']['cited_answer'][1]['citations']:
+            answer_with_citations = f"{answer}"
+        else:
+            citations = result['answer']['cited_answer'][1]['citations']
+            for citation in citations:
+                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 and res_title not in titles:
+                    res_titles.append(res_title)
+                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> "  # Append superscript citation numbers to the answer text
+
+
+        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}"
+
+        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 is no information relevant to your inquiry in my current resources. Please contact [FORTIFIED customer support](https://fortifiedhome.org/contact/) for further assistance."
     return markdown_list