Update app.py

app.py

@@ -1,10 +1,8 @@
 """
 LocaleNLP Translation Service
 ============================
-
 A multi-language translation application supporting English, Wolof, Hausa, and Darija.
 Features text, audio, and document translation with automatic chaining for all language pairs.
-
 Author: LocaleNLP
 """
 
@@ -12,6 +10,8 @@ import os
 import re
 import logging
 import tempfile
+import csv
+import requests
 from typing import Optional, Dict, Tuple, Any, Union
 from pathlib import Path
 from dataclasses import dataclass
@@ -93,6 +93,9 @@ SUPPORTED_FILE_TYPES = [
 # Audio file extensions
 AUDIO_EXTENSIONS = [".wav", ".mp3", ".m4a"]
 
+# Evaluation CSV URL
+EVALUATION_CSV_URL = "https://github.com/mgolomanta/Models_Evaluation/blob/main/evaluation.csv"
+
 # ================================
 # Logging Configuration
 # ================================
@@ -429,6 +432,72 @@ class AudioProcessor:
         result = model.transcribe(audio_file_path)
         return result["text"]
 
+# ================================
+# Evaluation Service
+# ================================
+
+class EvaluationService:
+    """Handles evaluation submissions and storage."""
+    
+    EVALUATION_CSV_PATH = "evaluation.csv"
+    
+    @staticmethod
+    def save_evaluation(
+        source_lang: str,
+        target_lang: str,
+        user_input: str,
+        model_output: str,
+        notation: Optional[str] = None,
+        correct_answer: Optional[str] = None
+    ) -> str:
+        """
+        Save evaluation to CSV file.
+        
+        Args:
+            source_lang: Source language name
+            target_lang: Target language name
+            user_input: User input text
+            model_output: Model output text
+            notation: Optional notation value
+            correct_answer: Optional correct answer
+            
+        Returns:
+            Status message
+        """
+        try:
+            # Check if CSV file exists, create it with headers if not
+            file_exists = os.path.exists(EvaluationService.EVALUATION_CSV_PATH)
+            
+            with open(EvaluationService.EVALUATION_CSV_PATH, mode='a', newline='', encoding='utf-8') as file:
+                writer = csv.writer(file)
+                
+                # Write headers if file is new
+                if not file_exists:
+                    writer.writerow([
+                        "source_language_name",
+                        "target_language_name", 
+                        "user_input",
+                        "model_output",
+                        "notation_value",
+                        "correct_answer"
+                    ])
+                
+                # Write evaluation data
+                writer.writerow([
+                    source_lang,
+                    target_lang,
+                    user_input,
+                    model_output,
+                    notation if notation else "",
+                    correct_answer if correct_answer else ""
+                ])
+            
+            return "✅ Evaluation submitted successfully!"
+            
+        except Exception as e:
+            logger.error(f"Failed to save evaluation: {e}")
+            return f"❌ Error saving evaluation: {str(e)}"
+
 # ================================
 # Main Application
 # ================================
@@ -441,6 +510,7 @@ class TranslationApp:
         self.content_processor = ContentProcessor()
         self.translation_service = TranslationService(self.model_manager)
         self.audio_processor = AudioProcessor(self.model_manager)
+        self.evaluation_service = EvaluationService()
     
     def process_input(
         self,
@@ -480,6 +550,20 @@ class TranslationApp:
             
         return ""
     
+    def submit_evaluation(
+        self,
+        source_lang: str,
+        target_lang: str,
+        user_input: str,
+        model_output: str,
+        notation: Optional[str],
+        correct_answer: Optional[str]
+    ) -> str:
+        """Submit evaluation data."""
+        return self.evaluation_service.save_evaluation(
+            source_lang, target_lang, user_input, model_output, notation, correct_answer
+        )
+    
     def create_interface(self) -> gr.Blocks:
         """Create and return the Gradio interface."""
         
@@ -551,6 +635,27 @@ class TranslationApp:
                 interactive=False
             )
             
+            # Evaluation section
+            gr.Markdown("### 📝 Model Evaluation")
+            with gr.Group():
+                with gr.Row():
+                    notation = gr.Radio(
+                        choices=["1", "2", "3", "4", "5"],
+                        label="Notation (1-5 stars)",
+                        value=None
+                    )
+                    correct_translation = gr.Textbox(
+                        label="Correct Translation (if incorrect)",
+                        lines=3,
+                        placeholder="Enter the correct translation if the model output is wrong..."
+                    )
+                
+                submit_evaluation_btn = gr.Button("Submit Evaluation", variant="primary")
+                evaluation_status = gr.Textbox(
+                    label="Evaluation Status",
+                    interactive=False
+                )
+            
             # Event handlers
             def update_visibility(mode: str) -> Dict[str, Any]:
                 """Update component visibility based on input mode."""
@@ -601,6 +706,35 @@ class TranslationApp:
                     logger.error(f"Translation error: {e}")
                     return f"❌ Translation error: {str(e)}"
             
+            def handle_evaluation(
+                source_lang: str,
+                target_lang: str,
+                user_input: str,
+                model_output: str,
+                notation_value: Optional[str],
+                correct_answer: Optional[str]
+            ) -> str:
+                """Handle evaluation submission."""
+                if not user_input.strip() or not model_output.strip():
+                    return "⚠️ Please translate text before submitting evaluation."
+                
+                return self.submit_evaluation(
+                    source_lang,
+                    target_lang,
+                    user_input,
+                    model_output,
+                    notation_value,
+                    correct_answer
+                )
+            
+            def clear_evaluation_fields() -> Dict:
+                """Clear evaluation fields after submission."""
+                return {
+                    notation: gr.update(value=None),
+                    correct_translation: gr.update(value=""),
+                    evaluation_status: gr.update(value="Evaluation cleared. Ready for next submission.")
+                }
+            
             # Connect events
             input_mode.change(
                 fn=update_visibility,
@@ -617,6 +751,23 @@ class TranslationApp:
                 inputs=[extracted_text, input_lang, output_lang],
                 outputs=output_text
             )
+            
+            submit_evaluation_btn.click(
+                fn=handle_evaluation,
+                inputs=[
+                    input_lang, 
+                    output_lang, 
+                    extracted_text, 
+                    output_text, 
+                    notation, 
+                    correct_translation
+                ],
+                outputs=evaluation_status
+            ).then(
+                fn=clear_evaluation_fields,
+                inputs=[],
+                outputs=[notation, correct_translation, evaluation_status]
+            )
         
         return interface