anji19's picture
Update app.py
1873b55 verified
raw
history blame contribute delete
938 Bytes
from dotenv import load_dotenv
load_dotenv()
import os
import google.generativeai as genai
import gradio as gr
GOOGLE_API_KEY=os.environ.get("google_api_key")
genai.configure(api_key=(GOOGLE_API_KEY))
# Configure Google Generative AI
#genai.configure(api_key=os.getenv(GOOGLE_API_KEY))
model = genai.GenerativeModel("gemini-pro")
def get_gemini_responces(message, chat_history):
# Assuming the message is a question for the Gemini model
responce = model.generate_content(message)
# Append the model's response to the chat history
chat_history.append((message, responce.text))
return "", chat_history
# Define the Gradio interface with a Chatbot component
with gr.Blocks() as demo:
chatbot = gr.Chatbot()
msg = gr.Textbox()
clear = gr.ClearButton([msg, chatbot])
msg.submit(get_gemini_responces, [msg, chatbot], [msg, chatbot])
# Launch the Gradio app
if __name__ == "__main__":
demo.launch()