Update backend/ai_decoder.py

backend/ai_decoder.py

@@ -1,24 +1,21 @@
 import os
-from huggingface_hub import InferenceClient
+from openai import OpenAI
 
-# Initialize the client (Reads from HF_TOKEN environment variable)
-client = InferenceClient(api_key=os.getenv("HF_TOKEN"))
+# Automatically reads the OPENAI_API_KEY from your Space Secrets
+client = OpenAI() 
 
 def decode_semantic_intent(corrupted_text: str) -> str:
-    prompt = f"""You are a 6G Semantic Communication Decoder. 
-A message was mathematically destroyed by wireless noise.
-The traditional mathematical demodulator outputted this gibberish: '{corrupted_text}'
-
-Based on linguistic probability, reconstruct the original intent of the message perfectly. 
-Only output the corrected sentence. Do not add any conversational text."""
+    prompt = f"A message was destroyed by wireless noise. The demodulator outputted: '{corrupted_text}'. Reconstruct the original intent perfectly. Only output the corrected sentence."
 
     try:
-        response = client.text_generation(
-            prompt, 
-            model="mistralai/Mistral-7B-Instruct-v0.2", 
-            max_new_tokens=50,
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "You are a 6G Semantic Communication Decoder."},
+                {"role": "user", "content": prompt}
+            ],
             temperature=0.1
         )
-        return response.strip()
+        return response.choices[0].message.content.strip()
     except Exception as e:
         return f"AI Decoding Failed: {str(e)}"