Update app.py

app.py

@@ -25,8 +25,7 @@ from concurrent.futures import ThreadPoolExecutor
 from hdb_scan import HDBSCAN
 import websockets
 from websockets.exceptions import ConnectionClosed
-from ot_code_editor import OTCodeEditor  # Assuming you have OTCodeEditor installed
-
+from code_editor import code_editor, generate_code_snippet
 # ========== Configuration ==========
 WORKSPACE = Path("/tmp/issue_workspace")
 WORKSPACE.mkdir(exist_ok=True)
@@ -253,6 +252,8 @@ def create_ui():
             with gr.Column(scale=1):
                 model_select = gr.Dropdown(choices=list(HF_MODELS.keys()), value="Mistral-8x7B", 
                                         label="AI Model", info="Choose your resolution strategy")
+                language_select = gr.Dropdown(choices=["python", "javascript", "java", "c", "cpp", "html", "css", "bash", "ruby", "go", "php", "rust", "typescript"], 
+                                               value="python", label="Select Language", info="Choose the programming language for the code editor")
                 crawl_btn = gr.Button("🚀 Scan Repository", variant="primary")
 
         with gr.Tabs():
@@ -297,14 +298,14 @@ def create_ui():
                 with gr.Row():
                     gr.Markdown("### 🏆 Achievement System")
                     badges = gr.HTML("<div class='badges'></div>")
-
+        
         # Enhanced Event Handlers
         async def generate_patch(issue_num):
             patch = await manager.generate_code_patch(issue_num)
             return gr.JSON(value=patch)
 
         def update_code_editor(files):
-            return code_editor(value=files)
+            return code_editor(value=files, language=language_select.value)
 
         issue_list.select(
             fn=lambda evt: (evt[0], manager.issues[evt[0]]['body']),
@@ -343,8 +344,8 @@ def create_ui():
         # ...
 
         issue_num.change(
-            fn=lambda issue_num: create_code_editor(issue_num),
-            inputs=[issue_num],
+            fn=lambda issue_num: create_code_editor(issue_num, language_select.value),
+            inputs=[issue_num, language_select],
             outputs=code_edit
         )