Update app.py

app.py

@@ -1,8 +1,6 @@
 # https://chatgpt.com/c/692f03df-88e8-8326-9e22-230b32dd2194
 
 import gradio as gr
-from openai import OpenAI
-import os
 
 # -----------------------------
 # Compliance configuration
@@ -55,13 +53,11 @@ def analyze_compliance(api_key: str, free_text: str, uploaded_file) -> str:
     text_to_check = ""
     if uploaded_file is not None:
         try:
-            # Gradio gives a file object or dict depending on version, handle both
             file_path = getattr(uploaded_file, "name", None)
             if file_path:
                 with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
                     text_to_check = f.read()
             else:
-                # Fallback: if it's bytes-like
                 if hasattr(uploaded_file, "read"):
                     raw_bytes = uploaded_file.read()
                     if isinstance(raw_bytes, bytes):
@@ -109,21 +105,49 @@ def analyze_compliance(api_key: str, free_text: str, uploaded_file) -> str:
         "## Fairness\n"
     )
 
+    # -----------------------------
+    # OpenAI call with robust import handling
+    # -----------------------------
     try:
-        # Call OpenAI
-        client = OpenAI(api_key=api_key)
-
-        response = client.chat.completions.create(
-            model="gpt-4.1-mini",  # You can change to "gpt-4.1" if you prefer
-            messages=[
-                {"role": "system", "content": system_message},
-                {"role": "user", "content": user_message},
-            ],
-            temperature=0.1,
-        )
-
-        result = response.choices[0].message.content
-        return result
+        # Try new SDK style first (openai>=1.x)
+        try:
+            from openai import OpenAI  # type: ignore
+            client = OpenAI(api_key=api_key)
+
+            response = client.chat.completions.create(
+                model="gpt-4.1-mini",  # change to "gpt-4.1" if desired
+                messages=[
+                    {"role": "system", "content": system_message},
+                    {"role": "user", "content": user_message},
+                ],
+                temperature=0.1,
+            )
+            result = response.choices[0].message.content
+            return result
+
+        except ModuleNotFoundError:
+            # Fallback: older SDK style `import openai`
+            try:
+                import openai  # type: ignore
+
+                openai.api_key = api_key
+                response = openai.ChatCompletion.create(
+                    model="gpt-4.1-mini",
+                    messages=[
+                        {"role": "system", "content": system_message},
+                        {"role": "user", "content": user_message},
+                    ],
+                    temperature=0.1,
+                )
+                result = response["choices"][0]["message"]["content"]
+                return result
+            except ModuleNotFoundError:
+                return (
+                    "⚠️ **OpenAI library not installed.**\n\n"
+                    "Please add `openai` to your `requirements.txt`:\n\n"
+                    "```text\nopenai>=1.0.0\n```\n"
+                    "and rebuild/restart the app."
+                )
 
     except Exception as e:
         return f"⚠️ **OpenAI API Error:** {e}"