Update app.py

app.py

@@ -342,42 +342,42 @@ class RepoAnalyzer:
             history[-1][1] = f"⚠️ Error: {str(e)}"
             yield history
     
-    def process_code_blocks(text):
-        """Process markdown code blocks to use custom artifact styling"""
-        import re
+def process_code_blocks(text):
+    """Process markdown code blocks to use custom artifact styling"""
+    import re
         
         # Pattern for code blocks with language specification
-        pattern = r"```(\w+)\n(.*?)```"
+    pattern = r"```(\w+)\n(.*?)```"
         
-        def replace_code_block(match):
-            language = match.group(1)
-            code = match.group(2)
-            escaped_code = code.replace('`', r'\`')
+    def replace_code_block(match):
+        language = match.group(1)
+        code = match.group(2)
+        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
+        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)
-        return processed_text
+    processed_text = re.sub(pattern, replace_code_block, text, flags=re.DOTALL)
+    return processed_text
 
-    def create_ui():
-        # 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:
-            gr.Markdown("""
+def create_ui():
+    # 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:
+        gr.Markdown("""
             <style>
             .container { max-width: 100% !important; padding: 1rem; }
             .mobile-full { width: 100% !important; }
@@ -460,176 +460,177 @@ class RepoAnalyzer:
             </script>
             """)
     
-            with gr.Row(elem_classes="container"):
-                gr.Markdown(f"# 🤖 Repository Chat Analysis\n\n📅 {local_time}")
-    
-            with gr.Tabs() as tabs:
-                with gr.Tab("🛠️ Konfigurasi", elem_classes="mobile-full"):
-                    provider = gr.Radio(
-                        choices=[AIProvider.XAI, AIProvider.GEMINI, AIProvider.OLLAMA],
-                        label="Penyedia AI",
-                        value=AIProvider.XAI,
-                        interactive=True,
-                        elem_classes="mobile-full"
-                    )
-                    
-                    with gr.Group() as api_settings:
-                        with gr.Group():
-                            with gr.Row():
-                                xai_key = gr.Textbox(
-                                    label="X.AI (Grok) API Key",
-                                    type="password",
-                                    placeholder="Opsional - Klik icon (?) untuk info",
-                                    show_label=True,
-                                    scale=3
-                                )
-                                with gr.Column(scale=1):
-                                    gr.Markdown(XAI_API_HELP)
-                        
-                        with gr.Group():
-                            with gr.Row():
-                                gemini_key = gr.Textbox(
-                                    label="Gemini API Key",
-                                    type="password",
-                                    placeholder="Opsional - Kosongkan untuk gunakan key default",
-                                    show_label=True,
-                                    scale=3
-                                )
-                                with gr.Column(scale=1):
-                                    gr.Markdown(GEMINI_API_HELP)
-    
-                    # Model selection with better layout
-                    with gr.Row():
-                        model_dropdown = gr.Dropdown(
-                            label="Model AI",
-                            choices=XAI_MODELS,
-                            value="grok-2-latest",
-                            interactive=True,
-                            elem_classes="mobile-full"
-                        )
-    
-                with gr.Tab("📊 Analisis Repository", elem_classes="mobile-full"):
+        # Header
+        with gr.Row(elem_classes="container"):
+            gr.Markdown(f"# 🤖 Repository Chat Analysis\n\n📅 {local_time}")
+
+        # Main Interface
+        with gr.Tabs() as tabs:
+            # Configuration Tab
+            with gr.Tab("🛠️ Konfigurasi", elem_classes="mobile-full"):
+                provider = gr.Radio(
+                    choices=[AIProvider.XAI, AIProvider.GEMINI, AIProvider.OLLAMA],
+                    label="Penyedia AI",
+                    value=AIProvider.XAI,
+                    interactive=True,
+                    elem_classes="mobile-full"
+                )
+                
+                with gr.Group() as api_settings:
+                    # XAI API Key
                     with gr.Group():
                         with gr.Row():
-                            repo_url = gr.Textbox(
-                                label="URL Repository GitHub",
-                                placeholder="https://github.com/username/repository",
-                                elem_classes="mobile-full"
+                            xai_key = gr.Textbox(
+                                label="X.AI (Grok) API Key",
+                                type="password",
+                                placeholder="Opsional - Klik icon (?) untuk info",
+                                show_label=True,
+                                scale=3
                             )
-                        
+                            with gr.Column(scale=1):
+                                gr.Markdown(XAI_API_HELP)
+                    
+                    # Gemini API Key
+                    with gr.Group():
                         with gr.Row():
-                            with gr.Column(scale=2):
-                                github_token = gr.Textbox(
-                                    label="Token GitHub", 
-                                    type="password",
-                                    placeholder="Klik icon (?) untuk panduan",
-                                    elem_classes="mobile-full"
-                                )
+                            gemini_key = gr.Textbox(
+                                label="Gemini API Key",
+                                type="password",
+                                placeholder="Opsional - Kosongkan untuk gunakan key default",
+                                show_label=True,
+                                scale=3
+                            )
                             with gr.Column(scale=1):
-                                branch = gr.Textbox(
-                                    label="Branch (opsional)",
-                                    placeholder="main",
-                                    elem_classes="mobile-full"
-                                )
-    
-                    clone_button = gr.Button(
-                        "🔄 Clone Repository",
-                        variant="primary",
+                                gr.Markdown(GEMINI_API_HELP)
+
+                # Model Selection
+                with gr.Row():
+                    model_dropdown = gr.Dropdown(
+                        label="Model AI",
+                        choices=XAI_MODELS,
+                        value="grok-2-latest",
+                        interactive=True,
                         elem_classes="mobile-full"
                     )
+
+            # Repository Analysis Tab
+            with gr.Tab("📊 Analisis Repository", elem_classes="mobile-full"):
+                # Repository Input Section
+                with gr.Group():
+                    with gr.Row():
+                        repo_url = gr.Textbox(
+                            label="URL Repository GitHub",
+                            placeholder="https://github.com/username/repository",
+                            elem_classes="mobile-full"
+                        )
                     
-                    clone_status = gr.Markdown(
-                        value="",
-                        label="Status Repository",
-                        elem_classes="mobile-full"
-                    )
-    
-                    # File selection components
-                    with gr.Group():
-                        gr.Markdown("### 📎 File yang Dipilih")
-                        
-                        with gr.Row():
-                            file_selector = gr.Dropdown(
-                                label="Pilih File dari Repository",
-                                choices=[],
-                                multiselect=True,
+                    with gr.Row():
+                        with gr.Column(scale=2):
+                            github_token = gr.Textbox(
+                                label="Token GitHub", 
+                                type="password",
+                                placeholder="Klik icon (?) untuk panduan",
                                 elem_classes="mobile-full"
                             )
-                        
-                        file_list = gr.HTML(
-                            value="<div class='file-list'>Belum ada file yang dipilih</div>",
-                            label="Daftar File Terpilih"
-                        )
-    
-                    # Enhanced chat interface with fullscreen support
-            with gr.Group():
-                chat_history = gr.Chatbot(
-                    label="📝 Riwayat Chat",
-                    elem_classes="fullscreen-chat",
-                    height=None,  # Let CSS handle the height
-                    show_label=True,
-                    type="messages"  # Fix deprecation warning
+                        with gr.Column(scale=1):
+                            branch = gr.Textbox(
+                                label="Branch (opsional)",
+                                placeholder="main",
+                                elem_classes="mobile-full"
+                            )
+
+                # Clone Repository Section
+                clone_button = gr.Button(
+                    "🔄 Clone Repository",
+                    variant="primary",
+                    elem_classes="mobile-full"
                 )
                 
-                with gr.Group(elem_classes="chat-input"):
-                    chat_input = gr.Textbox(
-                        label="💭 Tanyakan tentang Repository",
-                        placeholder="Ketik pertanyaan Anda di sini...",
-                        lines=3,
-                        elem_classes="mobile-full"
-                    )
+                clone_status = gr.Markdown(
+                    value="",
+                    label="Status Repository",
+                    elem_classes="mobile-full"
+                )
+
+                # File Selection Section
+                with gr.Group():
+                    gr.Markdown("### 📎 File yang Dipilih")
+                    
                     with gr.Row():
-                        clear_button = gr.Button("🧹 Bersihkan", variant="secondary")
-                        send_button = gr.Button("📤 Kirim", variant="primary")
+                        file_selector = gr.Dropdown(
+                            label="Pilih File dari Repository",
+                            choices=[],
+                            multiselect=True,
+                            elem_classes="mobile-full"
+                        )
+                    
+                    file_list = gr.HTML(
+                        value="<div class='file-list'>Belum ada file yang dipilih</div>",
+                        label="Daftar File Terpilih"
+                    )
 
-        # Connect event handlers
+        # Chat Interface (Outside tabs for full width)
+        with gr.Group():
+            chat_history = gr.Chatbot(
+                label="📝 Riwayat Chat",
+                elem_classes="fullscreen-chat",
+                height=None,
+                show_label=True,
+                type="messages"
+            )
+            
+            with gr.Group(elem_classes="chat-input"):
+                chat_input = gr.Textbox(
+                    label="💭 Tanyakan tentang Repository",
+                    placeholder="Ketik pertanyaan Anda di sini...",
+                    lines=3,
+                    elem_classes="mobile-full"
+                )
+                with gr.Row():
+                    clear_button = gr.Button("🧹 Bersihkan", variant="secondary")
+                    send_button = gr.Button("📤 Kirim", variant="primary")
+
+        # Event Handlers
         def clear_chat_history():
             return None
 
-        clear_button.click(
-            fn=clear_chat_history,
-            outputs=chat_history
-        )
-
         def handle_clone(repo_url, github_token, branch):
             if not repo_url:
                 return "⚠️ URL repository diperlukan!", [], []
-            
             success, message = analyzer.clone_repository(repo_url, github_token, branch)
-            
             if success:
-                # Get list of files from cloned repository
                 files = list(analyzer.repo_content.keys())
                 return message, files, []
-            
             return message, [], []
 
-        # Connect clone button click event
-        clone_button.click(
-            fn=handle_clone,
-            inputs=[repo_url, github_token, branch],
-            outputs=[clone_status, file_selector, file_list]
-        )
-
         def update_file_list(selected):
             if not selected:
                 return "<div class='file-list'>Belum ada file yang dipilih</div>"
-            
             html = "<div class='file-list'>"
             for file in selected:
                 html += f"<div class='file-item'><span>{file}</span></div>"
             html += "</div>"
             return html
 
-        # Update file list when selection changes
+        # Connect Events
+        clear_button.click(
+            fn=clear_chat_history,
+            outputs=chat_history
+        )
+
+        clone_button.click(
+            fn=handle_clone,
+            inputs=[repo_url, github_token, branch],
+            outputs=[clone_status, file_selector, file_list]
+        )
+
         file_selector.change(
             fn=update_file_list,
             inputs=[file_selector],
             outputs=[file_list]
         )
 
-        # Connect chat events
-        send_event = send_button.click(
+        send_button.click(
             fn=handle_chat,
             inputs=[
                 chat_input,
@@ -647,7 +648,7 @@ class RepoAnalyzer:
             outputs=chat_input
         )
 
-        input_event = chat_input.submit(
+        chat_input.submit(
             fn=handle_chat,
             inputs=[
                 chat_input,
@@ -665,12 +666,27 @@ class RepoAnalyzer:
             outputs=chat_input
         )
 
+        # Update model list when provider changes
+        def update_model_list(provider_choice):
+            if provider_choice == AIProvider.XAI:
+                return gr.Dropdown(choices=XAI_MODELS, value="grok-2-latest")
+            elif provider_choice == AIProvider.GEMINI:
+                return gr.Dropdown(choices=GEMINI_MODELS, value="gemini-pro")
+            else:  # OLLAMA
+                return gr.Dropdown(choices=OLLAMA_MODELS, value="llama2")
+
+        provider.change(
+            fn=update_model_list,
+            inputs=[provider],
+            outputs=[model_dropdown]
+        )
+
         return app
 
 if __name__ == "__main__":
     print(f"""
     🚀 Memulai Repository Chat Analysis
-    📅 {datetime.now().strftime("%Y-%m-%d %H:%M:%S")}
+    📅 Current Date and Time (UTC): {datetime.utcnow().strftime("%Y-%m-%d %H:%M:%S")}
     """)
     
     app = create_ui()