Update app.py

app.py

@@ -2,65 +2,69 @@ import streamlit as st
 import google.generativeai as genai
 import os
 
-# Set the API key directly in the script (this should be done securely in a real application)
-os.environ["API_KEY"] = google_api_key
+# Retrieve the API key from environment variables
+api_key = os.getenv("GOOGLE_API_KEY")
 
-# Configure Google Gemini API
-genai.configure(api_key=os.environ["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")
+    # 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"
-)
+    # 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"
-)
+    # 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"
-)
+    # 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"
-)
+    # 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"
-)
+    # 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}
-"""
+    # 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.")
+    # 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')
+    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.")
+    # Footer
+    st.sidebar.write("---")
+    st.sidebar.write("This application helps you generate a prompt to write code for an AI application.")