Update app.py

app.py

@@ -21,7 +21,10 @@ CURRENT_TIME = "2025-05-23 12:57:22"
 CURRENT_USER = "ErRickow"
 
 # Default API Keys (fallback if user doesn't provide their own)
-DEFAULT_XAI_KEY = os.getenv("XAI_API_KEY", "xai-vfjhklL384Z4HKdItsZomqpFlXubTZJAFnISQUpV7dE8lRnWwYBVPSCxSTlu08wDbAcv720bx2dDiQ9x")
+DEFAULT_XAI_KEY = os.getenv(
+    "XAI_API_KEY",
+    "xai-vfjhklL384Z4HKdItsZomqpFlXubTZJAFnISQUpV7dE8lRnWwYBVPSCxSTlu08wDbAcv720bx2dDiQ9x",
+)
 DEFAULT_GEMINI_KEY = os.getenv("GEMINI_API_KEY")
 
 # API settings
@@ -37,7 +40,7 @@ OLLAMA_MODELS = [
     "starling-lm",
     "dolphin-phi",
     "phi",
-    "orca-mini"
+    "orca-mini",
 ]
 
 XAI_MODELS = [
@@ -110,37 +113,41 @@ Note:
 - Masukkan API key Anda sendiri jika default mencapai limit
 """
 
+
 class AIProvider:
     OLLAMA = "ollama"
     GEMINI = "gemini"
     XAI = "xai"
 
+
 class RepoAnalyzer:
     def __init__(self):
         self.current_repo = None
         self.repo_content = {}
         self.chat_history = []
 
-    async def stream_xai_response(self, prompt: str, api_key: str = None, model: str = "grok-2-latest") -> AsyncGenerator[str, None]:
+    async def stream_xai_response(
+        self, prompt: str, api_key: str = None, model: str = "grok-2-latest"
+    ) -> AsyncGenerator[str, None]:
         """Stream response dari X.AI (Grok) API"""
         try:
             # Use default key if none provided
             actual_key = api_key if api_key else DEFAULT_XAI_KEY
-            
+
             if not actual_key:
                 yield "⚠️ API Key X.AI diperlukan. Gunakan key Anda sendiri atau tunggu reset limit default key."
                 return
 
-            client = AsyncOpenAI(
-                api_key=actual_key,
-                base_url=XAI_BASE_URL
-            )
+            client = AsyncOpenAI(api_key=actual_key, base_url=XAI_BASE_URL)
 
             # Prepare messages with repository context if available
             messages = [
-                {"role": "system", "content": "Anda adalah asisten AI yang membantu menganalisis repository code. Berikan respons dalam Bahasa Indonesia."}
+                {
+                    "role": "system",
+                    "content": "Anda adalah asisten AI yang membantu menganalisis repository code. Berikan respons dalam Bahasa Indonesia.",
+                }
             ]
-            
+
             if self.current_repo:
                 context = f"Repository: {self.current_repo}\n\n"
                 repo_files = "\n".join(list(self.repo_content.keys()))
@@ -150,9 +157,7 @@ class RepoAnalyzer:
             messages.append({"role": "user", "content": prompt})
 
             stream = await client.chat.completions.create(
-                model=model,
-                messages=messages,
-                stream=True
+                model=model, messages=messages, stream=True
             )
 
             full_response = ""
@@ -170,7 +175,9 @@ class RepoAnalyzer:
             print(error_msg)
             yield error_msg
 
-    async def stream_gemini_response(self, prompt: str, api_key: str) -> AsyncGenerator[str, None]:
+    async def stream_gemini_response(
+        self, prompt: str, api_key: str
+    ) -> AsyncGenerator[str, None]:
         """Stream response dari Gemini API"""
         try:
             if not api_key:
@@ -178,110 +185,126 @@ class RepoAnalyzer:
                 return
 
             genai.configure(api_key=api_key)
-            model = genai.GenerativeModel('gemini-pro')
-            
+            model = genai.GenerativeModel("gemini-pro")
+
             # Tambahkan konteks repository jika ada
             if self.current_repo:
                 context = f"Repository: {self.current_repo}\n\n"
                 repo_files = "\n".join(list(self.repo_content.keys()))
                 context += f"Files in repository:\n{repo_files}\n\n"
                 prompt = context + prompt
-            
+
             response = model.generate_content(
                 prompt,
-                generation_config={
-                    "temperature": 0.7,
-                    "top_p": 0.8,
-                    "top_k": 40
-                },
-                stream=True
+                generation_config={"temperature": 0.7, "top_p": 0.8, "top_k": 40},
+                stream=True,
             )
-            
+
             full_response = ""
             async for chunk in response:
                 if chunk.text:
                     full_response += chunk.text
                     yield chunk.text
-            
+
             self.chat_history.append({"role": "user", "content": prompt})
             self.chat_history.append({"role": "assistant", "content": full_response})
-            
+
         except Exception as e:
             error_msg = f"⚠️ Error dalam Gemini API: {str(e)}\n\nPastikan API Key valid dan memiliki kuota yang cukup."
             print(error_msg)
             yield error_msg
 
-    def clone_repository(self, repo_url: str, github_token: str, branch: str = None) -> tuple[bool, str]:
+    def clone_repository(
+        self, repo_url: str, github_token: str, branch: str = None
+    ) -> tuple[bool, str]:
         """Clone repository GitHub dengan autentikasi"""
         if not repo_url:
             return False, "⚠️ URL repository diperlukan"
 
-        repo_name = repo_url.split('/')[-1].replace('.git', '')
-        
+        repo_name = repo_url.split("/")[-1].replace(".git", "")
+
         if os.path.exists(repo_name):
-            subprocess.run(['rm', '-rf', repo_name], check=True)
+            subprocess.run(["rm", "-rf", repo_name], check=True)
 
         try:
-            owner_repo = '/'.join(repo_url.split('/')[-2:])
-            
+            owner_repo = "/".join(repo_url.split("/")[-2:])
+
             # Cek apakah repository private
-            headers = {'Authorization': f'token {github_token}'} if github_token else {}
-            repo_check = requests.get(f"https://api.github.com/repos/{owner_repo}", headers=headers)
-            
+            headers = {"Authorization": f"token {github_token}"} if github_token else {}
+            repo_check = requests.get(
+                f"https://api.github.com/repos/{owner_repo}", headers=headers
+            )
+
             if repo_check.status_code == 404:
                 return False, "⚠️ Repository tidak ditemukan. Periksa URL repository."
             elif repo_check.status_code == 401:
-                return False, "⚠️ Token GitHub tidak valid. Klik icon bantuan (?) untuk panduan mendapatkan token."
-            elif repo_check.status_code == 403 and repo_check.json().get('private', False):
-                return False, "⚠️ Ini adalah repository private. Token GitHub dengan akses 'repo' diperlukan."
+                return (
+                    False,
+                    "⚠️ Token GitHub tidak valid. Klik icon bantuan (?) untuk panduan mendapatkan token.",
+                )
+            elif repo_check.status_code == 403 and repo_check.json().get(
+                "private", False
+            ):
+                return (
+                    False,
+                    "⚠️ Ini adalah repository private. Token GitHub dengan akses 'repo' diperlukan.",
+                )
 
-            auth_url = f"https://{github_token}@github.com/{owner_repo}" if github_token else f"https://github.com/{owner_repo}"
+            auth_url = (
+                f"https://{github_token}@github.com/{owner_repo}"
+                if github_token
+                else f"https://github.com/{owner_repo}"
+            )
 
-            cmd = ['git', 'clone']
+            cmd = ["git", "clone"]
             if branch:
-                cmd.extend(['--branch', branch])
+                cmd.extend(["--branch", branch])
             cmd.append(auth_url)
 
             process = subprocess.run(
                 cmd,
                 capture_output=True,
                 text=True,
-                env=dict(os.environ, GIT_ASKPASS='echo', GIT_TERMINAL_PROMPT='0')
+                env=dict(os.environ, GIT_ASKPASS="echo", GIT_TERMINAL_PROMPT="0"),
             )
 
             if process.returncode == 0:
                 self.current_repo = repo_name
                 # Scan dan simpan konten repository
                 file_count = 0
-                for file_path in Path(repo_name).rglob('*'):
-                    if file_path.is_file() and '.git' not in str(file_path):
+                for file_path in Path(repo_name).rglob("*"):
+                    if file_path.is_file() and ".git" not in str(file_path):
                         success, content = self.read_file_safely(str(file_path))
                         if success:
                             self.repo_content[str(file_path)] = content
                             file_count += 1
-                
-                return True, f"✅ Repository berhasil di-clone!\n\nNama: {repo_name}\nJumlah file: {file_count}\n\nAnda sekarang bisa mengajukan pertanyaan tentang repository ini."
+
+                return (
+                    True,
+                    f"✅ Repository berhasil di-clone!\n\nNama: {repo_name}\nJumlah file: {file_count}\n\nAnda sekarang bisa mengajukan pertanyaan tentang repository ini.",
+                )
             else:
                 return False, f"⚠️ Gagal clone repository:\n{process.stderr}"
-                
+
         except Exception as e:
             return False, f"⚠️ Error: {str(e)}"
 
     def read_file_safely(self, file_path: str) -> tuple[bool, str]:
         """Baca file dengan aman menggunakan berbagai encoding"""
-        encodings = ['utf-8', 'latin-1', 'cp1252']
+        encodings = ["utf-8", "latin-1", "cp1252"]
         for encoding in encodings:
             try:
-                with open(file_path, 'r', encoding=encoding) as f:
+                with open(file_path, "r", encoding=encoding) as f:
                     content = f.read()
                     return True, content
             except Exception as e:
                 continue
         return False, "Tidak dapat membaca file dengan encoding yang didukung"
 
+
 def create_ui():
     analyzer = RepoAnalyzer()
-    
+
     with gr.Blocks(title="Repository Chat Analysis", theme=gr.themes.Soft()) as app:
         gr.Markdown("""
         <style>
@@ -310,9 +333,9 @@ def create_ui():
                 provider = gr.Radio(
                     choices=[AIProvider.XAI, AIProvider.GEMINI, AIProvider.OLLAMA],
                     label="Penyedia AI",
-                    value=AIProvider.XAI
+                    value=AIProvider.XAI,
                 )
-                
+
                 with gr.Group() as api_settings:
                     with gr.Row():
                         xai_key = gr.Textbox(
@@ -320,18 +343,18 @@ def create_ui():
                             type="password",
                             placeholder="Opsional - Klik icon (?) untuk info",
                             show_label=True,
-                            scale=3
+                            scale=3,
                         )
                         with gr.Column(scale=1):
                             gr.Markdown(XAI_API_HELP)
-                    
+
                     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
+                            scale=3,
                         )
                         with gr.Column(scale=1):
                             gr.Markdown(GEMINI_API_HELP)
