Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import os
 import time
 import json
-from together import Together
+import requests
 
 # Set environment variables
 # Note: For Hugging Face Spaces, you'll set these in the Secrets tab
@@ -21,11 +21,11 @@ def get_together_client():
     api_key = os.environ.get("TOGETHER_API_KEY")
     if not api_key:
         raise ValueError("TOGETHER_API_KEY not found. Please set it in the environment variables.")
-    return Together(api_key=api_key)
+    return api_key
 
-# Function to call LLM
+# Function to call LLM using REST API directly
 def call_llm(model, prompt, temperature=0.7, max_tokens=1500):
-    client = get_together_client()
+    api_key = get_together_client()
     
     # System message for Salesforce expertise
     system_message = "You are a Salesforce development expert specializing in B2B Commerce migrations, CloudCraze to B2B Lightning Experience conversions, and Apex code optimization. Provide clear, step-by-step guidance and corrected code."
@@ -33,18 +33,33 @@ def call_llm(model, prompt, temperature=0.7, max_tokens=1500):
     start_time = time.time()
     
     try:
-        response = client.chat.completions.create(
-            model=model,
-            messages=[
+        headers = {
+            "Authorization": f"Bearer {api_key}",
+            "Content-Type": "application/json"
+        }
+        
+        data = {
+            "model": model,
+            "messages": [
                 {"role": "system", "content": system_message},
                 {"role": "user", "content": prompt}
             ],
-            temperature=temperature,
-            max_tokens=max_tokens,
-            top_p=0.9
+            "temperature": temperature,
+            "max_tokens": max_tokens,
+            "top_p": 0.9
+        }
+        
+        response = requests.post(
+            "https://api.together.xyz/v1/chat/completions", 
+            headers=headers, 
+            json=data
         )
         
-        completion_text = response.choices[0].message.content
+        if response.status_code != 200:
+            return f"Error: API returned status code {response.status_code}: {response.text}"
+        
+        response_data = response.json()
+        completion_text = response_data["choices"][0]["message"]["content"]
         
         end_time = time.time()
         duration = end_time - start_time