feat: added the reading of the excel file, and the possibility to download a copy of the file.

config.py

@@ -7,6 +7,7 @@ config = {
     'default_template_index': 0,
     'styled_docs_path': 'temp/styles_files',
     'generated_docs_path': 'temp/generated_files',
+    'excel_doc_path' : 'temp/requirements_file',
     'options': ["Recentrer les tableaux", "Justifier le texte (Normal)"],
     'max_styles': 300,
     'log_msg': {

src/control/controller.py

@@ -1,5 +1,6 @@
 import asyncio
 import os
+import shutil
 from typing import Dict
 import random
 import datetime
@@ -13,6 +14,7 @@ import src.tools.semantic_db as semantic_db
 from src.tools.wiki import Wiki
 from src.llm.llm_tools import get_wikilist, get_public_paragraph, get_private_paragraph
 from src.tools.semantic_db import add_texts_to_collection, query_collection
+from src.tools.excel_tools import excel_to_json
 import gradio as gr
 from src.retriever.retriever import Retriever
 
@@ -22,6 +24,7 @@ class Controller:
         self.templates_path = config['templates_path']
         self.generated_docs_path = config['generated_docs_path']
         self.styled_docs_path = config['styled_docs_path']
+        self.excel_doc_path = config['excel_doc_path']
         self.new_docs = []
         self.gen_docs = []
         self.input_csv = ""
@@ -72,7 +75,9 @@ class Controller:
         self.gen_docs = []
         self.log = []
         path_to_clear = os.path.abspath(self.generated_docs_path)
+        second_path_to_clear = os.path.abspath(self.excel_doc_path)
         [os.remove(f"{path_to_clear}/{doc}") for doc in os.listdir(path_to_clear)]
+        [os.remove(f"{second_path_to_clear}/{doc}") for doc in os.listdir(second_path_to_clear)]
 
     def set_template(self, template_name: str = ""):
         if not template_name:
@@ -252,14 +257,14 @@ class Controller:
         gen_paths = list(set(gen_paths))
         return gen_paths
     
-            
- 
-
 
 
     """
     Requirements
     """
+    def clear_input_csv(self):
+        self.input_csv = ""
+        [os.remove(f"{self.excel_doc_path}/{doc}") for doc in os.listdir(self.excel_doc_path)]
 
     def set_input_csv(self, csv_path: str):
         """
@@ -282,4 +287,21 @@ class Controller:
         """
         fills the collection with the blocks of the documents
         """
-        Retriever(doc=doc, collection=collection)
+        Retriever(doc=doc, collection=collection)
+
+    def generate_response_to_requirements(self):
+        excel_content = self.get_requirements_from_csv()
+        print(excel_content)
+        excel_name = self.input_csv
+        if '/' in excel_name:
+            excel_name = excel_name.split('/')[-1]
+        elif '\\' in excel_name:
+            excel_name = excel_name.split('\\')[-1]
+        #copy the document and generate the new one
+        shutil.copy(self.input_csv, self.excel_doc_path)
+
+
+
+    def get_requirements_from_csv(self):
+        excel_content = excel_to_json(self.input_csv)
+        return excel_content

src/tools/excel_tools.py

@@ -0,0 +1,5 @@
+import pandas as pd
+
+def excel_to_json(file_path):
+    df = pd.read_excel(file_path)
+    return df.to_json(orient='records', force_ascii=False)

src/view/view.py

@@ -133,10 +133,10 @@ def run(config: Dict, controller: Controller):
                 """
                 
                 with gr.Accordion("Générer la réponse aux exigences (en cours de développement)", open=False, visible=True) as exigences_acc:
-                    input_csv_comp = gr.File(file_count="single", file_types=[".csv", "xlsx"], visible=True, label="Fichiers d'exigences (csv, xlsx only)")
+                    input_csv_comp = gr.File(file_count='multiple', file_types=[".xlsx"], label="Fichier d'exigences (xlsx only)")
                     with gr.Row():
                         verif_btn = gr.Button("Générer la réponse aux exigences (en cours de développement)", visible=False)
-                    output_csv_comp = gr.File(file_count="single", file_types=[".csv", "xlsx"], visible=False)
+                    output_csv_comp = gr.File(file_count="single", file_types=[".xlsx"], visible=False)
 
                 gr.Markdown("")
                 gr.Markdown("")
@@ -175,8 +175,12 @@ def run(config: Dict, controller: Controller):
         """
 
         def input_csv_fn(input_csv_):
-            if not input_csv_.name.endswith('.csv') and not input_csv_.name.endswith('.xlsx'):
-                raise gr.Error(f'File {input_csv_.name} is not a csv or xlsx file, please upload only csv or xlsx files')
+            if len(input_csv_) > 1:
+                raise gr.Error(f'Please upload only one file')
+            else:
+                input_csv_ = input_csv_[0]
+            if not input_csv_.name.endswith('.xlsx'):
+                raise gr.Error(f'File {input_csv_.name} is not a xlsx file, please upload only xlsx files')
             else:       
                 controller.set_input_csv(input_csv_)
                 update_ = {
@@ -189,6 +193,32 @@ def run(config: Dict, controller: Controller):
                                 inputs=[input_csv_comp],
                                 outputs=[verif_btn],
                                 )
+        
+        def input_csv_clear_fn():
+            controller.clear_input_csv()
+            update_ = {
+                verif_btn: gr.update(visible=False),
+            }
+            return update_
+        
+        input_csv_comp.clear(
+            input_csv_clear_fn,
+            inputs=[],
+            outputs=[verif_btn]
+        )
+        
+        def generate_requirements_excel():
+            controller.generate_response_to_requirements()
+            output_path = [f"{controller.excel_doc_path}/{f}" for f in os.listdir(controller.excel_doc_path)]
+            update_ = {
+                output_csv_comp: gr.update(value=output_path, visible=True),
+            }
+            return update_
+        
+        verif_btn.click(generate_requirements_excel,
+                        inputs=[],
+                        outputs=[output_csv_comp],
+                        )
 
         def input_files_upload_fn(input_files_):
             for files in input_files_: