Update app.py

app.py

@@ -93,16 +93,16 @@ def _clean_filename(filename_line_content):
     path_match = re.match(r'^([\w\-\.\s\/\\]+)', text) # Adjusted to be more general
     if path_match:
         # Further clean if it looks like "path/to/file (description)"
-        return re.split(r'\s*\(', path_match.group(1).strip(), 1).strip()
+        return re.split(r'\s*\(', path_match.group(1).strip(), 1).strip() # Corrected split index
 
     # Fallback for more complex lines, like "### File: `src/app.py` (main application)"
     backtick_match = re.search(r'`([^`]+)`', text)
     if backtick_match:
         potential_fn = backtick_match.group(1).strip()
-        cleaned_fn = re.split(r'\s*\(|\s{2,}', potential_fn, 1).strip()
+        cleaned_fn = re.split(r'\s*\(|\s{2,}', potential_fn, 1).strip() # Corrected split index
         if cleaned_fn: return cleaned_fn
 
-    filename_candidate = re.split(r'\s*\(|\s{2,}', text, 1).strip()
+    filename_candidate = re.split(r'\s*\(|\s{2,}', text, 1).strip() # Corrected split index
     filename_candidate = filename_candidate.strip('`\'":;,') # Clean common wrapping chars
     return filename_candidate if filename_candidate else text # Fallback to original if all else fails
 
