AI Assistant

export class ChatGpt extends OBC.Component{
  enabled: boolean = true;
  static uuid = OBC.UUID.create();
  fileData : string | null = null; 

  constructor(components:OBC.Components){
    super(components);
    components.add(ChatGpt.uuid, this);
  }
}

const reader = new FileReader();
let fileCOntent : string = ""
reader.onload = function(event){
  const myData = event.target?.result;
  if(typeof myData ==="string"){
    fileCOntent = myData;
    gpt.fileData = myData;
  }
}
reader.readAsText(file);

modifyDataDile = ()=>{
  if(! this.fileData) return;
  const validENtities = new Set([
    "IFCMATERIAL",
    "IfcRelAssociatesMaterial".toUpperCase(),
    "IfcBuildingStorey".toUpperCase(),
    "IfcRelContainedInSpatialStructure".toUpperCase(),
    "IFCWALL",
    "IFCSLAB",
    "IFCBEAM",
  ]);

  var result = this.fileData.split("\n").filter(line=>{
    const pattern = new RegExp("^#\\d+=(\\w+)");
    const match = line.match(/^#\d+\s*=\s*(\w+)/);
    if(!match) return false;
    const entityName = match[1].trim();
    return validENtities.has(entityName.toUpperCase());
  }).join("\n");
  return result;
}

sendQuerryToGPT = async (data:GptData)=>{
  const {message, fileData} = data;
  const apiKey = import.meta.env.VITE_OPENAI_API_KEY;

  const response = await fetch("https://api.openai.com/v1/chat/completions",{
    method:"POST",
    headers:{
      "Content-Type": "application/json",
      "Authorization": `Bearer ${apiKey}`
    },
    body: JSON.stringify({
      model:"gpt-3.5-turbo",
      messages:[
        { 
          role:"system",  
          content:`Based on the given data answear question about ifc file.
                  You should only create the response based on the information given.
                  Information that is not found on given file should not be presented.
                  Please answear the questions using as few words as possible
        ` 
        },
        { 
          role: "user", 
          content: `Here is the file content:\n${this.fileData}\n\nNow answer this question: ${message} using only given content` 
        }
      ]
    })
  });
  const responseData = await response.json();
  return responseData;
}

return BUI.html`
  <bim-toolbar-section label="AI Panel" icon="material-symbols:inventory" style="width:100%">
    <div style="display: flex; flex-direction:column; gap: 0.375rem; min-width:350px">
      <bim-panel-section  fixed>
        <bim-text-input name="name" id="querryDataField" label="Write Question" vertical></bim-text-input>
        ${textArea}
        <div style="display: flex; gap:0.375rem">
          <bim-button name="querry" label="Querry" icon="bx:show-alt" @click=${(e) => onAddClick(e)}></bim-button>
          <bim-button label="Reset" icon="bx:show-alt" @click=${() => resetTextInputs()}></bim-button>
        </div>
      </bim-panel-section>
    </div>
  </bim-toolbar-section>
`

const onAddClick = async (e:Event)=>{
  const btn = e.target as BUI.Button;
  const panelSection = btn.closest("bim-panel-section");
  const querry = panelSection?.value.name;
  
  if(!modelData.fileData){
    alert("No file data, cant answear");
    return;
  }
  
  const querryResponse = await modelData.getQuerry({message:querry, fileData:modelData.fileData});
  textArea.innerText = querryResponse.choices[0].message.content;
}

AddIfcFileDataToGenerallString = async (model: FragmentsGroup, type: number, backupType?: number)=>{
  const indexer = this.components.get(OBC.IfcRelationsIndexer);
  let generallString: string = "";
  let wallsData = await model.getAllPropertiesOfType(type);
  if(!wallsData && backupType) wallsData = await model.getAllPropertiesOfType(backupType);
  if(wallsData==null || Object.keys(wallsData).length==0) return;
  
  for(const [expressId, propeties] of Object.entries(wallsData)){
    generallString += JSON.stringify(propeties);
    
    // Get building storey information
    const levelOfThisWall = indexer.getEntitiesWithRelation(model, "ContainsElements", parseInt(expressId));
    for(var propOfWall of levelOfThisWall){
      const property = await model.getProperties(propOfWall);
      if(property.type === WEBIFC.IFCBUILDINGSTOREY){
        generallString += JSON.stringify(property);
      }
    }

    // Get property sets and quantities
    const propertiesRelations = indexer.getEntityRelations(model,parseInt(expressId),"IsDefinedBy");
    for(const relID of propertiesRelations){
      const relatedEntity = await model.getProperties(relID);
      if(!relatedEntity) continue;
      if(relatedEntity.type === WEBIFC.IFCPROPERTYSET || relatedEntity.type === WEBIFC.IFCELEMENTQUANTITY){
        generallString += JSON.stringify(relatedEntity);
      }
    }
  }
  return generallString;
}

console.log("number of words: ", this.fileData?.length);
console.log("number of words after short: ", this.modifyDataDile()?.length);

VITE_OPENAI_API_KEY=your_openai_api_key_here