Create art_explorer.ppy

art_explorer.ppy

@@ -0,0 +1,66 @@
+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
+
+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 = []
+
+  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 {}
+                  )}
+              ]
+          )
+          
+          self.current_context = response.model_dump()
+          self.exploration_history.append({
+              "query": query,
+              "zoom_level": zoom_level,
+              "response": self.current_context
+          })
+          
+          return response
+      
+      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", [])
+      }