Update app.py

app.py

@@ -18,16 +18,17 @@ import random
 import json
 from datetime import datetime
 from pathlib import Path
-from openai import OpenAI
+
 import streamlit as st
 import pandas as pd
+import openai
 from huggingface_hub import upload_file, HfApi
 
 # ---------------------------
 # CONFIGURABLE PARAMETERS
 # ---------------------------
 # Classic Beer Game choices: choose 24 or 36 depending on experiment design
-DEFAULT_WEEKS = 36
+DEFAULT_WEEKS = 24
 TRANSPORT_DELAY = 2       # shipments take 2 weeks to arrive
 ORDER_DELAY = 1           # orders incur 1-week processing delay (modeled via pipeline)
 INITIAL_INVENTORY = 12
@@ -56,7 +57,7 @@ def fmt(o):
 # Hugging Face upload helper
 # ---------------------------
 HF_TOKEN = os.getenv("HF_TOKEN")
-HF_REPO_ID = os.getenv("HF_REPO_ID")  #  "Lilli98/beer-game-logs"
+HF_REPO_ID = os.getenv("HF_REPO_ID")  # e.g., "XinyuLi/beer-game-logs"
 hf_api = HfApi()
 
 def upload_log_to_hf(local_path: Path, participant_id: str):
@@ -86,13 +87,7 @@ def upload_log_to_hf(local_path: Path, participant_id: str):
 # ---------------------------
 # OpenAI helper
 # ---------------------------
-
-OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
-
-if not OPENAI_API_KEY:
-    raise ValueError("Missing OPENAI_API_KEY environment variable!")
-
-client = OpenAI(api_key=OPENAI_API_KEY)
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
 def call_llm_for_order(role: str, local_state: dict, info_sharing_visible: bool, demand_history: list, max_tokens=40, temperature=0.7):
     """
@@ -119,7 +114,7 @@ def call_llm_for_order(role: str, local_state: dict, info_sharing_visible: bool,
     )
 
     try:
-        resp = client.chat.completions.create(
+        resp = openai.ChatCompletion.create(
             model=OPENAI_MODEL,
             messages=[
                 {"role": "system", "content": "You are an automated Beer Game agent who decides weekly orders."},
@@ -129,7 +124,7 @@ def call_llm_for_order(role: str, local_state: dict, info_sharing_visible: bool,
             temperature=temperature,
             n=1
         )
-        raw = resp.choices[0].message.content.strip()
+        raw = resp.choices[0].message.get("content", "").strip()
     except Exception as e:
         raw = f"OPENAI_ERROR: {str(e)}"
         # fallback later
@@ -483,12 +478,12 @@ with col_sidebar:
             # quick test prompt
             try:
                 test_prompt = "You are a helpful agent. Reply with '42'."
-                resp = client.chat.completions.create(
+                resp = openai.ChatCompletion.create(
                     model=OPENAI_MODEL,
                     messages=[{"role":"user","content":test_prompt}],
                     max_tokens=10
                 )
-                st.write("LLM raw:", resp.choices[0].message.content)
+                st.write("LLM raw:", resp.choices[0].message.get("content"))
             except Exception as e:
                 st.error(f"LLM test failed: {e}")