Update app.py

app.py

@@ -386,7 +386,7 @@ def create_ui():
                                 placeholder="Klik icon (?) untuk panduan",
                                 elem_classes="mobile-full",
                             )
-                            gr.Markdown(GITHUB_TOKEN_HELP)  # Add help text
+                            gr.Markdown(GITHUB_TOKEN_HELP)
                         with gr.Column(scale=1):
                             branch = gr.Textbox(
                                 label="Branch (opsional)",
@@ -402,7 +402,7 @@ def create_ui():
                     value="", label="Status Repository", elem_classes="mobile-full"
                 )
 
-                # File Selection
+                # File Selection with improved handling
                 with gr.Group():
                     gr.Markdown("### 📎 File yang Dipilih")
 
@@ -411,9 +411,10 @@ def create_ui():
                             label="Pilih File dari Repository",
                             choices=[],
                             multiselect=True,
-                            value=[],  # Initialize with empty list
+                            value=[],
+                            allow_custom_value=True,  # Allow custom values
+                            max_choices=None,  # Allow multiple selections
                             elem_classes="mobile-full",
-                            allow_custom_value=False,  # Prevent custom values
                         )
 
                     file_list = gr.HTML(
@@ -421,14 +422,58 @@ def create_ui():
                         label="Daftar File Terpilih",
                     )
 
-                # Example questions
-                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"
-                """)
+                # Event Handlers
+                def handle_clone(repo_url, github_token, branch):
+                    if not repo_url:
+                        return (
+                            "⚠️ URL repository diperlukan!",
+                            gr.Dropdown(choices=[]),
+                            "<div class='file-list'>Belum ada file yang dipilih</div>",
+                        )
+
+                    success, message = analyzer.clone_repository(
+                        repo_url, github_token, branch
+                    )
+
+                    if success:
+                        # Get list of files and sort them
+                        files = sorted(list(analyzer.repo_content.keys()))
+                        # Update dropdown with new choices
+                        return (
+                            message,
+                            gr.Dropdown(choices=files, value=[]),
+                            "<div class='file-list'>Belum ada file yang dipilih</div>",
+                        )
+
+                    return (
+                        message,
+                        gr.Dropdown(choices=[]),
+                        "<div class='file-list'>Belum ada file yang dipilih</div>",
+                    )
+
+                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
+
+                # Connect clone button
+                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]
+                )
 
                 # Chat Interface
                 with gr.Group():
@@ -453,152 +498,105 @@ def create_ui():
                         '<div id="loading" style="display:none">Memproses permintaan...</div>'
                     )
 
-            # Event Handlers
-            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
-                    return (
-                        message,
-                        files,
-                        "<div class='file-list'>Belum ada file yang dipilih</div>",
-                    )
-
-                return (
+                async def handle_chat(
                     message,
-                    [],
-                    "<div class='file-list'>Belum ada file yang dipilih</div>",
-                )
-
-            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
-
-            # Connect Events
-            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]
-            )
-
-            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,
+                    history,
+                    provider_choice,
+                    model_name,
                     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
+                    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:
+                        file_context = ""
+                        if selected_files:
+                            file_context = "\n\nFile yang dipilih:\n"
+                            for file in selected_files:
+                                content = analyzer.repo_content.get(file, "")
+                                if content:  # Only include files that exist
+                                    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__":