Update app.py

app.py

@@ -6,7 +6,6 @@ import os
 import json
 import logging
 import re
-import pickle
 from typing import List, Tuple
 import gradio as gr
 from openai import OpenAI
@@ -315,7 +314,15 @@ def generate_response(question: str, model: str) -> str:
         return "An error occurred processing your request."
 
 # --- Gradio Interface ---
-def chat_interface(question: str, history: List[Tuple[str, str]], model: str):
+# Define the mapping from display names to actual model identifiers
+model_mapping = {
+    "Gemini-2.0-Flash": "gemini-2.0-flash",
+    "Meta-llama-3-70b-instruct(GWDG)": "meta-llama-3-70b-instruct",
+    "llama3-70b-8192(Groq)": "llama3-70b-8192"
+}
+
+def chat_interface(question: str, history: List[Tuple[str, str]], display_model: str):
+    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)]
 
@@ -325,7 +332,7 @@ with gr.Blocks(title="AskNature BioRAG Expert", theme=gr.themes.Soft()) as demo:
         chatbot = gr.Chatbot(label="Dialogue History", height=500)
     with gr.Row():
         question = gr.Textbox(placeholder="Ask about biomimicry (e.g. 'How does Werewool use coral proteins to make fibers?')", label="Inquiry", scale=4)
-        model_selector = gr.Dropdown(choices=["gemini-2.0-flash", "meta-llama-3-70b-instruct(GWDG)", "llama3-70b-8192(Groq)"], label="Generation Model", value="gemini-2.0-flash")
+        model_selector = gr.Dropdown(choices=list(model_mapping.keys()), label="Generation Model", value="Gemini-2.0-Flash")
         clear_btn = gr.Button("Clear History", variant="secondary")
 
     gr.Markdown("""
@@ -338,4 +345,4 @@ with gr.Blocks(title="AskNature BioRAG Expert", theme=gr.themes.Soft()) as demo:
     clear_btn.click(lambda: [], None, chatbot)
 
 if __name__ == "__main__":
-    demo.launch(show_error=True)
+    demo.launch(show_error=True)