sube api a hf spaces

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]

README.md

@@ -1,10 +1,12 @@
 ---
-title: Real State Api
-emoji: ⚡
-colorFrom: gray
-colorTo: yellow
+title: Real State Api 
+emoji: 🏢
+colorFrom: yellow
+colorTo: green
 sdk: docker
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# D360 Challenge
+
+

app/agents/real_state_agent.py

@@ -0,0 +1,16 @@
+from langchain.agents.agent_types import AgentType
+from langchain_experimental.agents.agent_toolkits import create_pandas_dataframe_agent
+from langchain_openai import ChatOpenAI
+from app.utils.data_preparation import load_data
+
+
+df = load_data()
+
+
+real_state_agent = create_pandas_dataframe_agent(
+    llm=ChatOpenAI(temperature=0, model="gpt-3.5-turbo-0613"),
+    df=df,
+    # prefix = prefix,
+    verbose=True,
+    agent_type=AgentType.OPENAI_FUNCTIONS,
+)

app/models/prediction_models.py

@@ -0,0 +1,9 @@
+from pydantic import BaseModel
+
+
+class PredictionRequest(BaseModel):
+    question: str
+
+
+class PredictionResponse(BaseModel):
+    response: str

app/routes/home.py

@@ -0,0 +1,8 @@
+from fastapi import APIRouter
+
+router = APIRouter()
+
+
+@router.get("/")
+def read_root():
+    return {"message": "D360 Challenge by Santiago Battezzati"}

app/routes/prediction.py

@@ -0,0 +1,24 @@
+from fastapi import APIRouter
+from app.models.prediction_models import PredictionRequest, PredictionResponse
+from typing import List
+from app.agents.real_state_agent import real_state_agent
+
+
+import os
+import openai
+
+openai.api_key = os.getenv("OPENAI_API_KEY")
+
+
+router = APIRouter()
+
+
+## MODIFICAR ESTO:
+@router.post("/", response_model=List[PredictionResponse])
+def predict(request: PredictionRequest):
+    response = real_state_agent.run(request.question)
+
+    # ver el type esto sigue siendo list, pero ahora dije que queiro una response string:
+    response_data = [{"response": response}]
+
+    return response_data

app/utils/data_preparation.py

@@ -0,0 +1,6 @@
+import pandas as pd
+
+
+def load_data():
+    df = pd.read_csv("app/data/listings.csv")
+    return df

app/utils/validations.py

@@ -0,0 +1,21 @@
+# BORRAR ESTE MODULO:
+from fastapi import HTTPException
+
+
+def check_country_code(request):
+    if request.country not in ["CL", "MX"]:
+        raise HTTPException(
+            status_code=400, detail=f"Invalid country code: {request.country}"
+        )
+    else:
+        return print("correct country code")
+
+
+def check_valid_ids(request, df):
+    invalid_ids = set(request.invoiceId) - set(df.index)
+    if invalid_ids:
+        raise HTTPException(
+            status_code=400, detail=f"Invalid invoiceId(s): {invalid_ids}"
+        )
+    else:
+        return print("invoice ids are valid")

main.py

@@ -0,0 +1,18 @@
+from fastapi import FastAPI
+from app.routes import prediction
+from app.routes import home
+
+app = FastAPI()
+
+
+app.include_router(
+    home.router,
+    tags=["home"],
+)
+
+
+app.include_router(
+    prediction.router,
+    prefix="/predict",
+    tags=["prediction"],
+)

requirements.txt

@@ -0,0 +1,17 @@
+fastapi==0.108.0
+langchain==0.1.1
+langchain-experimental==0.0.49
+langchain-openai==0.0.3
+openai==1.8.0
+pandas==2.1.4
+pydantic==2.5.3
+pydantic_core==2.14.6
+#jupyterlab==4.0.10
+tabulate==0.9.0
+uvicorn==0.25.0
+
+
+### 
+## agregar open ai y langchain y langchain experimental... 
+## tabulate... 
+