expansion fixed

app.py

@@ -76,22 +76,21 @@ def llm_expand_query(query):
 
 
   prompt = f"""
-    Expand this query into a the query format used for a fulltext search
+    Expand this query into a the query format used for a search
     over the INSPIRE HEP database. Propose alternatives of the query to
-    maximize the recall and join those variantes using OR operators and
-    prepend each variant with the ft prefix. Just provide the expanded
-    query, without explanations.
+    maximize the recall and join those variantes using OR operators. 
+    Just provide the expanded query, without explanations.
 
     Example of query:
     how far are black holes?
 
     Expanded query:
-    ft "how far are black holes" OR ft "distance from black holes" OR ft
-    "distances to black holes" OR ft "measurement of distance to black
-    holes"  OR ft "remoteness of black holes"  OR ft "distance to black
-    holes"  OR ft "how far are singularities"  OR ft "distance to
-    singularities"  OR ft "distances to event horizon"  OR ft "distance
-    from Schwarzschild radius" OR ft "black hole distance"
+    "how far are black holes" OR "distance from black holes" OR
+    "distances to black holes" OR "measurement of distance to black
+    holes"  OR "remoteness of black holes" OR "distance to black
+    holes"  OR "how far are singularities" OR "distance to
+    singularities" OR "distances to event horizon" OR "distance
+    from Schwarzschild radius" OR "black hole distance"
 
     Query: {query}
 
@@ -213,9 +212,9 @@ def clean_refs(answer, results):
 def search(query, progress=gr.Progress()):
     time.sleep(1)
     progress(0, desc="Expanding query...")
-    query = llm_expand_query(query)
+    expanded_query = llm_expand_query(query)
     progress(0.25, desc="Searching INSPIRE HEP...")
-    results = search_inspire(query)
+    results = search_inspire(expanded_query)
     progress(0.50, desc="Generating answer...")
     context = results_context(results)
     prompt = user_prompt(query, context)
@@ -244,6 +243,7 @@ with gr.Blocks() as demo:
             search_btn = gr.Button("Search")
             examples = gr.Examples([["Which one is closest star?"], ["In which particles does the Higgs Boson decay to?"]], query)
             gr.HTML("<img src='https://sinai.ujaen.es/sites/default/files/SINAI%20-%20logo%20tx%20azul%20%5Baf%5D.png' width='200'></img>")
+            gr.HTML("<img src='https://diariodigital.ujaen.es/sites/default/files/general/logo-uja.svg' width='200'></img>")
         with gr.Column():
            results = gr.Markdown("Answer will appear here...", label="Search Results", )
         search_btn.click(fn=search, inputs=query, outputs=results, api_name="search", show_progress=True)