Update app.py

app.py

@@ -371,6 +371,8 @@ def process_code_blocks(text):
     processed_text = re.sub(pattern, replace_code_block, text, flags=re.DOTALL)
     return processed_text
 
+analyzer = RepoAnalyzer()
+
 def create_ui():
     # Get local time
     local_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
@@ -597,89 +599,182 @@ 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:
-                files = list(analyzer.repo_content.keys())
-                return message, files, []
+                # Get files with metadata for better display
+                files = []
+                file_metadata = []
+                
+                for file_path in analyzer.repo_content.keys():
+                    files.append(file_path)
+                    content = analyzer.repo_content[file_path]
+                    size = len(content.encode('utf-8'))
+                    file_metadata.append({
+                        'path': file_path,
+                        'size': f"{size / 1024:.1f} KB",
+                        'lines': len(content.splitlines())
+                    })
+                
+                # Create rich HTML display for clone status
+                status_html = f"""
+                <div style="padding: 1rem; background: #f0f8ff; border-radius: 8px; border: 1px solid #add8e6;">
+                    <h3 style="margin-top: 0;">✅ Repository berhasil di-clone!</h3>
+                    <p><strong>Nama:</strong> {analyzer.current_repo}</p>
+                    <p><strong>Jumlah file:</strong> {len(files)}</p>
+                    <p><strong>Status:</strong> Siap untuk analisis</p>
+                </div>
+                """
+                
+                # Create initial file list display
+                file_list_html = create_file_list_html(file_metadata)
+                
+                return status_html, files, file_list_html
             return message, [], []
 
-        def update_file_list(selected):
-            if not selected:
+        def create_file_list_html(file_metadata):
+            if not file_metadata:
                 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 += "<style>\n"
+            html += ".file-list { max-height: 300px; overflow-y: auto; }\n"
+            html += ".file-item { display: flex; justify-content: space-between; padding: 8px; border-bottom: 1px solid #eee; }\n"
+            html += ".file-item:hover { background: #f5f5f5; }\n"
+            html += ".file-info { display: flex; gap: 1rem; color: #666; font-size: 0.9em; }\n"
+            html += "</style>"
+            
+            for metadata in file_metadata:
+                html += f"""
+                <div class='file-item'>
+                    <div class='file-path'>{metadata['path']}</div>
+                    <div class='file-info'>
+                        <span>{metadata['size']}</span>
+                        <span>{metadata['lines']} lines</span>
+                    </div>
+                </div>
+                """
             html += "</div>"
             return html
 
-        # 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]
-        )
-
-        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
-        )
+        def update_file_list(selected):
+            if not selected:
+                return "<div class='file-list'>Belum ada file yang dipilih</div>"
+            
+            file_metadata = []
+            for file_path in selected:
+                if file_path in analyzer.repo_content:
+                    content = analyzer.repo_content[file_path]
+                    size = len(content.encode('utf-8'))
+                    file_metadata.append({
+                        'path': file_path,
+                        'size': f"{size / 1024:.1f} KB",
+                        'lines': len(content.splitlines())
+                    })
+            
+            return create_file_list_html(file_metadata)
 
-        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
-        )
+        # Modified handle_chat function with better file context handling
+        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:
+                # Improved file context formatting
+                file_context = ""
+                if selected_files:
+                    file_context = "\n\nAnalisis berdasarkan file yang dipilih:\n"
+                    for file in selected_files:
+                        content = analyzer.repo_content.get(file, "")
+                        file_context += f"\n### File: {file}\n```\n{content}\n```\n"
+                
+                enhanced_message = f"{message}\n{file_context}"
+                
+                # Rest of the handle_chat function remains the same
+                full_response = ""
+                if provider_choice == AIProvider.XAI:
+                    async for chunk in analyzer.stream_xai_response(enhanced_message, xai_key, model_name):
+                        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)}"
+                yield history
 
-        # 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]
-        )
+        # Add additional CSS for improved repository display
+        gr.Markdown("""
+            <style>
+            /* Additional styles for repository content */
+            .repo-status {
+                margin: 1rem 0;
+                padding: 1rem;
+                background: #f0f8ff;
+                border-radius: 8px;
+                border: 1px solid #add8e6;
+            }
+            .file-list {
+                border: 1px solid #ddd;
+                border-radius: 8px;
+                margin-top: 1rem;
+                background: white;
+            }
+            .file-item {
+                transition: background-color 0.2s;
+            }
+            .file-info {
+                opacity: 0.7;
+            }
+            .file-path {
+                font-family: monospace;
+                overflow: hidden;
+                text-overflow: ellipsis;
+                white-space: nowrap;
+            }
+            
+            /* Improved scrollbars for file list */
+            .file-list::-webkit-scrollbar {
+                width: 8px;
+                height: 8px;
+            }
+            .file-list::-webkit-scrollbar-track {
+                background: #f1f1f1;
+                border-radius: 4px;
+            }
+            .file-list::-webkit-scrollbar-thumb {
+                background: #888;
+                border-radius: 4px;
+            }
+            .file-list::-webkit-scrollbar-thumb:hover {
+                background: #555;
+            }
+            </style>
+        """)
 
         return app