Update app.py

app.py

@@ -112,6 +112,48 @@ AVAILABLE_MODELS = {
         "max_tokens": 128000,
         "output_tokens": 1200,
         "temperature": 0.7
+    },
+    "gpt-4o": {
+        "description": "Latest GPT-4 Omni model",
+        "max_tokens": 128000,
+        "output_tokens": 1200,
+        "temperature": 0.7
+    },
+    "gpt-4o-mini": {
+        "description": "Efficient version of GPT-4o",
+        "max_tokens": 128000,
+        "output_tokens": 1000,
+        "temperature": 0.7
+    },
+    "o1-mini": {
+        "description": "OpenAI Reasoning Model - Mini",
+        "max_tokens": 180000,
+        "output_tokens": 1000,
+        "temperature": 0.7
+    },
+    "o1": {
+        "description": "OpenAI Reasoning Model - Standard",
+        "max_tokens": 200000,
+        "output_tokens": 1200,
+        "temperature": 0.7
+    },
+    "o1-pro": {
+        "description": "OpenAI Reasoning Model - Professional",
+        "max_tokens": 200000,
+        "output_tokens": 1500,
+        "temperature": 0.7
+    },
+    "o3-mini": {
+        "description": "OpenAI Advanced Reasoning - Mini",
+        "max_tokens": 200000,
+        "output_tokens": 1000,
+        "temperature": 0.7
+    },
+    "o3-mini-high": {
+        "description": "OpenAI Advanced Reasoning - Enhanced",
+        "max_tokens": 200000,
+        "output_tokens": 1200,
+        "temperature": 0.7
     }
 }
 
@@ -122,13 +164,14 @@ def get_ai_response(prompt, history):
         return get_demo_response(prompt)
     
     try:
-        # Format messages for OpenAI
+        # Format messages for API
         messages = []
         
         # Add system message
+        system_message = "You are a helpful assistant that provides clear, concise, and accurate information."
         messages.append({
             "role": "system", 
-            "content": "You are a helpful assistant that provides clear, concise, and accurate information."
+            "content": system_message
         })
         
         # Add conversation history