Spaces:
Running
Running
File size: 2,507 Bytes
756bb52 72897f5 38c454e 2743781 72897f5 756bb52 72897f5 756bb52 72897f5 756bb52 72897f5 756bb52 72897f5 756bb52 72897f5 756bb52 72897f5 756bb52 72897f5 756bb52 72897f5 756bb52 72897f5 756bb52 72897f5 |
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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 |
import streamlit as st
import google.generativeai as genai
import os
# Retrieve the API key from environment variables
api_key = os.getenv("google_api_key")
# Check if API key is provided
if not api_key:
st.error("API key is not set. Please set the GOOGLE_API_KEY environment variable.")
else:
# Configure Google Gemini API
genai.configure(api_key=api_key)
# Sidebar for input questions
st.sidebar.title("Prompt Engineering for AI Application")
# 1. Functionality of the AI application
functionality = st.sidebar.text_input(
"What specific functionality do you want the AI application to have?",
placeholder="e.g., image classification, text generation"
)
# 2. Preferred programming language or framework
language = st.sidebar.text_input(
"What programming language or framework do you prefer to use?",
placeholder="e.g., Python, TensorFlow"
)
# 3. Specific libraries, tools, or models
libraries = st.sidebar.text_input(
"Are there any specific libraries, tools, or models you want to integrate?",
placeholder="e.g., OpenAI's GPT, Whisper"
)
# 4. Input data or format
input_data = st.sidebar.text_input(
"What input data or format will the application work with?",
placeholder="e.g., text, images"
)
# 5. Specific requirements or constraints
requirements = st.sidebar.text_input(
"Do you have any specific requirements or constraints for the output?",
placeholder="e.g., performance, accuracy"
)
# Combine the user inputs into a prompt
prompt = f"""
Write a code to develop an AI application with the following details:
- Functionality: {functionality}
- Programming language or framework: {language}
- Specific libraries, tools, or models: {libraries}
- Input data or format: {input_data}
- Specific requirements or constraints: {requirements}
"""
# Main content
st.title("AI Code Generator using Google's Gemini Model")
st.write("Fill out the details in the sidebar to generate a proper prompt.")
if st.sidebar.button("Generate Code"):
model = genai.GenerativeModel('gemini-1.5-flash')
response = model.generate_content(prompt)
st.subheader("Generated Code:")
st.code(response.text, language='python')
# Footer
st.sidebar.write("---")
st.sidebar.write("This application helps you generate a prompt to write code for an AI application.")
|