Upload folder using huggingface_hub

main.py

@@ -88,30 +88,32 @@ def llm_response(query):
     titles, links, res_titles, res_links = [], [], [], []
     try:
         result = chain.invoke({"input": query})
+        if not result['answer']['cited_answer']:
+            return "There are no articles relevant to your inquiry..."
         answer = result['answer']['cited_answer'][0]["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)
-        # markdown_list = f"{answer}\n\nCitations:\n" + "\n".join(
-        #     [f"- [{title}]({link})" for title, link in zip(titles, links)])
-
-        # 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 result['answer']['cited_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> "
+        else:
+            answer_with_citations = f"{answer}"
 
         # Build the references section with clickable links
         citations_section = "\n\nCitations:\n" + "\n".join(