devopscloudram/vision
0
1from dotenv import load_dotenv
2
3load_dotenv()
4import streamlit as st
5import os
6import google.generativeai as genai
7
8genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
9
10#function to load gemini pro model adn get resp
11model=genai.GenerativeModel("gemini-pro")
12def get_resp(question):
13 resp=model.generate_content(question)
14 return resp.text
15
16#UI code
17st.set_page_config(page_title="Ramz GenAI")
18st.header("Ramz GenAI")
19input=st.text_input("Input:",key="input")
20
21submit=st.button("Ask :question:")
22
23if submit:
24 resp=get_resp(input)
25 st.write(resp)