@@ -342,7 +365,7 @@ def create_ui():
                         label="Model AI",
                         choices=XAI_MODELS,
                         value="grok-2-latest",
-                        interactive=True
+                        interactive=True,
                     )
 
             # Repository Analysis Tab
@@ -351,47 +374,40 @@ def create_ui():
                     with gr.Row():
                         repo_url = gr.Textbox(
                             label="URL Repository GitHub",
-                            placeholder="https://github.com/username/repository"
+                            placeholder="https://github.com/username/repository",
                         )
-                    
+
                     with gr.Row():
                         with gr.Column(scale=2):
                             github_token = gr.Textbox(
-                                label="Token GitHub", 
+                                label="Token GitHub",
                                 type="password",
-                                placeholder="Klik icon (?) untuk panduan"
+                                placeholder="Klik icon (?) untuk panduan",
                             )
                         with gr.Column(scale=1):
                             branch = gr.Textbox(
-                                label="Branch (opsional)",
-                                placeholder="main"
+                                label="Branch (opsional)", placeholder="main"
                             )
 
-                clone_button = gr.Button(
-                    "🔄 Clone Repository",
-                    variant="primary"
-                )
-                
-                clone_status = gr.Markdown(
-                    value="",
-                    label="Status Repository"
-                )
+                clone_button = gr.Button("🔄 Clone Repository", variant="primary")
+
+                clone_status = gr.Markdown(value="", label="Status Repository")
 
                 # File Selection
                 with gr.Group():
                     gr.Markdown("### 📎 File yang Dipilih")
