Update Gradio_UI.py

Gradio_UI.py

@@ -13,14 +13,26 @@ from smolagents.utils import _is_package_available
 
 import gradio as gr
 from fpdf import FPDF
+from langdetect import detect
 
-# === PDF Export Function ===
-def export_text_to_pdf(text, output_path="ai_act_register.pdf"):
+# === PDF Export Function with Language Option ===
+def export_text_to_pdf(text, output_path="ai_act_register.pdf", language="fr"):
     pdf = FPDF()
     pdf.add_page()
     pdf.set_auto_page_break(auto=True, margin=15)
     pdf.set_font("Arial", size=12)
 
+    # Optional: Translate heading depending on language
+    if language == "en":
+        title = "AI Act Compliance Register"
+    else:
+        title = "Registre de Conformité AI Act"
+
+    pdf.set_font("Arial", 'B', 14)
+    pdf.cell(0, 10, title, ln=True)
+    pdf.ln(10)
+    pdf.set_font("Arial", size=12)
+
     for line in text.split('\n'):
         pdf.multi_cell(0, 10, line)
 
@@ -107,6 +119,7 @@ def stream_to_gradio(agent, task: str, reset_agent_memory: bool = False, additio
         raise ModuleNotFoundError("Install 'gradio': pip install 'smolagents[gradio]'")
 
     final_output_text = ""
+
     for step_log in agent.run(task, stream=True, reset=reset_agent_memory, additional_args=additional_args):
         if hasattr(agent.model, "last_input_token_count"):
             if isinstance(step_log, ActionStep):
@@ -118,8 +131,13 @@ def stream_to_gradio(agent, task: str, reset_agent_memory: bool = False, additio
             if message.role == "assistant" and isinstance(message.content, str):
                 final_output_text += message.content + "\n"
 
-    pdf_path = export_text_to_pdf(final_output_text)
-    yield gr.ChatMessage(role="assistant", content=f"✅ Document AI Act exporté en PDF : {pdf_path}")
+    try:
+        language = detect(final_output_text)
+    except:
+        language = "fr"
+
+    pdf_path = export_text_to_pdf(final_output_text, language=language)
+    yield gr.ChatMessage(role="assistant", content=f"✅ Document AI Act exporté en PDF ({language.upper()}) : {pdf_path}")
     yield gr.File(value=pdf_path, label="Télécharger le registre PDF")
 
 # === UI ===
@@ -129,7 +147,7 @@ class GradioUI:
             raise ModuleNotFoundError("Install 'gradio': pip install 'smolagents[gradio]'")
         self.agent = agent
         self.file_upload_folder = file_upload_folder
-        if self.file_upload_folder and not os.path.exists(file_upload_folder):
+        if self.file_upload_folder and not os.path.exists(self.file_upload_folder):
             os.mkdir(self.file_upload_folder)
 
     def interact_with_agent(self, prompt, messages):