Update app.py

app.py

@@ -19,6 +19,14 @@ from datasets import load_dataset, get_dataset_config_info, Dataset
 from PIL import Image
 import time
 import uuid
+import asyncio
+
+# Gemini API requires async
+try:
+    asyncio.get_running_loop()
+except RuntimeError:
+    loop = asyncio.new_event_loop()
+    asyncio.set_event_loop(loop)
 
 # Page config with beautiful theme
 st.set_page_config(
@@ -417,15 +425,20 @@ Groq_Token = os.getenv("GROQ_API_KEY")
 hf_token = os.getenv("HF_TOKEN")
 gemini_token = os.getenv("GEMINI_TOKEN")
 
+# Model order is decided by this
 models = {
-    "gpt-oss-20b": "openai/gpt-oss-20b",
     "gpt-oss-120b": "openai/gpt-oss-120b",
-    "llama3.1": "llama-3.1-8b-instant",
+    "gpt-oss-20b": "openai/gpt-oss-20b",
+    "llama4 maverik":"meta-llama/llama-4-maverick-17b-128e-instruct",
     "llama3.3": "llama-3.3-70b-versatile",
     "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"
+    "gemini-2.5-flash": "gemini-2.5-flash",
+    "gemini-2.5-pro": "gemini-2.5-pro",
+    "gemini-2.5-flash-lite": "gemini-2.5-flash-lite",
+    "gemini-2.0-flash": "gemini-2.0-flash",
+    "gemini-2.0-flash-lite": "gemini-2.0-flash-lite",
+    # "llama4 scout":"meta-llama/llama-4-scout-17b-16e-instruct"
+    # "llama3.1": "llama-3.1-8b-instant"
 }
 
 self_path = os.path.dirname(os.path.abspath(__file__))
@@ -550,10 +563,10 @@ if not available_models:
     st.error("No API keys available! Please set up your API keys in the .env file")
     st.stop()
 
-# Set DeepSeek-R1 as default if available
+# Set GPT-OSS-120B as default if available
 default_index = 0
-if "deepseek-R1" in available_models:
-    default_index = available_models.index("deepseek-R1")
+if "gpt-oss-120b" in available_models:
+    default_index = available_models.index("gpt-oss-120b")
 
 # Simple header - just title and model selector
 col1, col2 = st.columns([3, 1])
@@ -935,6 +948,4 @@ if st.session_state.get("processing"):
         <p style='margin: 0; font-size: 0.75rem; color: #475569;'><strong>Locations:</strong> 300+ cities across India</p>
         <p style='margin: 0; font-size: 0.75rem; color: #475569;'><strong>Records:</strong> 100,000+ measurements</p>
     </div>
-    """, unsafe_allow_html=True)
-
-
+    """, unsafe_allow_html=True)