Update app.py

app.py

@@ -20,6 +20,7 @@ from paper.download import (
 from paper.parser import extract_text_and_figures
 from gen.gemini import get_basic_qa, get_deep_qa
 from constants.styles import STYLE
+from constants.js import UPDATE_SEARCH_RESULTS, UPDATE_IF_TYPE
 
 from apscheduler.schedulers.background import BackgroundScheduler
 
@@ -321,117 +322,6 @@ def search(search_in, max_results=3):
         )
     )
 
-UPDATE_SEARCH_RESULTS = f"""
-function search(searchIn, maxResults = 3) {{
-    if (searchIn.trim().length > 0) {{
-        const results = [];
-        let titles = {list(titles)};
-
-        for (const title of titles) {{ // Assuming 'titles' is an array defined elsewhere
-            if (results.length > 10) {{
-                break;
-            }} else {{
-                if (title.toLowerCase().includes(searchIn.toLowerCase())) {{ // JavaScript's equivalent to Python's 'in'
-                    results.push(title);
-                }}
-            }}
-        }}
-
-        // Handle UI elements (Explanation below)
-        const resultElements = [1,2,3,4,5,6,7,8,9,10].map(index => {{
-            return results[index - 1] || '';
-        }});
-
-        if (resultElements[0] == '') {{
-            document.getElementById('search_r1').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r1').style.display = 'block';
-        }}
-
-        if (resultElements[1] == '') {{
-            document.getElementById('search_r2').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r2').style.display = 'block';
-        }}
-
-        if (resultElements[2] == '') {{
-            document.getElementById('search_r3').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r3').style.display = 'block';
-        }}
-
-        if (resultElements[3] == '') {{
-            document.getElementById('search_r4').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r4').style.display = 'block';
-        }}
-
-        if (resultElements[4] == '') {{
-            document.getElementById('search_r5').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r5').style.display = 'block';
-        }}
-
-        if (resultElements[5] == '') {{
-            document.getElementById('search_r6').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r6').style.display = 'block';
-        }}
-
-        if (resultElements[6] == '') {{
-            document.getElementById('search_r7').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r7').style.display = 'block';
-        }}
-
-        if (resultElements[7] == '') {{
-            document.getElementById('search_r8').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r8').style.display = 'block';
-        }}
-
-        if (resultElements[8] == '') {{
-            document.getElementById('search_r9').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r9').style.display = 'block';
-        }}
-
-        if (resultElements[9] == '') {{
-            document.getElementById('search_r10').style.display = 'none';
-        }} else {{
-            document.getElementById('search_r10').style.display = 'block';
-        }}
-
-        return resultElements; 
-    }} else {{
-        document.getElementById('search_r1').style.display = 'none';
-        document.getElementById('search_r2').style.display = 'none';
-        document.getElementById('search_r3').style.display = 'none';
-        document.getElementById('search_r4').style.display = 'none';
-        document.getElementById('search_r5').style.display = 'none';
-        document.getElementById('search_r6').style.display = 'none';
-        document.getElementById('search_r7').style.display = 'none';
-        document.getElementById('search_r8').style.display = 'none';
-        document.getElementById('search_r9').style.display = 'none';
-        document.getElementById('search_r10').style.display = 'none';
-
-        return ['', '', '', '', '', '', '', '', '', '']
-    }}
-}}
-"""
-
-UPDATE_IF_TYPE = f"""
-function chage_if_type(if_type) {{
-    if (if_type == 'Q&As') {{
-        document.getElementById('chat_block').style.display = 'none';
-        document.getElementById('qna_block').style.display = 'block';
-    }} else {{
-        document.getElementById('chat_block').style.display = 'block';
-        document.getElementById('qna_block').style.display = 'none';        
-    }}
-}}
-"""
-
 def set_date(title):
     paper = title2qna[title]
     date = paper["target_date"].strftime("%Y-%m-%d")
@@ -666,7 +556,7 @@ with gr.Blocks(css=STYLE, theme=gr.themes.Soft()) as demo:
             search_r1, search_r2, search_r3, search_r4, search_r5,
             search_r6, search_r7, search_r8, search_r9, search_r10
         ],
-        js=UPDATE_SEARCH_RESULTS,
+        js=UPDATE_SEARCH_RESULTS % str(list(titles)),
         fn=None
     )