fixed some bugs

app.py

@@ -1,20 +1,22 @@
-import openai
+import google.generativeai as genai
 import streamlit as st
-from openai import OpenAI, OpenAIError
 import os
+from dotenv import load_dotenv
+
+# Load API key from environment variables
+load_dotenv()
+api_key = os.getenv("GCP_API_KEY")
 
-# Set OpenAI API Key
-api_key = os.getenv("OPENAI_API_KEY")
 if not api_key:
-    st.error("API key is missing. Set OPENAI_API_KEY as an environment variable.")
+    st.error("API key is missing. Set GCP_API_KEY as an environment variable.")
     st.stop()
 
-client = OpenAI(api_key=api_key)
-
+# Configure Gemini API
+genai.configure(api_key=api_key)
 
 # Streamlit UI
-st.title("AI Text Generator")
-st.write("Generate text using OpenAI's language model.")
+st.title("AI Text Generator using Gemini API")
+st.write("Generate text using Google's Gemini AI model.")
 
 # User Inputs
 prompt = st.text_area("Enter your prompt:", "Once upon a time...")
@@ -31,21 +33,22 @@ if st.button("Generate Text"):
         st.error("Please enter a prompt.")
     else:
         full_prompt = f"{prompt}\n\nFormat: {output_format}\nTone: {tone}\nCreative Mode: {creative_mode}"
+        
         try:
-            response = client.chat.completions.create(
-                model="gpt-3.5-turbo",
-                messages=[{"role": "user", "content": full_prompt}],
-                temperature=creativity,
-                max_tokens=max_length,
-                n=n_responses
+            model = genai.GenerativeModel("gemini-pro")  # Use the latest Gemini model
+            response = model.generate_content(
+                full_prompt,
+                generation_config=genai.types.GenerationConfig(
+                    temperature=creativity,  # Now correctly used inside generation_config
+                    max_output_tokens=max_length
+                )
             )
-        except OpenAIError as e:
-            if "insufficient_quota" in str(e):
-                st.error("You have exceeded your API quota. Please check your OpenAI account billing and usage.")
-            elif "model_not_found" in str(e):
-                st.error("The specified AI model does not exist or you do not have access to it.")
-            else:
-                st.error("An error occurred while generating text. Please try again later.")
+            
+            st.subheader("Generated Response")
+            st.write(response.text)
+        
+        except Exception as e:
+            st.error(f"An error occurred: {e}")
 
 # Feedback Section
 st.subheader("Feedback")

requirements.txt

@@ -1,2 +1,3 @@
 streamlit
-openai
+google-generativeai
+python-dotenv