10 billionth update

app.py

@@ -1,25 +1,27 @@
 import gradio as gr
-import openai
+import google.generativeai as genai
 import os
 
-# Set up OpenAI API key (use an environment variable for security)
-api_key = os.getenv("OPENAI_API_KEY")
+# Load environment variables manually
+from dotenv import load_dotenv
+load_dotenv()
+
+# Set up Gemini API key (use an environment variable for security)
+api_key = os.getenv("GEMINI_API_KEY")
 if not api_key:
-    print("Warning: OPENAI_API_KEY is not set. Please add it in Hugging Face Space secrets.")
+    print("Warning: GEMINI_API_KEY is not set. Please add it in Hugging Face Space secrets.")
 else:
     print(f"API Key detected, length: {len(api_key)}")
-    openai.api_key = api_key  # Set the API key globally
+    genai.configure(api_key=api_key)
 
 def chat_with_ai(prompt):
-    """Generates a response from OpenAI's GPT-3.5."""
+    """Generates a response from Google's Gemini AI."""
     if not api_key:
         return "Error: Missing API key. Please configure it in the Hugging Face Space settings."
     try:
-        response = openai.ChatCompletion.create(
-            model="gpt-3.5-turbo",
-            messages=[{"role": "user", "content": prompt}]
-        )
-        return response["choices"][0]["message"]["content"]
+        model = genai.GenerativeModel("gemini-pro")
+        response = model.generate_content(prompt)
+        return response.text
     except Exception as e:
         return f"Error: {str(e)}"
 
@@ -32,4 +34,4 @@ iface = gr.Interface(
 
 # Launch the app
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch()

requirements.txt

@@ -1,2 +1,3 @@
-openai==0.28
-gradio
+google-generativeai
+gradio
+python-dotenv