Update app.py

app.py

@@ -281,7 +281,6 @@ class RepoAnalyzer:
 
 def create_ui():
     analyzer = RepoAnalyzer()
-    selected_files = []
     
     with gr.Blocks(title="Repository Chat Analysis", theme=gr.themes.Soft()) as app:
         gr.Markdown("""
@@ -302,57 +301,170 @@ def create_ui():
             gr.Markdown(f"""
             # AI Github Repository Chat 
             
-            📅 Waktu: {CURRENT_TIME}  
+            📅 {datetime.now().strftime("%Y-%m-%d %H:%M:%S")} UTC
             """)
 
-        with gr.Tab("🛠️ Konfigurasi"):
-            provider = gr.Radio(
-                choices=[AIProvider.XAI, AIProvider.GEMINI, AIProvider.OLLAMA],
-                label="Penyedia AI",
-                value=AIProvider.XAI
-            )
-            
-            with gr.Group() as api_settings:
-                with gr.Row():
-                    xai_key = gr.Textbox(
-                        label="X.AI (Grok) API Key",
-                        type="password",
-                        placeholder="Opsional - Klik icon (?) untuk info. Kosongkan untuk gunakan key default",
-                        show_label=True
-                    )
-                    gr.Markdown(XAI_API_HELP)
+        with gr.Tabs() as tabs:  # Main tabs container
+            # Configuration Tab
+            with gr.Tab("🛠️ Konfigurasi"):
+                provider = gr.Radio(
+                    choices=[AIProvider.XAI, AIProvider.GEMINI, AIProvider.OLLAMA],
+                    label="Penyedia AI",
+                    value=AIProvider.XAI
+                )
                 
+                with gr.Group() as api_settings:
+                    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.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 gr.Row():
-                    gemini_key = gr.Textbox(
-                        label="Gemini API Key",
-                        type="password",
-                        placeholder="Opsional - Kosongkan untuk gunakan key default",
-                        show_label=True
+                    model_dropdown = gr.Dropdown(
+                        label="Model AI",
+                        choices=XAI_MODELS,
+                        value="grok-2-latest",
+                        interactive=True
                     )
-                    gr.Markdown(GEMINI_API_HELP)
-
-            # Model selection based on provider
-            with gr.Row():
-                model_dropdown = gr.Dropdown(
-                    label="Model AI",
-                    choices=XAI_MODELS,
-                    value="grok-2-latest",
-                    interactive=True
+
+            # Repository Analysis Tab
+            with gr.Tab("📊 Analisis Repository"):
+                with gr.Group():
+                    with gr.Row():
+                        repo_url = gr.Textbox(
+                            label="URL Repository GitHub",
+                            placeholder="https://github.com/username/repository"
+                        )
+                    
+                    with gr.Row():
+                        with gr.Column(scale=2):
+                            github_token = gr.Textbox(
+                                label="Token GitHub", 
+                                type="password",
+                                placeholder="Klik icon (?) untuk panduan"
+                            )
+                        with gr.Column(scale=1):
+                            branch = gr.Textbox(
+                                label="Branch (opsional)",
+                                placeholder="main"
+                            )
+
+                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
+                        )
+                    
+                    file_list = gr.HTML(
+                        value="<div class='file-list'>Belum ada file yang dipilih</div>",
+                        label="Daftar File Terpilih"
+                    )
+
+                # Chat Interface
+                with gr.Group():
+                    chat_history = gr.Chatbot(
+                        label="📝 Riwayat Chat",
+                        height=500,
+                        show_label=True,
+                        type="messages"  # Fix deprecation warning
+                    )
+                    
+                    with gr.Row():
+                        chat_input = gr.Textbox(
+                            label="💭 Tanyakan tentang Repository",
+                            placeholder="Ketik pertanyaan Anda di sini...",
+                            lines=3
+                        )
+                        send_button = gr.Button("📤 Kirim", variant="primary")
+
+                    loading_indicator = gr.HTML(
+                        '<div id="loading" style="display:none">Memproses permintaan...</div>'
+                    )
 
-            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")
+        # Event Handlers
+        def handle_clone(repo_url, github_token, branch):
+            if not repo_url:
+                return "⚠️ URL repository diperlukan!", [], []
             
-            provider.change(
-                fn=update_model_list,
-                inputs=[provider],
-                outputs=[model_dropdown]
-            )
+            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, []
+            
+            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>"
+            html += "</div>"
+            return html
+
+        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-1.5-mini")
+            else:  # OLLAMA
+                return gr.Dropdown(choices=OLLAMA_MODELS, value="llama2")
+
+        # Connect Events
+        provider.change(
+            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]
+        )
+
+        file_selector.change(
+            fn=update_file_list,
+            inputs=[file_selector],
+            outputs=[file_list]
+        )
 
             with gr.Tab("📊 Analisis Repository", elem_classes="mobile-full"):
                 with gr.Group():