Update app.py

app.py

@@ -595,17 +595,19 @@ with gr.Blocks(title="Archaeological Query Engine") as app:
                         elem_classes=["results-output"]
                     )
             
-            # Function to handle the Generate Answer button click
-            def on_generate_answer(city, question, max_sources, api_key, temperature):
+              def on_generate_answer(city, question, max_sources, api_key, temperature):
                 if not question or question.strip() == "":
                     return "Please enter a question to generate an answer."
                 
-                # Use the provided Groq API key directly
-                # Replace this with your actual Groq API key
-                groq_api_key = Groq(api_key=os.environ.get("GROQ_API"))
+                # Get the Groq API key from environment variable
+                groq_api_key = os.environ.get("GROQ_API")
+                
+                if not groq_api_key:
+                    return "Error: GROQ_API environment variable not set. Please set your Groq API key in the environment."
                 
                 try:
-                                      return generate_answer_with_groq(city, question, max_sources, groq_api_key, temperature)
+                    # Use the API key string (not a Groq client object)
+                    return generate_answer_with_groq(city, question, max_sources, groq_api_key, temperature)
                 except Exception as e:
                     return f"Error: {str(e)}"