Update src/streamlit_app.py

src/streamlit_app.py

@@ -101,7 +101,7 @@ if query:
         selected = data[idx]
 
         # Show Image
-        st.image(selected['image'], caption="Most relevant medical image", use_column_width=True)
+        st.image(selected['image'], caption="Most relevant medical image", use_container_width=True)
 
         # Show Text
         st.markdown(f"**Case Description:** {selected[TEXT_COLUMN]}")
@@ -109,13 +109,17 @@ if query:
         # GPT Explanation
         if openai.api_key:
             prompt = f"Explain this case in plain English: {selected[TEXT_COLUMN]}"
-            response = openai.ChatCompletion.create(
-                model="gpt-4",
-                messages=[{"role": "user", "content": prompt}],
-                temperature=0.5,
-                max_tokens=150
+            from openai import OpenAI
+            client = OpenAI(api_key=openai.api_key)
+
+            response = client.chat.completions.create(
+            model="gpt-4o",  # or "gpt-4" if you need the older GPT-4
+            messages=[{"role": "user", "content": prompt}],
+            temperature=0.5,
+            max_tokens=150
             )
-            explanation = response["choices"][0]["message"]["content"]
+            explanation = response.choices[0].message.content
+
             st.markdown(f"### 🤖 Explanation by GPT:\n{explanation}")
         else:
             st.warning("OpenAI API key not found. Please set OPENAI_API_KEY as a secret environment variable.")