enhacnemnet

main.py

@@ -1,7 +1,7 @@
 from fastapi import FastAPI, HTTPException, UploadFile, File,Request,Depends,status,BackgroundTasks
 from fastapi.security import OAuth2PasswordBearer
 from pydantic import BaseModel
-from typing import Optional
+from typing import Optional, List
 from uuid import uuid4
 import os
 from dotenv import load_dotenv
@@ -11,6 +11,7 @@ import json
 from prompt import *
 from fastapi.middleware.cors import CORSMiddleware
 import requests
+import pandas as pd
 
 
 
@@ -49,7 +50,57 @@ class UserInput(BaseModel):
     stream: Optional[bool] = False
     messages: Optional[list[dict]] = []
 
+class Artwork(BaseModel):
+    name: str
+    artist: str
+    image_url: str
+    date: str
+    description: str
+
+
+# Global variable to store the data
+artworks_data = []
+
+def load_data():
+    global artworks_data
+
+    # Provide the path to your local spreadsheet
+    spreadsheet_path = "data.xlsx"
+
+    # Read the spreadsheet into a DataFrame
+    df = pd.read_excel(spreadsheet_path, sheet_name='Sheet1')  # Adjust sheet_name as needed
+
+    df = df.fillna(False)
+    # Convert DataFrame to a list of dictionaries
+    df_filtered = df[df['Publication'] == True]
+
+    artworks_data = df_filtered.to_dict(orient='records')
+
+    print("Data loaded successfully")
+
+load_data()
+
 #endpoinds
+@app.get("/artworks/{artist_name}")
+async def get_artworks_by_artist(artist_name: str):
+    artist_name_lower = artist_name.lower()
+    results = []
+
+    for artwork in artworks_data:
+        if artist_name_lower in artwork['Artiste'].lower():
+            result = {
+                'name':artwork['Titre français'],
+                'artist':artwork['Artiste'],
+                'image_url':artwork['Image_URL'],
+                'date':str(artwork['Date']),  # Ensure date is a string
+                'description':artwork['Media']
+            }
+            results.append(result)
+
+    if not results:
+        raise HTTPException(status_code=404, detail="Artist not found")
+
+    return results
 
 @app.post("/generate_sphinx")
 async def generate_sphinx():