@@ -200,8 +200,7 @@ def _convert_gr_history_to_api_messages(system_prompt, gr_history, current_user_
     return messages
 
 def get_latest_bot_message_as_json(gr_history):
-    # Check if chat_history and the last bot message exist
-    if not gr_history or not gr_history[-1] or len(gr_history[-1]) < 2 or not gr_history[-1]:
+    if not gr_history or not gr_history[-1] or len(gr_history[-1]) < 2 or not gr_history[-1]: # Check bot message at index 1
         return json.dumps([])
     return json.dumps([{"role": BOT_ROLE_NAME, "content": gr_history[-1]}], indent=2)
 
@@ -233,7 +232,7 @@ def _generate_ui_outputs_from_cache(owner, space_name):
         formatted_md_val = f"# Space: {owner or 'owner'}/{space_name or 'space'}\n## File Structure\n{bbb}\n📁 Root\n{bbb}\n\n*No files found or defined for this Space.*" if owner and space_name else "*Load or define a Space to see its Markdown structure.*"
     return formatted_md_val, preview_md_val, gr.update(value=download_file, interactive=download_file is not None)
 
-def handle_groq_chat_submit(user_message, chat_history, api_key_input, model_select, system_prompt, hf_owner_name, hf_repo_name, _current_formatted_markdown): # _current_formatted_markdown not used
+def handle_groq_chat_submit(user_message, chat_history, api_key_input, model_select, system_prompt, hf_owner_name, hf_repo_name, _current_formatted_markdown):
     global parsed_code_blocks_state_cache
     _chat_msg_in, _chat_hist, _status = "", list(chat_history), "Initializing..."
     _detected_files_update, _formatted_output_update, _download_btn_update = gr.update(), gr.update(), gr.update(interactive=False, value=None)
@@ -253,27 +252,35 @@ def handle_groq_chat_submit(user_message, chat_history, api_key_input, model_sel
         _status = f"Waiting for {model_select}..."; yield (_chat_msg_in, _chat_hist, _status, _detected_files_update, _formatted_output_update, _download_btn_update)
         response = requests.post(GROQ_API_ENDPOINT, headers=headers, json=payload, timeout=180); response.raise_for_status()
         api_resp_json = response.json()
-        if api_resp_json.get("choices") and api_resp_json["choices"].get("message"): # Corrected indexing
-            bot_response_actual = api_resp_json["choices"]["message"].get("content") # Corrected indexing
-            if bot_response_actual:
-                _chat_hist[-1] = (user_message, bot_response_actual); _status = "AI response received. Processing files..."
-                yield (_chat_msg_in, _chat_hist, _status, _detected_files_update, _formatted_output_update, _download_btn_update)
-                latest_bot_message_json = get_latest_bot_message_as_json(_chat_hist)
-                parsing_res = _parse_chat_stream_logic(latest_bot_message_json, existing_files_state=parsed_code_blocks_state_cache) # Use cache
-                if parsing_res["error_message"]:
-                    _status = f"Parsing Error: {parsing_res['error_message']}"
-                    _detected_files_update = gr.Markdown(f"## Parsing Error\n`{escape_html_for_markdown(parsing_res['error_message'])}`")
-                else:
-                    _formatted_output_update, _detected_files_update, _download_btn_update = _generate_ui_outputs_from_cache(hf_owner_name, hf_repo_name)
-                    _status = "Processing complete. Previews updated."
-                yield (_chat_msg_in, _chat_hist, _status, _detected_files_update, _formatted_output_update, _download_btn_update); return
-            else: error_msg = "API Error: Empty message content."
-        else: error_msg = f"API Error: Unexpected response: {api_resp_json}"
+        
+        bot_response_actual = None
+        if api_resp_json.get("choices") and len(api_resp_json["choices"]) > 0:
+            message_obj = api_resp_json["choices"].get("message")
+            if message_obj and isinstance(message_obj, dict): # Check if message_obj is a dict
+                 bot_response_actual = message_obj.get("content")
+
+        if bot_response_actual:
+            _chat_hist[-1] = (user_message, bot_response_actual); _status = "AI response received. Processing files..."
+            yield (_chat_msg_in, _chat_hist, _status, _detected_files_update, _formatted_output_update, _download_btn_update)
+            latest_bot_message_json = get_latest_bot_message_as_json(_chat_hist)
+            parsing_res = _parse_chat_stream_logic(latest_bot_message_json, existing_files_state=parsed_code_blocks_state_cache)
+            if parsing_res["error_message"]:
+                _status = f"Parsing Error: {parsing_res['error_message']}"
+                _detected_files_update = gr.Markdown(f"## Parsing Error\n`{escape_html_for_markdown(parsing_res['error_message'])}`")
+            else:
+                _formatted_output_update, _detected_files_update, _download_btn_update = _generate_ui_outputs_from_cache(hf_owner_name, hf_repo_name)
+                _status = "Processing complete. Previews updated."
+            yield (_chat_msg_in, _chat_hist, _status, _detected_files_update, _formatted_output_update, _download_btn_update); return
+        else:
+            error_msg = f"API Error: Unexpected response structure or empty message content. Response: {json.dumps(api_resp_json)}"
+        
         _chat_hist[-1] = (user_message, error_msg); _status = error_msg
     except requests.exceptions.HTTPError as e: error_msg = f"API HTTP Error: {e} - {e.response.text if e.response else 'No details'}"
     except requests.exceptions.RequestException as e: error_msg = f"API Request Error: {e}"
     except Exception as e: error_msg = f"Unexpected error in chat submit: {e}"
-    if _chat_hist and len(_chat_hist[-1]) == 2 and _chat_hist[-1] is None: _chat_hist[-1] = (user_message, error_msg) # Check length before assignment
+    
+    if _chat_hist and len(_chat_hist[-1]) == 2 and _chat_hist[-1] is None:
+        _chat_hist[-1] = (user_message, error_msg)
     _status = error_msg
     yield (_chat_msg_in, _chat_hist, _status, _detected_files_update, _formatted_output_update, _download_btn_update)
 
@@ -464,7 +471,7 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="teal", secondary_hue="orange"))
                 with gr.TabItem("✏️ Edit Space Files"):
                     gr.Markdown("Select a file to view, edit, or delete. Changes are committed to HF Hub.")
                     file_browser_dropdown = gr.Dropdown(label="Select File in Space", choices=[], interactive=True, visible=False, info="Load/build Space first.")
-                    file_content_editor = gr.Code(label="File Content Editor", language="python", lines=15, interactive=True)
+                    file_content_editor = gr.Code(label="File Content Editor", language="python", lines=15, interactive=True) # Changed language to python from plaintext
                     commit_message_input = gr.Textbox(label="Commit Message", placeholder="e.g., Updated app.py", value="Update via AI Space Editor")
                     with gr.Row(): update_file_button = gr.Button("Commit Changes", variant="primary", scale=2); delete_file_button = gr.Button("🗑️ Delete Selected File", variant="stop", scale=1)
                     edit_status_display = gr.Textbox(label="File Edit/Delete Status", interactive=False, lines=2, value="*Select file...*")