Update art_explorer.py

art_explorer.py

@@ -1,66 +1,52 @@
-import os
-import openai
-import instructor
-from typing import Optional, Dict, Any, List
-from models import ArtHistoryResponse
-from prompts import SYSTEM_PROMPT, CONTEXTUAL_ZOOM_PROMPT
+from typing import Optional, List, Dict
+import json
+from groq import Groq
+from .prompts import SYSTEM_PROMPT, format_exploration_prompt, DEFAULT_RESPONSE
 
-class ArtExplorer:
-  def __init__(self):
-      self.client = instructor.patch(openai.OpenAI(
-          base_url="https://api.groq.com/openai/v1",
-          api_key=os.environ.get("GROQ_API_KEY")
-      ))
-      self.current_context = {}
-      self.exploration_history = []
+class ExplorationPathGenerator:
+  def __init__(self, api_key: str):
+      self.client = Groq(api_key=api_key)
+
+  def generate_exploration_path(
+      self, 
+      user_query: str, 
+      selected_path: Optional[List[Dict[str, str]]] = None,
+      exploration_parameters: Optional[Dict] = None
+  ) -> Dict:
+      if selected_path is None:
+          selected_path = []
+      if exploration_parameters is None:
+          exploration_parameters = {
+              "depth": 5,
+              "domain": None,
+              "previous_explorations": []
+          }
+
+      formatted_prompt = format_exploration_prompt(
+          user_query=user_query,
+          selected_path=selected_path,
+          exploration_parameters=exploration_parameters
+      )
 
-  def explore_art_period(
-      self,
-      query: str,
-      zoom_level: int = 1,
-      previous_context: Optional[Dict] = None
-  ) -> ArtHistoryResponse:
       try:
           response = self.client.chat.completions.create(
-              model="mixtral-8x7b-32768",
-              response_model=ArtHistoryResponse,
               messages=[
-                  {"role": "system", "content": SYSTEM_PROMPT},
-                  {"role": "user", "content": CONTEXTUAL_ZOOM_PROMPT.format(
-                      query=query,
-                      zoom_level=zoom_level,
-                      previous_context=previous_context or {}
-                  )}
-              ]
+                  {
+                      "role": "system",
+                      "content": SYSTEM_PROMPT
+                  },
+                  {
+                      "role": "user",
+                      "content": formatted_prompt
+                  }
+              ],
+              model="mixtral-8x7b-32768",  # or your preferred GROQ model
+              temperature=0.7,
+              max_tokens=2000
           )
           
-          self.current_context = response.model_dump()
-          self.exploration_history.append({
-              "query": query,
-              "zoom_level": zoom_level,
-              "response": self.current_context
-          })
-          
-          return response
-      
+          result = response.choices[0].message.content
+          return json.loads(result)
       except Exception as e:
-          print(f"Error in exploration: {e}")
-          return None
-
-  def get_location_data(self) -> List[Dict[str, Any]]:
-      if not self.current_context:
-          return []
-      
-      return [loc.dict() for loc in self.current_context.get("locations", [])]
-
-  def get_timeline_data(self) -> Dict[str, Any]:
-      if not self.current_context:
-          return {}
-      
-      time_period = self.current_context.get("time_period", {})
-      return {
-          "period_name": time_period.get("name", ""),
-          "start_year": time_period.get("start_year"),
-          "end_year": time_period.get("end_year"),
-          "events": self.current_context.get("historical_events", [])
-      }
+          print(f"Error generating exploration path: {e}")
+          return DEFAULT_RESPONSE