import os
import google.generativeai as genai
import gradio as gr

# Replace with your actual API key (ensure you are using the correct endpoint)
genai.configure(api_key=os.environ["GOOGLE_API_KEY"]) 

generation_config = {
  "temperature": 0.7,
  "top_p": 0.95,
  "top_k": 64,
  "max_output_tokens": 128,
  "response_mime_type": "text/plain",
}

# Create the Gemini 1.5 Flash model
model = genai.GenerativeModel(
  model_name="gemini-1.5-flash",
  generation_config=generation_config,
)

# Define a function to generate responses
def generate_response(chatbot):
  prompt = ""
  for turn in chatbot:
      prompt += f"{turn[0]}: {turn[1]}\n"  # Format the conversation history

  response = model.generate_text(
      prompt=prompt,
      temperature=0.7,
      max_output_tokens=50,
  )

  return response.result

# Create Gradio interface
with gr.Blocks() as interface:
  chatbot = gr.Chatbot() 
  msg = gr.Textbox() 
  clear = gr.ClearButton([msg, chatbot]) 

  msg.submit(generate_response, [chatbot], chatbot) 

interface.launch()