Update utils.py

utils.py

@@ -1,21 +1,17 @@
-# utils.py
-
-from groq import Groq
 import os
+import openai
+import requests
 
-# Initialize Groq client
-client = Groq(api_key=os.environ["sk-QOjwnewUPqnUMgDwrVkUNAqgZS4RcEBanCXO6gha0tIwC6s9"])
+# Set the OpenAI API key for GPT-3 or any LLM models you choose
+openai.api_key = os.getenv("OPENAI_API_KEY")
 
-def generate_tutor_output(subject, difficulty, student_input, model):
-    model_dict = {
-        "OpenAI": "Meta-Llama-3.1-8B-Instruct", 
-        "LLAMA3 70B": "llama3-groq-70b-8192-tool-use-preview",
-        "Mixtral 8x7B": "mixtral-8x7b-32768"
-    }
-    current_model = model_dict.get(model, None)
+# Hugging Face API token for Stable Diffusion
+HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN")
+STABLE_DIFFUSION_MODEL = "CompVis/stable-diffusion-v-1-4-original"
 
-    if not current_model:
-        return "Invalid model selected"
+def generate_tutor_output(subject, difficulty, student_input, model):
+    model_dict = {"OpenAI": "gpt-3.5-turbo", "LLAMA3 70B": "llama3-groq-70b-8192-tool-use-preview", "Mixtral 8x7B": "mixtral-8x7b-32768"}
+    current_model = model_dict.get(model, "gpt-3.5-turbo")
 
     prompt = f"""
     You are an expert tutor in {subject} at the {difficulty} level. 
@@ -29,19 +25,22 @@ def generate_tutor_output(subject, difficulty, student_input, model):
     Format your response as a JSON object with keys: "lesson", "question", "feedback"
     """
 
-    completion = client.chat.completions.create(
-        messages=[
-            {
-                "role": "system",
-                "content": f"You are the world's best AI tutor, renowned for your ability to explain complex concepts in an engaging, clear, and memorable way. You specialize in {subject} at the {difficulty} level.",
-            },
-            {
-                "role": "user",
-                "content": prompt,
-            }
-        ],
+    response = openai.ChatCompletion.create(
         model=current_model,
-        max_tokens=1000,
+        messages=[
+            {"role": "system", "content": f"You are a highly skilled {subject} tutor for {difficulty} level."},
+            {"role": "user", "content": prompt}
+        ]
     )
+    return response.choices[0].message['content']
+
+def generate_visual_answer(prompt):
+    url = f"https://api-inference.huggingface.co/models/{STABLE_DIFFUSION_MODEL}"
+    headers = {"Authorization": f"Bearer {HUGGINGFACE_TOKEN}"}
 
-    return completion.choices[0].message.content
+    response = requests.post(url, headers=headers, json={"inputs": prompt})
+    
+    if response.status_code == 200:
+        return response.json()[0]["url"]
+    else:
+        return "Error generating image: " + response.text