Upload folder using huggingface_hub

frontend/app.py

@@ -181,12 +181,13 @@ def Search(request, search_results=[]):
     print(
         f"Search: Fetching results for query: {query_value}, ranking: {ranking_value}"
     )
-
     return Div(
         Div(
             SearchBox(query_value=query_value, ranking_value=ranking_value),
             Div(
-                LoadingMessage(),  # Show the loading message initially
+                LoadingMessage()
+                if not search_results
+                else SearchResult(search_results),
                 id="search-results",  # This will be replaced by the search results
             ),
             cls="grid",

main.py

@@ -98,7 +98,14 @@ def get(request):
                 cls="grid",
             )
         )
-
+    # Generate a unique query_id based on the query and ranking value
+    query_id = generate_query_id(query_value + ranking_value)
+    # See if results are already in cache
+    if result_cache.get(query_id) is not None:
+        print(f"Results for query_id {query_id} already in cache")
+        result = result_cache.get(query_id)
+        search_results = get_results_children(result)
+        return Layout(Search(request, search_results))
     # Show the loading message if a query is provided
     return Layout(Search(request))  # Show SearchBox and Loading message initially
 
@@ -116,15 +123,12 @@ async def get(request, query: str, nn: bool = True):
     # Generate a unique query_id based on the query and ranking value
     query_id = generate_query_id(query + ranking_value)
     # See if results are already in cache
-    if result_cache.get(query_id):
+    if result_cache.get(query_id) is not None:
         print(f"Results for query_id {query_id} already in cache")
         result = result_cache.get(query_id)
         search_results = get_results_children(result)
-        # If task is completed, return the results, but no query_id
-        if task_cache.get(query_id):
-            return SearchResult(search_results, None)
-        # If task is not completed, return the results with query_id
         return SearchResult(search_results, query_id)
+    # Run the embedding and query against Vespa app
     task_cache.set(query_id, False)
     model = app.manager.model
     processor = app.manager.processor