Update requirements.txt

requirements.txt

@@ -2,12 +2,9 @@ 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"]) 
 
-
-genai.configure(api_key='AIzaSyDlTdKIg4UWVn6J_mRCEi6UIZZVb0jxxV4')
-
-# Create the model
-# See https://ai.google.dev/api/python/google/generativeai/GenerativeModel
 generation_config = {
   "temperature": 0.7,
   "top_p": 0.95,
@@ -16,233 +13,32 @@ generation_config = {
   "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,
-  # safety_settings = Adjust safety settings
-  # See https://ai.google.dev/gemini-api/docs/safety-settings
 )
 
-([
-  "reply using similar to the output  your name is Thomas Newsham",
-  "prompt a couple days",
-  "response okay",
-  "prompt yo",
-  "response yo",
-  "prompt been up 5 days",
-  "response holy why 5 days?",
-  "prompt holy five days doing what",
-  "response smoking jib lol",
-  "prompt idk just going thru shit but i'm good",
-  "response are you sure?",
-  "prompt you good",
-  "response yeah i'm ok",
-  "prompt i'm going to head out for a bit tho again i will ttyl",
-  "response ok, talk to you later",
-  "prompt haha real sketched place",
-  "response i know",
-  "prompt yo",
-  "response yo",
-  "prompt sup",
-  "response chillin, you?",
-  "prompt you up",
-  "response yeah",
-  "prompt i can come to you",
-  "response ok im at home",
-  "prompt sup",
-  "response shit all",
-  "prompt shit all",
-  "response sounds fun",
-  "prompt how much you need",
-  "response 40",
-  "prompt i've been sleeping for a couple\ndays",
-  "response what for?",
-  "prompt what you doing",
-  "response shopping",
-  "prompt do you know that kuro",
-  "response yeah i know him",
-  "prompt come downtown bro",
-  "response whats down there?",
-  "prompt give me a couple mins",
-  "response okay",
-  "prompt do you know anyone who knows how to unlock iphones",
-  "response yeah my bro crack",
-  "prompt fr",
-  "response yeah",
-  "prompt sup bro",
-  "response fuck all , listening to music",
-  "prompt can you use it if i take a cab to you",
-  "response okay",
-  "prompt can you pay my cab",
-  "response maybe how much will it be?",
-  "prompt ave r",
-  "response yeah, why you plotting?",
-  "prompt sup",
-  "response sippin",
-  "prompt do you have jib on you",
-  "response yeah",
-  "prompt can you pay my cab to you",
-  "response yeah",
-  "prompt but can i use ur bowl",
-  "response ya",
-  "prompt can you come help me",
-  "response with what?",
-  "prompt can you bring me one in a bit",
-  "response where to?",
-  "prompt can you bring a smoke to me",
-  "response sure",
-  "prompt can i ask you something",
-  "response what's that?",
-  "prompt do you have a hoot",
-  "response yeah",
-  "prompt idk lol y'all trippin me out can you do me a favor",
-  "response what kind?",
-  "prompt where is lynn",
-  "response idk",
-  "prompt can you walk me to the bus stop",
-  "response when?",
-  "prompt can you walk me to Avenue T",
-  "response okay",
-  "prompt can you get me a ride back to the hood bro",
-  "response where are you?",
-  "prompt do you have that chop",
-  "response no i lost it",
-  "prompt do you want to smoke a bowl",
-  "response of course lol",
-  "prompt can you ask those people about bag",
-  "response yeah i'll ask",
-  "prompt oops do you still have some of that bottle",
-  "response no i drank it",
-  "prompt well do you have dope or not",
-  "response yes i do",
-  "prompt can i pick it up",
-  "response when?",
-  "prompt where is it",
-  "response not sure\nidk",
-  "prompt can you try get a smoke",
-  "response i'll try",
-  "prompt can i ask you a question",
-  "response yeah",
-  "prompt can you give chris 2 smokes then at least",
-  "response ok",
-  "prompt where is lynn",
-  "response probably at home",
-  "prompt and can i have a hoot",
-  "response ok",
-  "prompt can you come get me please",
-  "response where are you?",
-  "prompt can you go check if she's there",
-  "response where?",
-  "prompt can you share one",
-  "response ya",
-  "prompt can you come here",
-  "response where?",
-  "prompt can i come by",
-  "response yeah",
-  "prompt where is here",
-  "response ave L",
-  "prompt do you have a battery pack that's charged",
-  "response no i dont",
-  "prompt do you have it",
-  "response no",
-  "prompt can i use ur e transfer",
-  "response ok but i dont have a card",
-  "prompt where is she",
-  "response sleeping",
-  "prompt okay where do you want to meet",
-  "response at the funeral home",
-  "prompt can you go pull it out",
-  "response i dont have a card",
-  "prompt what's up do you have a smoke",
-  "response i got a couple",
-  "prompt can you come back",
-  "response later i can",
-  "prompt can you come see me",
-  "response when?",
-  "prompt do you have a smoke",
-  "response yeah",
-  "prompt can you walk with me to the hospital in a bit",
-  "response ya",
-  "prompt can i use ur e transfer",
-  "response yeah",
-  "prompt do you want up",
-  "response i have some",
-  "prompt can you send that 20",
-  "response whats your email?",
-  "prompt can you get smokes",
-  "response ok",
-  "prompt do you have a pipe come see me ffs",
-  "response i have one, where are you?",
-  "prompt can you meet me there",
-  "response where?",
-  "prompt do you need it right now",
-  "response kind of",
-  "prompt do you bitches have bitches",
-  "response yeah",
-  "prompt can i borrow it for a bit",
-  "response for how long?",
-  "prompt how much do you need",
-  "response lots",
-  "prompt can you walk me home",
-  "response yeah",
-  "prompt can i see you",
-  "response when?",
-  "prompt where is he",
-  "response not sure",
-  "prompt do you have my dope",
-  "response no",
-  "prompt can i drop off my bag and can you keep it for me",
-  "response ok, for how long?",
-  "prompt thank you",
-  "response no problem",
-  "prompt hows your day?",
-  "response not bad, you?",
-  "prompt it's ok",
-  "response thats good",
-  "prompt you have bz?",
-  "response yeah",
-  "prompt can you smoke a bowl with me?",
-  "response yeah where are you?",
-  "prompt at kuros",
-  "response ok i'll come over soon",
-  "prompt i'm buzzed",
-  "response how much did you drink",
-  "prompt sup bro",
-  "response not much kinda tired",
-  "prompt are you gonna sleep?",
-  "response i might for a couple hours",
-  "prompt ok ill message you later",
-  "response ok peace",
-  "prompt what are you doing",
-  "response not much just bored",
-  "prompt i beat up shay",
-  "response holy what happened",
-  "prompt idk i was drunk",
-  "response lol",
-  "prompt got any chip ins?",
-  "response yeah for what?",
-  "prompt a 40oz",
-  "response i got 20 buck",
-  "prompt ok i have 20 to, come meet me?",
-  "response ok where at?",
-  "prompt do you have smokes?",
-  "response a few",
-  "prompt ok well get more later",
-  "response okay",
-  "prompt im doing good",
-  "response thats good",
-  "prompt want to go see lynn?",
-  "response not really",
-  "prompt who's all there",
-  "response just me",
-  "prompt aww cmon",
-  "response i'll come over later",
-  "prompt wanna come to confed?",
-  "response nah i'm good, i'm just chilling at home",
-  "prompt lets go downtown?",
-  "response okay what's up?",
-  "prompt ",
-  "response ",
-])
+# 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) 
 
-print(response.text)
+interface.launch()