.

app.py

@@ -262,6 +262,10 @@ def get_ai_response(query: str, context: str, model: str) -> str:
             if result is None:
                 result = "Failed to get response from llama3-70b-8192"
         # Append the model name to the response for clarity
+        # get the key name model from model mapping
+        for key, value in model_mapping.items():
+            if value == model:
+                model = key
         result += f"\n\n**Model:** {model}"
         return result
     except Exception as e:
@@ -302,7 +306,6 @@ def get_groq_llama3_response(query: str) -> str:
     except requests.exceptions.RequestException as e:
         logger.error(f"Groq API Error: {str(e)}")
         return "An error occurred while contacting Groq's Llama 3 model."
-
 # --- Pipeline ---
 documents = load_and_chunk_data(data_file_name)
 retriever = EnhancedRetriever(documents)
@@ -327,6 +330,7 @@ def chat_interface(question: str, history: List[Tuple[str, str]], display_model:
     model = model_mapping.get(display_model, "gemini-2.0-flash")  # Default to Gemini if not found
     response = generate_response(question, model)
     return "", history + [(question, response)]
+
 with gr.Blocks(title="AskNature BioRAG Expert", theme=gr.themes.Soft()) as demo:
     gr.Markdown("# 🌿 AskNature RAG-based Chatbot")
     with gr.Row():