Update app.py

app.py

@@ -97,22 +97,20 @@ def get_chat_response(query: str, knowledge_base: dict) -> str:
         # Generate prompt
         prompt = generate_openai_prompt(query, context)
         
-        # Get OpenAI client
-        client = get_openai_client()
-        
-        # Generate response
-        response = client.ChatCompletion.create(
-            model="gpt-4-mini",
+        # Generate response using the new API
+        response = openai.ChatCompletion.create(
+            model="gpt-4",
             messages=[
-                {"role": "system", "content": prompt},
-                {"role": "user", "content": query}
+                {"role": "system", "content": "You are Manyue's AI assistant. Answer questions about Manyue."},
+                {"role": "user", "content": query},
+                {"role": "assistant", "content": prompt}
             ],
             max_tokens=200,
             temperature=0.7
         )
         
         # Extract and clean response
-        response_text = response.choices[0].message.content.strip()
+        response_text = response['choices'][0]['message']['content'].strip()
         
         return response_text
         
@@ -120,6 +118,11 @@ def get_chat_response(query: str, knowledge_base: dict) -> str:
         st.error(f"Error generating response: {str(e)}")
         return "I apologize, but I encountered an error. Please try asking again."
 
+        
+    except Exception as e:
+        st.error(f"Error generating response: {str(e)}")
+        return "I apologize, but I encountered an error. Please try asking again."
+
 def initialize_session_state():
     """Initialize session state variables"""
     if "messages" not in st.session_state: