add gemini

README.md

@@ -19,7 +19,7 @@
    Set your OpenAI API key as an environment variable and start the application:
 
    ```shell
-   OPENAI_API_KEY=<your key> python3 app.py
+   GCP_PROJECT_ID=<your project id> OPENAI_API_KEY=<your key> python3 app.py
    ```
 
-   Replace <your key> with your actual OpenAI API key.
+   Replace <your project id> and <your key> with your GCP project ID and OpenAI API key respectively.

app.py

@@ -9,7 +9,7 @@ from fastchat.serve.gradio_web_server import bot_response
 import gradio as gr
 
 # TODO(#1): Add more models.
-SUPPORTED_MODELS = ["gpt-4", "gpt-4-turbo", "gpt-3.5-turbo"]
+SUPPORTED_MODELS = ["gpt-4", "gpt-4-turbo", "gpt-3.5-turbo", "gemini-pro"]
 
 
 def user(user_message):

requirments.txt

@@ -6,6 +6,7 @@ altair==5.2.0
 annotated-types==0.6.0
 anyio==4.2.0
 attrs==23.2.0
+cachetools==5.3.2
 certifi==2023.11.17
 charset-normalizer==3.3.2
 click==8.1.7
@@ -20,8 +21,21 @@ fonttools==4.47.2
 frozenlist==1.4.1
 fschat==0.2.35
 fsspec==2023.12.2
+google-api-core==2.16.1
+google-auth==2.27.0
+google-cloud-aiplatform==1.40.0
+google-cloud-bigquery==3.17.1
+google-cloud-core==2.4.1
+google-cloud-resource-manager==1.11.0
+google-cloud-storage==2.14.0
+google-crc32c==1.5.0
+google-resumable-media==2.7.0
+googleapis-common-protos==1.62.0
 gradio==3.50.2
 gradio_client==0.6.1
+grpc-google-iam-v1==0.13.0
+grpcio==1.60.0
+grpcio-status==1.60.0
 h11==0.14.0
 httpcore==1.0.2
 httpx==0.26.0
@@ -42,15 +56,18 @@ multidict==6.0.4
 networkx==3.2.1
 nh3==0.2.15
 numpy==1.26.3
-openai==1.10.0
+openai==0.28.0
 orjson==3.9.12
 packaging==23.2
 pandas==2.2.0
 peft==0.8.1
 pillow==10.2.0
 prompt-toolkit==3.0.43
+proto-plus==1.23.0
 protobuf==4.25.2
 psutil==5.9.8
+pyasn1==0.5.1
+pyasn1-modules==0.3.0
 pydantic==1.10.14
 pydantic_core==2.16.1
 pydub==0.25.1
@@ -65,10 +82,12 @@ regex==2023.12.25
 requests==2.31.0
 rich==13.7.0
 rpds-py==0.17.1
+rsa==4.9
 ruff==0.1.15
 safetensors==0.4.2
 semantic-version==2.10.0
 sentencepiece==0.1.99
+shapely==2.0.2
 shellingham==1.5.4
 shortuuid==1.0.11
 six==1.16.0