Update app.py

app.py

@@ -16,7 +16,7 @@ from jinja2 import Environment, FileSystemLoader
 import numpy as np
 from sentence_transformers import CrossEncoder
 
-from backend.query_llm import generate_hf, generate_openai
+from backend.query_llm import generate_hf, generate_openai,generate_gemini
 from backend.semantic_search import table, retriever
 
 VECTOR_COLUMN_NAME = "vector"
@@ -93,8 +93,8 @@ def bot(history, api_kind):
 
     if api_kind == "HuggingFace":
          generate_fn = generate_hf
-    elif api_kind == "OpenAI":
-         generate_fn = generate_openai
+    elif api_kind == "Gemini":
+         generate_fn = generate_gemini
     elif api_kind is None:
          gr.Warning("API name was not provided")
          raise ValueError("API name was not provided")
@@ -143,7 +143,7 @@ with gr.Blocks(theme='WeixuanYuan/Soft_dark') as demo:
                 )
         txt_btn = gr.Button(value="Submit text", scale=1)
 
-    api_kind = gr.Radio(choices=["HuggingFace"], value="HuggingFace")
+    api_kind = gr.Radio(choices=["HuggingFace","Gemini"], value="HuggingFace")
 
     prompt_html = gr.HTML()
     try: