Update app.py

app.py

@@ -279,8 +279,8 @@ class RepoAnalyzer:
         return False, "Tidak dapat membaca file dengan encoding yang didukung"
 
 def create_ui():
-    # Get UTC time
-    utc_time = datetime.utcnow().strftime("%Y-%m-%d %H:%M:%S")
+    # Get local time
+    local_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     analyzer = RepoAnalyzer()
     
     with gr.Blocks(title="Repository Chat Analysis", theme=gr.themes.Soft()) as app:
@@ -368,11 +368,7 @@ def create_ui():
         """)
 
         with gr.Row(elem_classes="container"):
-            gr.Markdown(f"""
-            # 🤖 Repository Chat Analysis
-            
-            📅 Current Date and Time (UTC): {utc_time}
-            """)
+            gr.Markdown(f"# 🤖 Repository Chat Analysis\n\n📅 {local_time}")
 
         with gr.Tabs() as tabs:
             with gr.Tab("🛠️ Konfigurasi", elem_classes="mobile-full"):
@@ -496,67 +492,69 @@ def create_ui():
                     '<div id="loading" style="display:none">Memproses permintaan...</div>'
                 )
 
-        async def handle_chat(message, history, provider_choice, model_name, xai_key, gemini_key, selected_files):
-            if not analyzer.current_repo:
-                yield history + [[message, "⚠️ Mohon clone repository terlebih dahulu sebelum mengajukan pertanyaan."]]
-                return
-            
-            history = history or []
-            history.append([message, ""])
-            
-            try:
-                # Add context about selected files to the prompt
-                file_context = ""
-                if selected_files:
-                    file_context = "\n\nFile yang dipilih:\n"
-                    for file in selected_files:
-                        content = analyzer.repo_content.get(file, "")
-                        # Wrap code blocks in custom artifact styling
-                        file_context += f"""
-<div class="wrapper-artifact">
-    <div class="header-artifact">
-        <span>{file}</span>
-        <button class="copy-button" onclick="copyToClipboard(`{content.replace('`', '\\`')}`)">Copy</button>
-    </div>
-    <div class="content-artifact">
-        <pre><code>{content}</code></pre>
-    </div>
-</div>
-"""
+async def handle_chat(message, history, provider_choice, model_name, xai_key, gemini_key, selected_files):
+    if not analyzer.current_repo:
+        yield history + [[message, "⚠️ Mohon clone repository terlebih dahulu sebelum mengajukan pertanyaan."]]
+        return
+    
+    history = history or []
+    history.append([message, ""])
+    
+    try:
+        # Add context about selected files to the prompt
+        file_context = ""
+        if selected_files:
+            file_context = "\n\nFile yang dipilih:\n"
+            for file in selected_files:
+                content = analyzer.repo_content.get(file, "")
+                escaped_content = content.replace('`', r'\`')
                 
-                enhanced_message = f"{message}\n{file_context}"
+                html = (
+                    '<div class="wrapper-artifact">'
+                    f'<div class="header-artifact">'
+                    f'<span>{file}</span>'
+                    f'<button class="copy-button" onclick="copyToClipboard(`{escaped_content}`)">Copy</button>'
+                    '</div>'
+                    '<div class="content-artifact">'
+                    f'<pre><code>{content}</code></pre>'
+                    '</div>'
+                    '</div>'
+                )
+                file_context += html
+        
+        enhanced_message = f"{message}\n{file_context}"
+        
+        full_response = ""
+        if provider_choice == AIProvider.XAI:
+            async for chunk in analyzer.stream_xai_response(enhanced_message, xai_key, model_name):
+                # Wrap code blocks in custom styling
+                chunk = process_code_blocks(chunk)
+                full_response += chunk
+                await asyncio.sleep(0.08)
+                history[-1][1] = full_response
+                yield history
                 
-                full_response = ""
-                if provider_choice == AIProvider.XAI:
-                    async for chunk in analyzer.stream_xai_response(enhanced_message, xai_key, model_name):
-                        # Wrap code blocks in custom styling
-                        chunk = process_code_blocks(chunk)
-                        full_response += chunk
-                        await asyncio.sleep(0.08)
-                        history[-1][1] = full_response
-                        yield history
-                        
-                elif provider_choice == AIProvider.GEMINI:
-                    async for chunk in analyzer.stream_gemini_response(enhanced_message, gemini_key or DEFAULT_GEMINI_KEY):
-                        chunk = process_code_blocks(chunk)
-                        full_response += chunk
-                        await asyncio.sleep(0.08)
-                        history[-1][1] = full_response
-                        yield history
-                        
-                else:  # OLLAMA
-                    response = analyze_with_ollama(model_name, enhanced_message)
-                    response = process_code_blocks(response)
-                    words = response.split()
-                    for i in range(len(words)):
-                        full_response = " ".join(words[:i+1])
-                        await asyncio.sleep(0.08)
-                        history[-1][1] = full_response
-                        yield history
-                    
-            except Exception as e:
-                history[-1][1] = f"⚠️ Error: {str(e)}"
+        elif provider_choice == AIProvider.GEMINI:
+            async for chunk in analyzer.stream_gemini_response(enhanced_message, gemini_key or DEFAULT_GEMINI_KEY):
+                chunk = process_code_blocks(chunk)
+                full_response += chunk
+                await asyncio.sleep(0.08)
+                history[-1][1] = full_response
                 yield history
+                
+        else:  # OLLAMA
+            response = analyze_with_ollama(model_name, enhanced_message)
+            response = process_code_blocks(response)
+            words = response.split()
+            for i in range(len(words)):
+                full_response = " ".join(words[:i+1])
+                await asyncio.sleep(0.08)
+                history[-1][1] = full_response
+                yield history
+                
+    except Exception as e:
+        history[-1][1] = f"⚠️ Error: {str(e)}"
+        yield history
 
 def process_code_blocks(text):
     """Process markdown code blocks to use custom artifact styling"""
@@ -568,17 +566,20 @@ def process_code_blocks(text):
     def replace_code_block(match):
         language = match.group(1)
         code = match.group(2)
-        return f"""
-<div class="wrapper-artifact">
-    <div class="header-artifact">
-        <span>{language}</span>
-        <button class="copy-button" onclick="copyToClipboard(`{code.replace('`', '\\`')}`)">Copy</button>
-    </div>
-    <div class="content-artifact">
-        <pre><code>{code}</code></pre>
-    </div>
-</div>
-"""
+        escaped_code = code.replace('`', r'\`')
+        
+        html = (
+            '<div class="wrapper-artifact">'
+            f'<div class="header-artifact">'
+            f'<span>{language}</span>'
+            f'<button class="copy-button" onclick="copyToClipboard(`{escaped_code}`)">Copy</button>'
+            '</div>'
+            '<div class="content-artifact">'
+            f'<pre><code>{code}</code></pre>'
+            '</div>'
+            '</div>'
+        )
+        return html
     
     # Replace all code blocks in the text
     processed_text = re.sub(pattern, replace_code_block, text, flags=re.DOTALL)