Update app.py

app.py

@@ -549,49 +549,52 @@ 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():
-                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"]
-                )
-
+        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"]
+                    )
+            
         # Function to handle the Generate Answer button click
         def on_generate_answer(city, question, max_sources, temperature):
             if not question or not question.strip():