Update api.py

api.py

@@ -1,24 +1,47 @@
 import os
 import traceback
+import subprocess
+import importlib
 from flask import Flask, request, jsonify
 from flask_cors import CORS
 from dotenv import load_dotenv
-from openai import OpenAI
 from langdetect import detect
 from deep_translator import GoogleTranslator
-import subprocess
+
+# ---------- Fix OpenAI version issue ----------
+try:
+    # Ensure modern OpenAI SDK is installed and loaded
+    subprocess.run(
+        ["pip", "install", "--no-cache-dir", "--upgrade", "openai==1.12.0"],
+        check=True,
+    )
+    import openai
+    importlib.reload(openai)
+    from openai import OpenAI
+    print("✅ Using OpenAI SDK version 1.12.0")
+except Exception as e:
+    print("⚠️ Could not ensure correct OpenAI version:", e)
+    from openai import OpenAI  # fallback import
 
 # Patch huggingface_hub automatically if Gradio overwrote it
 try:
     import huggingface_hub
     if not hasattr(huggingface_hub, "cached_download"):
         subprocess.run(
-            ["pip", "install", "--no-cache-dir", "huggingface-hub==0.24.5", "transformers==4.30.2", "sentence-transformers==2.2.2"],
-            check=True
+            [
+                "pip",
+                "install",
+                "--no-cache-dir",
+                "huggingface-hub==0.24.5",
+                "transformers==4.30.2",
+                "sentence-transformers==2.2.2",
+            ],
+            check=True,
         )
         print("✅ Downgraded huggingface-hub for sentence-transformers compatibility.")
 except Exception as e:
     print("⚠️ Could not auto-patch huggingface_hub:", e)
+
 from sentence_transformers import SentenceTransformer
 from pinecone import Pinecone
 
@@ -37,7 +60,11 @@ CORS(app, resources={r"/ask": {"origins": "*"}})
 # ---------- OpenAI Client ----------
 client = None
 if OPENAI_API_KEY:
-    client = OpenAI(api_key=OPENAI_API_KEY)
+    try:
+        client = OpenAI(api_key=OPENAI_API_KEY)
+        print("✅ OpenAI client initialized successfully.")
+    except Exception as e:
+        print("❌ Failed to initialize OpenAI client:", e)
 else:
     print("⚠️  OPENAI_API_KEY is missing in .env")
 
@@ -63,11 +90,13 @@ try:
             results = []
             for match in matches:
                 meta = match.get("metadata", {})
-                results.append({
-                    "context": meta.get("context", ""),
-                    "page": meta.get("page"),
-                    "score": match.get("score", 0)
-                })
+                results.append(
+                    {
+                        "context": meta.get("context", ""),
+                        "page": meta.get("page"),
+                        "score": match.get("score", 0),
+                    }
+                )
             return results
 
     retriever = PineconeRetriever(index, embedder)
@@ -94,7 +123,9 @@ def detect_language(question: str) -> str:
 
 def normalize_language(lang: str, text: str) -> str:
     """Fix incorrect language detection like 'wer is' → German."""
-    if lang == "nl" and any(word in text.lower() for word in ["wer", "was", "wie", "javid", "coaching"]):
+    if lang == "nl" and any(
+        word in text.lower() for word in ["wer", "was", "wie", "javid", "coaching"]
+    ):
         return "de"
     return lang
 
@@ -120,18 +151,24 @@ def format_answers(question: str, answer: str, results):
     pages = [f"Seite {r.get('page', '')}" for r in results if r.get("page")]
     source = ", ".join(pages) if pages else "No source"
     top_score = max([r.get("score", 0.0) for r in results], default=0.0)
-    return {"answers": [{"question": question, "answer": answer, "source": source, "bm25_score": top_score}]}
+    return {
+        "answers": [
+            {"question": question, "answer": answer, "source": source, "bm25_score": top_score}
+        ]
+    }
 
 # ---------- Routes ----------
 @app.route("/", methods=["GET"])
 def health():
-    return jsonify({
-        "status": "running",
-        "retriever_ready": bool(retriever),
-        "openai_key_loaded": bool(OPENAI_API_KEY),
-        "pinecone_key_loaded": bool(PINECONE_API_KEY),
-        "index_name": PINECONE_INDEX_NAME
-    })
+    return jsonify(
+        {
+            "status": "running",
+            "retriever_ready": bool(retriever),
+            "openai_key_loaded": bool(OPENAI_API_KEY),
+            "pinecone_key_loaded": bool(PINECONE_API_KEY),
+            "index_name": PINECONE_INDEX_NAME,
+        }
+    )
 
 @app.route("/ask", methods=["POST", "OPTIONS"])
 def ask():
@@ -181,7 +218,7 @@ def ask():
             model="gpt-4o-mini",
             messages=[
                 {"role": "system", "content": sys_prompt},
-                {"role": "user", "content": user_content}
+                {"role": "user", "content": user_content},
             ],
             max_tokens=700,
         )