-                    
+
                     with gr.Row():
                         file_selector = gr.Dropdown(
                             label="Pilih File dari Repository",
                             choices=[],
                             multiselect=True,
-                            value=[]  # Initialize with empty list
+                            value=[],  # Initialize with empty list
                         )
-                    
+
                     file_list = gr.HTML(
                         value="<div class='file-list'>Belum ada file yang dipilih</div>",
-                        label="Daftar File Terpilih"
+                        label="Daftar File Terpilih",
                     )
 
                 # Chat Interface
@@ -400,14 +416,14 @@ def create_ui():
                         label="📝 Riwayat Chat",
                         height=500,
                         show_label=True,
-                        type="messages"  # Fix deprecation warning
+                        type="messages",  # Fix deprecation warning
                     )
-                    
+
                     with gr.Row():
                         chat_input = gr.Textbox(
                             label="💭 Tanyakan tentang Repository",
                             placeholder="Ketik pertanyaan Anda di sini...",
-                            lines=3
+                            lines=3,
                         )
                         send_button = gr.Button("📤 Kirim", variant="primary")
 
@@ -419,20 +435,22 @@ def create_ui():
         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 = sorted(list(analyzer.repo_content.keys()))  # Sort files alphabetically
+                files = sorted(
+                    list(analyzer.repo_content.keys())
+                )  # Sort files alphabetically
                 return message, files, []
