File size: 736 Bytes
45dfc95 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import streamlit
import langchain
from langchain.llms import GenerativeModel
# Replace with your Gemini API key
API_KEY = "AIzaSyBI2C5bFa0JsvCCyabbyANg8LUjcpqUiVM"
# Configure the model
llm = GenerativeModel("google-llm/text-davinci-003", api_key=API_KEY)
def generate_response(user_input):
"""
Sends user input to Gemini and returns its response.
Args:
user_input: The user's message.
Returns:
The generated response from Gemini.
"""
prompt ={user_input}
response = llm.generate_content(prompt=prompt)
return response.content[0]["text"]
while True:
user_input = input("enter your text")
if user_input.lower() == "quit":
break
response = generate_response(user_input)
st.write( {response}) |