Added Gemini

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from neon_llm_chatgpt.chatgpt import ChatGPT
 from neon_llm_palm2.palm2 import Palm2
+from neon_llm_gemini.gemini import Gemini
 from neon_llm_claude.claude import Claude
 from utils import convert_history, PersonaConverter
 import os
@@ -35,6 +36,15 @@ config = {
 }
 palm2 = Palm2(config)
 
+config = {
+    "model": "gemini-pro",
+    "key_path": key_path,
+    "role": role,
+    "context_depth": 3,
+    "max_tokens": 256,    
+}
+gemini = Gemini(config)
+
 key_anthropic = os.environ['ANTHROPIC_API_KEY']
 config = {
     "key": key_anthropic,
@@ -48,7 +58,8 @@ claude = Claude(config)
 
 model_choices = {
     "openai": chatgpt,
-    "google": palm2,
+    "palm2": palm2,
+    "gemini": gemini,
     "anthropic": claude,
 }
 model_choices_list = list(model_choices.keys())