Merge branch 'main' of https://huggingface.co/spaces/pendrag/fenybot-ir

app.py

@@ -1,4 +1,6 @@
 from openai import OpenAI
+import google.generativeai as genai
+
 import os
 import requests
 import json
@@ -122,7 +124,10 @@ def llm_expand_query(query):
     presence_penalty=0
   )
 
-  return response.choices[0].message.content
+    return response.choices[0].message.content
+  else:
+     response = genai.GenerativeModel("gemini-1.5-flash").generate_content(prompt)
+     return response.text
 
 def llm_generate_answer(prompt):
   """ Generate a response from the LLM """
@@ -168,7 +173,11 @@ def llm_generate_answer(prompt):
     presence_penalty=0
   )
 
-  return response.choices[0].message.content
+    return response.choices[0].message.content
+  
+  else:
+     response = genai.GenerativeModel("gemini-1.5-flash").generate_content(system_desc + "\n\n" + prompt)
+     return response.text
 
 def clean_refs(answer, results):
   """ Clean the references from the answer """
@@ -202,9 +211,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)
@@ -217,21 +226,28 @@ def search(query, progress=gr.Progress()):
 
 # ----------- MAIN ------------------------------------------------------------
 
-client = OpenAI()
+if GENAI_API == "openai":
+  client = OpenAI()
+else:
+   genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
 
 with gr.Blocks() as demo:
     gr.Markdown("# Feynbot on INSPIRE HEP Search")
     gr.Markdown("""Specialized academic search tool that combines traditional 
                 database searching with AI-powered query expansion and result 
-                synthesis, focused on physics research papers.""")
+                synthesis, focused on High Energy Physics research papers.""")
     with gr.Row():
         with gr.Column():
             query = gr.Textbox(label="Search Query")
             search_btn = gr.Button("Search")
             examples = gr.Examples([["Which one is closest star?"], ["In which particles does the Higgs Boson decay to?"]], query)
+            with gr.Row():
+              gr.HTML("<a href='https://sinai.ujaen.es'><img src='https://sinai.ujaen.es/sites/default/files/SINAI%20-%20logo%20tx%20azul%20%5Baf%5D.png' width='200'></img></a>")
+              gr.HTML("<a href='https://www.ujaen.es'><img src='https://diariodigital.ujaen.es/sites/default/files/general/logo-uja.svg' width='180'></img></a>")
         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)
+    
 
 
 demo.launch()

requirements.txt

@@ -1,4 +1,5 @@
 gradio
 openai
 requests
-httpx<0.28
+httpx<0.28
+google-generativeai