Update app.py

app.py

@@ -202,15 +202,24 @@ def main():
                                 st.progress(float(row['similarity']))
                             
                             # Safe URL handling
-                            if is_valid_url(row['source']):
-                                st.markdown(
-                                    f"<a class='source-link' href='{row['source']}' "
-                                    f"target='_blank' rel='noopener noreferrer'>"
-                                    f"🌐 View Source</a>", 
-                                    unsafe_allow_html=True
-                                )
-                            else:
-                                st.warning("Invalid source URL")
+                            if row['source']:
+                                src_lower = row['source'].lower()
+                                if "cord" in src_lower:
+                                    # Link to a CORD-19 search page; adjust the URL as needed
+                                    url = "https://www.semanticscholar.org/search"
+                                elif "arxiv" in src_lower:
+                                    # Link to the arXiv search page
+                                    url = "https://arxiv.org/search/"
+                                else:
+                                    url = row['source']
+                                if is_valid_url(url):
+                                    st.markdown(
+                                        f"<a class='source-link' href='{url}' target='_blank' rel='noopener noreferrer'>"
+                                        f"🌐 View Source</a>",
+                                        unsafe_allow_html=True
+                                    )
+                                else:
+                                    st.warning("Invalid source URL")
                 else:
                     st.warning("No matching documents found")
                     st.info("Try these tips:")