ai-chatbot / app.py
rvivord's picture
Update app.py
93410a7 verified
import gradio as gr
import google.generativeai as genai
from dotenv import load_dotenv
import os
load_dotenv()
# print(os.getenv("GEMINI_API"))
genai.configure(api_key=os.getenv("GEMINI_API"))
# model = genai.GenerativeModel("gemini-2.5-flash-preview-04-17")
model = genai.GenerativeModel("gemini-2.0-flash")
conversation_history = []
def chate(message, history):
conversation_history.append({"role": "user", "parts": [message]})
try:
response = model.generate_content(conversation_history)
response = model.generate_content(conversation_history)
conversation_history.append({"role": "model", "parts": [response.text]})
return response.text
except Exception as e:
return "Error: " + str(e)
demo = gr.ChatInterface(
fn=chate,
title="VIVORD AI",
description="Ask anything you want!",
examples=["Hello", "What is Python?", "What is the weather in Phnom Penh?"],
theme="ocean",
type="messages",
)
demo.launch(share=True)