-            
+
             return message, [], []
 
         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>"
@@ -449,239 +467,241 @@ def create_ui():
 
         # Connect Events
         provider.change(
-            fn=update_model_list,
-            inputs=[provider],
-            outputs=[model_dropdown]
+            fn=update_model_list, inputs=[provider], outputs=[model_dropdown]
         )
 
         clone_button.click(
             fn=handle_clone,
             inputs=[repo_url, github_token, branch],
-            outputs=[clone_status, file_selector, file_list]
+            outputs=[clone_status, file_selector, file_list],
         )
 
         file_selector.change(
-            fn=update_file_list,
-            inputs=[file_selector],
-            outputs=[file_list]
+            fn=update_file_list, inputs=[file_selector], outputs=[file_list]
         )
 
-            with gr.Tab("📊 Analisis Repository", elem_classes="mobile-full"):
-                with gr.Group():
-                    with gr.Row():
-                        repo_url = gr.Textbox(
-                            label="URL Repository GitHub",
-                            placeholder="https://github.com/username/repository",
-                            elem_classes="mobile-full"
-                        )
-                    
-                    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"
-                            )
-                        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",
-                    elem_classes="mobile-full"
-                )
-                
-                clone_status = gr.Markdown(
-                    value="",
-                    label="Status Repository",
-                    elem_classes="mobile-full"
-                )
+        with gr.Tab("📊 Analisis Repository", elem_classes="mobile-full"):
+            with gr.Group():
+                with gr.Row():
+                    repo_url = gr.Textbox(
+                        label="URL Repository GitHub",
+                        placeholder="https://github.com/username/repository",
+                        elem_classes="mobile-full",
+                    )
 
-                # Add 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,
-                            elem_classes="mobile-full"
+                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"
+                    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", elem_classes="mobile-full"
+            )
+
+            clone_status = gr.Markdown(
+                value="", label="Status Repository", elem_classes="mobile-full"
+            )
+
+            # Add 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,
+                        elem_classes="mobile-full",
                     )
 
-                def update_file_list(selected: List[str]) -> str:
-                    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
-
-                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]
+                file_list = gr.HTML(
+                    value="<div class='file-list'>Belum ada file yang dipilih</div>",
+                    label="Daftar File Terpilih",
                 )
 
-                # Update file list when selection changes
-                file_selector.change(
-                    fn=update_file_list,
-                    inputs=[file_selector],
-                    outputs=[file_list]
+            def update_file_list(selected: List[str]) -> str:
+                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
+
+            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
                 )
 
