Update app.py

app.py

@@ -2,26 +2,17 @@ import os
 import gradio as gr
 from google import genai
 
-# Read API key
-GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
-
 def qa_ai_bot(message, history):
-    if not GEMINI_API_KEY:
-        return "❌ GEMINI_API_KEY not set in Space variables."
+    api_key = os.getenv("AIzaSyCH9Pf6Bqe2nDGPKcwaqK91VbiD6_CoSqw")
 
-    client = genai.Client(api_key=GEMINI_API_KEY)
-
-    prompt = f"""
-You are an expert QA Automation Engineer.
-Answer with examples when required.
+    if not api_key:
+        return "❌ GEMINI_API_KEY not set in Space variables."
 
-User Question:
-{message}
-"""
+    client = genai.Client(api_key=api_key)
 
     response = client.models.generate_content(
         model="gemini-1.5-flash",
-        contents=prompt
+        contents=f"You are a QA Automation expert.\n\n{message}"
     )
 
     return response.text
@@ -29,7 +20,7 @@ User Question:
 gr.ChatInterface(
     fn=qa_ai_bot,
     title="🤖 AI QA Automation Assistant",
-    description="Ask about Selenium, Cypress, API Testing, Bug Reports, Test Plans",
+    description="Ask about Selenium, Cypress, API Testing, Bug Reports",
     examples=[
         "Write Selenium POM login test",
         "Generate Cypress test for signup page",