Update app.py

app.py

@@ -92,13 +92,15 @@ class ProteinAnalyzer:
         return ''.join(protein)
     
     @staticmethod
-    def analyze_protein_with_llm(protein_sequence: str, cell_type: str) -> str:
+    def analyze_protein_with_llm(protein_sequence: str, cell_type: str, language: str = "en") -> str:
         """Analyze protein structure and function using Friendli LLM API"""
         
         # Get API token from environment
         token = os.getenv("FRIENDLI_TOKEN")
         if not token:
             logger.warning("FRIENDLI_TOKEN not found in environment variables")
+            if language == "ko":
+                return "단백질 분석 불가: API 토큰이 설정되지 않았습니다"
             return "Protein analysis unavailable: API token not configured"
         
         try:
@@ -108,8 +110,23 @@ class ProteinAnalyzer:
                 "Content-Type": "application/json"
             }
             
-            # Create prompt for protein analysis
-            prompt = f"""You are a bioinformatics expert. Analyze the following protein sequence and provide insights about its potential structure and function.
+            # Create prompt for protein analysis based on language
+            if language == "ko":
+                prompt = f"""당신은 생물정보학 전문가입니다. 다음 단백질 서열을 분석하고 잠재적인 구조와 기능에 대한 통찰력을 제공해주세요.
+
+단백질 서열: {protein_sequence}
+세포 유형: {cell_type}
+
+다음 내용을 포함해주세요:
+1. 서열 패턴을 기반으로 예측되는 단백질 패밀리 또는 도메인
+2. 잠재적인 구조적 특징 (알파 나선, 베타 시트, 루프)
+3. 가능한 생물학적 기능
+4. {cell_type} 세포 유형과의 관련성
+5. 주목할 만한 서열 모티프나 특성
+
+과학 애플리케이션에 표시하기에 적합하도록 간결하면서도 유익한 응답을 작성해주세요."""
+            else:
+                prompt = f"""You are a bioinformatics expert. Analyze the following protein sequence and provide insights about its potential structure and function.
 
 Protein sequence: {protein_sequence}
 Cell type context: {cell_type}
@@ -128,7 +145,7 @@ Keep the response concise but informative, suitable for display in a scientific
                 "messages": [
                     {
                         "role": "system",
-                        "content": "You are a knowledgeable bioinformatics assistant specializing in protein structure and function prediction."
+                        "content": "You are a knowledgeable bioinformatics assistant specializing in protein structure and function prediction." if language == "en" else "당신은 단백질 구조와 기능 예측을 전문으로 하는 지식이 풍부한 생물정보학 어시스턴트입니다."
                     },
                     {
                         "role": "user",
@@ -214,10 +231,10 @@ class DNADiffusionApp:
             logger.error(f"Generation failed: {e}")
             raise
     
-    def handle_generation_request(self, cell_type: str, guidance_scale: float):
+    def handle_generation_request(self, cell_type: str, guidance_scale: float, language: str = "en"):
         """Handle sequence generation request from Gradio"""
         try:
-            logger.info(f"Generating sequence for cell type: {cell_type}")
+            logger.info(f"Generating sequence for cell type: {cell_type}, language: {language}")
             
             # Generate DNA sequence
             sequence, metadata = self.generate_sequence(cell_type, guidance_scale)
@@ -233,7 +250,7 @@ class DNADiffusionApp:
             # Analyze protein with LLM
             logger.info("Analyzing protein structure and function...")
             protein_analysis = self.protein_analyzer.analyze_protein_with_llm(
-                protein_sequence, cell_type
+                protein_sequence, cell_type, language
             )
             
             # Add analysis to metadata
@@ -277,6 +294,7 @@ def create_demo():
             
             if (event.data.type === 'generate_request') {
                 console.log('Triggering generation for cell type:', event.data.cellType);
+                console.log('Language:', event.data.language);
                 
                 // Update the hidden cell type input
                 const radioInputs = document.querySelectorAll('#cell-type-input input[type="radio"]');
@@ -288,6 +306,15 @@ def create_demo():
                     }
                 });
                 
+                // Update the language input
+                const langInputs = document.querySelectorAll('#language-input input[type="radio"]');
+                langInputs.forEach(input => {
+                    if (input.value === event.data.language) {
+                        input.checked = true;
+                        input.dispatchEvent(new Event('change'));
+                    }
+                });
+                
                 // Small delay to ensure radio button update is processed
                 setTimeout(() => {
                     document.querySelector('#generate-btn').click();
@@ -340,6 +367,12 @@ def create_demo():
                 label="Cell Type",
                 elem_id="cell-type-input"
             )
+            language_input = gr.Radio(
+                ["en", "ko"],
+                value="en",
+                label="Language",
+                elem_id="language-input"
+            )
             guidance_input = gr.Slider(
                 minimum=1.0,
                 maximum=10.0,
@@ -366,7 +399,7 @@ def create_demo():
         # Wire up the generation
         generate_btn.click(
             fn=app.handle_generation_request,
-            inputs=[cell_type_input, guidance_input],
+            inputs=[cell_type_input, guidance_input, language_input],
             outputs=[sequence_output, metadata_output]
         ).then(
             fn=None,