Update app.py

app.py

@@ -263,11 +263,14 @@ hf_token = os.getenv("HF_TOKEN")
 gemini_token = os.getenv("GEMINI_TOKEN")
 
 models = {
+    "gpt-oss-20b": "openai/gpt-oss-20b",
+    "gpt-oss-120b": "openai/gpt-oss-120b",
     "llama3.1": "llama-3.1-8b-instant",
-    "mistral": "mistral-saba-24b",
     "llama3.3": "llama-3.3-70b-versatile",
-    "gemma": "gemma2-9b-it",
-    "gemini-pro": "gemini-1.5-pro",
+    "deepseek-R1": "deepseek-r1-distill-llama-70b",
+    "llama4 maverik":"meta-llama/llama-4-maverick-17b-128e-instruct",
+    "llama4 scout":"meta-llama/llama-4-scout-17b-16e-instruct",
+    "gemini-pro": "gemini-1.5-pro"
 }
 
 self_path = os.path.dirname(os.path.abspath(__file__))
@@ -419,10 +422,18 @@ with st.sidebar:
     
     # Filter available models
     available_models = []
+    model_names = list(models.keys())
+    groq_models = []
+    gemini_models = []
+    for model_name in model_names:
+        if "gemini" not in model_name:
+            groq_models.append(model_name)
+        else:
+            gemini_models.append(model_name)
     if Groq_Token and Groq_Token.strip():
-        available_models.extend(["llama3.1", "llama3.3", "mistral", "gemma"])
+        available_models.extend(groq_models)
     if gemini_token and gemini_token.strip():
-        available_models.append("gemini-pro")
+        available_models.extend(gemini_models)
     
     if not available_models:
         st.error("❌ No API keys available! Please set up your API keys in the .env file")
@@ -437,11 +448,17 @@ with st.sidebar:
     # Model descriptions
     model_descriptions = {
         "llama3.1": "🦙 Fast and efficient for general queries",
-        "llama3.3": "🦙 Most advanced Llama model",
+        "llama3.3": "🦙 Most advanced LLaMA model for complex reasoning",
         "mistral": "⚡ Balanced performance and speed",
         "gemma": "💎 Google's lightweight model",
-        "gemini-pro": "🧠 Google's most powerful model"
+        "gemini-pro": "🧠 Google's most powerful model",
+        "gpt-oss-20b": "📘 OpenAI's compact open-weight GPT for everyday tasks",
+        "gpt-oss-120b": "📚 OpenAI's massive open-weight GPT for nuanced responses",
+        "deepseek-R1": "🔍 DeepSeek's distilled LLaMA model for efficient reasoning",
+        "llama4 maverik": "🚀 Meta's LLaMA 4 Maverick — high-performance instruction model",
+        "llama4 scout": "🛰️ Meta's LLaMA 4 Scout — optimized for adaptive reasoning"
     }
+
     
     if model_name in model_descriptions:
         st.info(model_descriptions[model_name])