Update app.py

app.py

@@ -549,73 +549,74 @@ with gr.Blocks(title="Archaeological Query Engine") as app:
             )
         
         # Add new tab for AI-generated answers using Groq API
-        with gr.TabItem("AI Answers (Groq API)"):
-            gr.Markdown("Ask questions about the dataset and get AI-generated answers using the Groq API with the deepseek-r1-distill-llama-70b model.")
-            
-            with gr.Row():
-                with gr.Column():
-                    # API key is now hardcoded in the code
-                                     
-                    ai_city_dropdown = gr.Dropdown(
-                        choices=city_names,
-                        value=city_names[0] if city_names else None,
-                        label="Select City"
-                    )
-                    
-                    question_input = gr.Textbox(
-                        label="Ask a Question",
-                        placeholder="E.g., What was the historical significance of this ancient city?",
-                        lines=3
-                    )
-                    
-                    max_sources_slider = gr.Slider(
-                        minimum=1,
-                        maximum=10,
-                        value=3,
-                        step=1,
-                        label="Maximum Number of Sources to Consider",
-                        info="Higher values may provide more comprehensive answers but will take longer"
-                    )
-                    
-                    temperature_slider = gr.Slider(
-                        minimum=0.0,
-                        maximum=1.0,
-                        value=0.3,
-                        step=0.1,
-                        label="Temperature",
-                        info="Lower values create more focused answers, higher values create more creative ones"
-                    )
-                    
-                    generate_button = gr.Button("Generate Answer")
-                
-                with gr.Column():
-                    answer_output = gr.HTML(
-                        label="AI-Generated Answer",
-                        value="",
-                        elem_classes=["results-output"]
-                    )
-            
-              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."
-                
-                # 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:
-                    # 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)}"
-            
-            generate_button.click(
-                fn=on_generate_answer,
-                inputs=[ai_city_dropdown, question_input, max_sources_slider, gr.Textbox(visible=False), temperature_slider],
-                outputs=answer_output
-            )
+    with gr.TabItem("AI Answers (Groq API)"):
+        gr.Markdown(
+            "Ask questions about the dataset and get AI-generated answers using the Groq API "
+            "with the deepseek-r1-distill-llama-70b model."
+        )
+
+        with gr.Row():
+            with gr.Column():
+                ai_city_dropdown = gr.Dropdown(
+                    choices=city_names,
+                    value=city_names[0] if city_names else None,
+                    label="Select City"
+                )
+                question_input = gr.Textbox(
+                    label="Ask a Question",
+                    placeholder="E.g., What was the historical significance of this ancient city?",
+                    lines=3
+                )
+                max_sources_slider = gr.Slider(
+                    minimum=1,
+                    maximum=10,
+                    value=3,
+                    step=1,
+                    label="Maximum Number of Sources to Consider",
+                    info="Higher values may provide more comprehensive answers but will take longer"
+                )
+                temperature_slider = gr.Slider(
+                    minimum=0.0,
+                    maximum=1.0,
+                    value=0.3,
+                    step=0.1,
+                    label="Temperature",
+                    info="Lower values create more focused answers, higher values create more creative ones"
+                )
+                generate_button = gr.Button("Generate Answer")
+
+            with gr.Column():
+                answer_output = gr.HTML(
+                    label="AI-Generated Answer",
+                    value="",
+                    elem_classes=["results-output"]
+                )
+
+        # Function to handle the Generate Answer button click
+        def on_generate_answer(city, question, max_sources, temperature):
+            if not question or not question.strip():
+                return "Please enter a question to generate an answer."
+            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
+                )
+            except Exception as e:
+                return f"Error: {str(e)}"
+
+        generate_button.click(
+            fn=on_generate_answer,
+            inputs=[ai_city_dropdown,
+                    question_input,
+                    max_sources_slider,
+                    temperature_slider],
+            outputs=answer_output
+        )
 
     # Add CSS styling
     gr.HTML("""