-                gr.Markdown("""
+                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],
+            )
+
+            # Update file list when selection changes
+            file_selector.change(
+                fn=update_file_list, inputs=[file_selector], outputs=[file_list]
+            )
+
+            gr.Markdown("""
                 ### 💡 Contoh Pertanyaan:
                 - "Jelaskan struktur utama dari repository ini"
                 - "Analisis kode di file yang dipilih"
                 - "Bagaimana cara memperbaiki [masalah specific] di file-file ini?"
                 - "Bandingkan implementasi di file-file yang dipilih"
                 """)
-                
-                # Improved chat interface
-                with gr.Group():
-                    chat_input = gr.Textbox(
-                        label="💭 Tanyakan tentang Repository",
-                        placeholder="Ketik pertanyaan Anda di sini...",
-                        lines=3,
-                        elem_classes="mobile-full"
-                    )
-                    send_button = gr.Button(
-                        "📤 Kirim",
-                        variant="primary",
-                        elem_classes="mobile-full"
-                    )
-                
-                chat_history = gr.Chatbot(
-                    label="📝 Riwayat Chat",
-                    height=500,
-                    show_label=True,
-                    elem_classes="mobile-full"
-                )
 
-                loading_indicator = gr.HTML(
-                    '<div id="loading" style="display:none">Memproses permintaan...</div>'
+            # Improved chat interface
+            with gr.Group():
+                chat_input = gr.Textbox(
+                    label="💭 Tanyakan tentang Repository",
+                    placeholder="Ketik pertanyaan Anda di sini...",
+                    lines=3,
+                    elem_classes="mobile-full",
                 )
-
-                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, "")
-                                file_context += f"\n{file}:\n```\n{content}\n```\n"
-                        
-                        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):
-                                full_response += chunk
-                                # Add delay between chunks for readability
-                                await asyncio.sleep(0.05)
-                                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):
-                                full_response += chunk
-                                # Add delay between chunks for readability
-                                await asyncio.sleep(0.05)
-                                history[-1][1] = full_response
-                                yield history
-                                
-                        else:  # OLLAMA
-                            response = analyze_with_ollama(model_name, enhanced_message)
-                            # Simulate streaming for OLLAMA with delay
-                            words = response.split()
-                            for i in range(len(words)):
-                                full_response = " ".join(words[:i+1])
-                                await asyncio.sleep(0.05)
-                                history[-1][1] = full_response
-                                yield history
-                            
-                    except Exception as e:
-                        history[-1][1] = f"⚠️ Error: {str(e)}"
-                        yield history
-
-                # Connect chat events with file selection
-                send_event = send_button.click(
-                    fn=handle_chat,
-                    inputs=[
-                        chat_input,
-                        chat_history,
-                        provider,
-                        model_dropdown,
-                        xai_key,
-                        gemini_key,
-                        file_selector
-                    ],
-                    outputs=chat_history,
-                    show_progress=True
-                ).then(
-                    fn=lambda: gr.update(value=""),
-                    outputs=chat_input
+                send_button = gr.Button(
+                    "📤 Kirim", variant="primary", elem_classes="mobile-full"
                 )
 
-                input_event = chat_input.submit(
-                    fn=handle_chat,
-                    inputs=[
-                        chat_input,
-                        chat_history,
-                        provider,
-                        model_dropdown,
-                        xai_key,
-                        gemini_key,
-                        file_selector
-                    ],
-                    outputs=chat_history,
-                    show_progress=True
-                ).then(
-                    fn=lambda: gr.update(value=""),
-                    outputs=chat_input
-                )
+            chat_history = gr.Chatbot(
+                label="📝 Riwayat Chat",
+                height=500,
+                show_label=True,
+                elem_classes="mobile-full",
+            )
+
+            loading_indicator = gr.HTML(
+                '<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, "")
+                            file_context += f"\n{file}:\n```\n{content}\n```\n"
+
+                    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
+                        ):
+                            full_response += chunk
+                            # Add delay between chunks for readability
+                            await asyncio.sleep(0.05)
+                            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
+                        ):
+                            full_response += chunk
+                            # Add delay between chunks for readability
+                            await asyncio.sleep(0.05)
+                            history[-1][1] = full_response
+                            yield history
+
+                    else:  # OLLAMA
+                        response = analyze_with_ollama(model_name, enhanced_message)
+                        # Simulate streaming for OLLAMA with delay
+                        words = response.split()
+                        for i in range(len(words)):
+                            full_response = " ".join(words[: i + 1])
+                            await asyncio.sleep(0.05)
+                            history[-1][1] = full_response
+                            yield history
+
+                except Exception as e:
+                    history[-1][1] = f"⚠️ Error: {str(e)}"
+                    yield history
+
+            # Connect chat events with file selection
+            send_event = send_button.click(
+                fn=handle_chat,
+                inputs=[
+                    chat_input,
+                    chat_history,
+                    provider,
+                    model_dropdown,
+                    xai_key,
+                    gemini_key,
+                    file_selector,
+                ],
+                outputs=chat_history,
+                show_progress=True,
+            ).then(fn=lambda: gr.update(value=""), outputs=chat_input)
+
+            input_event = chat_input.submit(
+                fn=handle_chat,
+                inputs=[
+                    chat_input,
+                    chat_history,
+                    provider,
+                    model_dropdown,
+                    xai_key,
+                    gemini_key,
+                    file_selector,
+                ],
+                outputs=chat_history,
+                show_progress=True,
+            ).then(fn=lambda: gr.update(value=""), outputs=chat_input)
 
     return app
 
+
 if __name__ == "__main__":
     print(f"""
     🚀 Memulai Repository Chat Analysis
     """)
-    
+
     app = create_ui()
-    app.launch(share=True)
+    app.launch(share=True)