Remove notebook download feature (keep Markdown and Code)

app.py

@@ -26,7 +26,6 @@ from config import (
     SUPPORTED_LANGUAGES,
     SYSTEM_PROMPT,
     EXAMPLE_PROMPTS,
-    NOTEBOOK_TEMPLATE,
     MARKDOWN_TEMPLATE,
 )
 
@@ -135,24 +134,6 @@ def create_download_file(
             content = MARKDOWN_TEMPLATE.format(query=query, response=response)
             filename = f"geoai_response_{timestamp}.md"
             
-        elif file_format == "notebook":
-            code_blocks = extract_code_blocks(response)
-            if not code_blocks:
-                # If no code blocks, create a markdown cell
-                code_cells = json.dumps([f"# No code blocks found\\n# Response:\\n# {response[:500]}..."])
-            else:
-                # Create code cells from extracted blocks
-                cells = [block["code"] for block in code_blocks if block["language"] == "python"]
-                if not cells:
-                    cells = [code_blocks[0]["code"]] if code_blocks else ["# No Python code found"]
-                code_cells = json.dumps(cells)
-            
-            content = NOTEBOOK_TEMPLATE.format(
-                query=query.replace('"', '\\"').replace('\n', '\\n'),
-                code_cells=code_cells
-            )
-            filename = f"geoai_notebook_{timestamp}.ipynb"
-            
         elif file_format == "code":
             code_blocks = extract_code_blocks(response)
             if not code_blocks:
@@ -184,11 +165,6 @@ def download_as_markdown(response: str, query: str) -> Optional[str]:
     return create_download_file(response, query, "markdown")
 
 
-def download_as_notebook(response: str, query: str) -> Optional[str]:
-    """Download response as Jupyter Notebook."""
-    return create_download_file(response, query, "notebook")
-
-
 def download_as_code(response: str, query: str) -> Optional[str]:
     """Download response as code file."""
     return create_download_file(response, query, "code")
@@ -301,7 +277,6 @@ def create_app():
         gr.Markdown("### 📥 Download Response")
         with gr.Row():
             download_md_btn = gr.Button("📄 Markdown", size="sm")
-            download_nb_btn = gr.Button("📓 Notebook", size="sm")
             download_code_btn = gr.Button("💻 Code File", size="sm")
         
         download_file = gr.File(label="Download", visible=False)
@@ -343,13 +318,6 @@ def create_app():
                 return gr.File(value=filepath, visible=True)
             return gr.File(visible=False)
         
-        def handle_download_nb(response, query):
-            """Handle notebook download."""
-            filepath = download_as_notebook(response, query)
-            if filepath:
-                return gr.File(value=filepath, visible=True)
-            return gr.File(visible=False)
-        
         def handle_download_code(response, query):
             """Handle code file download."""
             filepath = download_as_code(response, query)
@@ -391,12 +359,6 @@ def create_app():
             [download_file],
         )
         
-        download_nb_btn.click(
-            handle_download_nb,
-            [last_response, last_query],
-            [download_file],
-        )
-        
         download_code_btn.click(
             handle_download_code,
             [last_response, last_query],