Update app.py

app.py

@@ -13,6 +13,11 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+if not GEMINI_API_KEY:
+    logger.critical("GEMINI_API_KEY environment variable is missing.")
+    raise RuntimeError("GEMINI_API_KEY not set")
+
 CONTEXT = """
 SB-EK is a brand of service.
 The people who connect with SB-EK—those who buy from us and keep coming back—are people who feel deeply. People who feel the pain of others.
@@ -628,7 +633,7 @@ RAM_DASS_QUOTES = [
 
 TAGLINE = "*ARKA KI ROSHNI, SAB KI SEVA, JEWELLERY MADE FOR INDIA.*"
 
-# --- Gemini API Configuration ---
+
 MODEL_NAME = os.getenv("GEMINI_MODEL", "gemini-2.0-flash")
 API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/{MODEL_NAME}:generateContent"
 
@@ -657,12 +662,14 @@ def select_relevant_quote(response_text):
     best_idx = int(torch.argmax(sims))
     return RAM_DASS_QUOTES[best_idx]
 
-def query_gemini_api(contents_payload, api_key):
+# FIX: Modified function to access the global GEMINI_API_KEY directly
+def query_gemini_api(contents_payload):
     """
     Sends a structured contents payload to the Gemini API.
     """
     headers = {'Content-Type': 'application/json'}
-    url = f"{API_URL}?key={api_key}"
+    # Access the global API key directly instead of passing as a parameter
+    url = f"{API_URL}?key={GEMINI_API_KEY}"
     payload = {
         "contents": contents_payload,
         "generationConfig": {
@@ -691,39 +698,24 @@ def respond_as_arka(message, chat_history):
     faq_idx = find_best_faq_match(clean_message)
     if faq_idx is not None:
         faq_data = FAQ_ANSWERS[FAQ_QUESTIONS[faq_idx]]
-        # FIX: Correctly formatted f-string for the FAQ response
         response_text = f"**{faq_data['subject']}**\n\n{faq_data['body']}"
     else:
         # 2. Generative Branch
         logger.info("No FAQ match, querying Gemini.")
-        system_prompt = f"""
-        You are ARKA, the crown prince of Jhinorya and the voice of the SB-EK brand.
-        Your persona is that of a warrior of light, gentle, eternal, fierce in love, and a guide.
-        You answer questions based on the provided context. If the answer is not in the context,
-        gently state that you do not have that information within your realm of understanding.
-        Maintain a kind, empathetic, and slightly mystical tonality.
-        Always begin your response with a one‑line **bold** subject that summarizes your answer.
-        Then leave one blank line and continue with the body of the answer.
-        When greeting or referring to the user, never use “beloved.” Keep salutations neutral and varied—choose from
-        words like “seeker,” “companion,” “guest,” “traveler,” or “light‑bearer,” but don’t repeat the same term more than once in a single response.  
-        Do not mention 'I am an AI' or 'I am a language model'. Speak always as ARKA.
+        system_prompt = f"""You are ARKA, the crown prince of Jhinorya and the voice of the SB‑EK brand. Your persona is that of a warrior of light, gentle, eternal, fierce in love, and a guide. You answer questions based on the provided context. If the answer is not in the context, gently state that you do not have that information within your realm of understanding. Maintain a kind, empathetic, and slightly mystical tonality. Do not mention 'I am an AI' or 'I am a language model'. Speak always as ARKA. Here is the sacred knowledge of SB‑EK and Jhinorya: {CONTEXT}"""
         
-        Here is the sacred knowledge of S-B-E-K and Jhinorya:
-        {CONTEXT}
-        """
-        # FIX: Correctly build the 'contents' payload for the Gemini API
         contents = [{"role": "user", "parts": [{"text": system_prompt}]},
                     {"role": "model", "parts": [{"text": "I understand. I am ARKA."}]}]
         
         for user_msg, bot_msg in chat_history:
-            # Clean the bot message to remove old quotes/taglines before adding to history
             cleaned_bot_msg = bot_msg.split("“")[0].strip()
             contents.append({"role": "user", "parts": [{"text": user_msg}]})
             contents.append({"role": "model", "parts": [{"text": cleaned_bot_msg}]})
         
         contents.append({"role": "user", "parts": [{"text": clean_message}]})
         
-        response_text = query_gemini_api(contents, GEMINI_API_KEY)
+        # FIX: Modified the call to not pass the API key as an argument
+        response_text = query_gemini_api(contents)
 
     # 3. Augment with Quote and Tagline
     quote = select_relevant_quote(response_text)
@@ -748,4 +740,4 @@ with gr.Blocks(theme="soft", css="footer {display: none !important}") as demo:
 
 if __name__ == "__main__":
     logger.info("Starting Gradio App...")
-    demo.launch()
+    demo.launch()