appkit-assistant 0.16.3__py3-none-any.whl → 0.17.1__py3-none-any.whl

appkit_assistant/state/thread_state.py

@@ -14,10 +14,12 @@ import json
 import logging
 import uuid
 from collections.abc import AsyncGenerator
+from datetime import UTC, datetime
 from typing import Any
 
 import reflex as rx
 
+from appkit_assistant.backend import file_manager
 from appkit_assistant.backend.model_manager import ModelManager
 from appkit_assistant.backend.models import (
     AIModel,
@@ -31,10 +33,11 @@ from appkit_assistant.backend.models import (
     ThinkingType,
     ThreadModel,
     ThreadStatus,
+    UploadedFile,
 )
 from appkit_assistant.backend.repositories import mcp_server_repo
+from appkit_assistant.backend.response_accumulator import ResponseAccumulator
 from appkit_assistant.backend.services.thread_service import ThreadService
-from appkit_assistant.logic.response_accumulator import ResponseAccumulator
 from appkit_assistant.state.thread_list_state import ThreadListState
 from appkit_commons.database.session import get_asyncdb_session
 from appkit_user.authentication.states import UserSession
@@ -69,6 +72,13 @@ class ThreadState(rx.State):
     thinking_expanded: bool = False
     current_activity: str = ""
 
+    # File upload state
+    uploaded_files: list[UploadedFile] = []
+
+    # Editing state
+    editing_message_id: str | None = None
+    edited_message_content: str = ""
+
     # Internal logic helper (not reactive)
     @property
     def _thread_service(self) -> ThreadService:
@@ -97,6 +107,7 @@ class ThreadState(rx.State):
     _initialized: bool = False
     _current_user_id: str = ""
     _skip_user_message: bool = False  # Skip adding user message (for OAuth resend)
+    _pending_file_cleanup: list[str] = []  # Files to delete after processing
 
     # -------------------------------------------------------------------------
     # Computed properties
@@ -135,6 +146,14 @@ class ThreadState(rx.State):
         model = ModelManager().get_model(self.selected_model)
         return model.supports_tools if model else False
 
+    @rx.var
+    def selected_model_supports_attachments(self) -> bool:
+        """Check if the currently selected model supports attachments."""
+        if not self.selected_model:
+            return False
+        model = ModelManager().get_model(self.selected_model)
+        return model.supports_attachments if model else False
+
     @rx.var
     def get_unique_reasoning_sessions(self) -> list[str]:
         """Get unique reasoning session IDs."""
@@ -432,11 +451,131 @@ class ThreadState(rx.State):
         self.thinking_items = []
         self.image_chunks = []
         self.show_thinking = False
+        self._clear_uploaded_files()
+
+    # -------------------------------------------------------------------------
+    # File upload handling
+    # -------------------------------------------------------------------------
+
+    @rx.event
+    async def handle_upload(self, files: list[rx.UploadFile]) -> None:
+        """Handle uploaded files from the browser.
+
+        Moves files to user-specific directory and adds them to state.
+        """
+        user_session: UserSession = await self.get_state(UserSession)
+        user_id = user_session.user.user_id if user_session.user else "anonymous"
+
+        for upload_file in files:
+            try:
+                # Save uploaded file to disk first
+                upload_data = await upload_file.read()
+                temp_path = rx.get_upload_dir() / upload_file.filename
+                temp_path.parent.mkdir(parents=True, exist_ok=True)
+                temp_path.write_bytes(upload_data)
+
+                # Move to user directory
+                final_path = file_manager.move_to_user_directory(
+                    str(temp_path), str(user_id)
+                )
+                file_size = file_manager.get_file_size(final_path)
+
+                uploaded = UploadedFile(
+                    filename=upload_file.filename,
+                    file_path=final_path,
+                    size=file_size,
+                )
+                self.uploaded_files = [*self.uploaded_files, uploaded]
+                logger.info("Uploaded file: %s", upload_file.filename)
+            except Exception as e:
+                logger.error("Failed to upload file %s: %s", upload_file.filename, e)
+
+    @rx.event
+    def remove_file_from_prompt(self, file_path: str) -> None:
+        """Remove an uploaded file from the prompt."""
+        # Delete the file from disk
+        file_manager.cleanup_uploaded_files([file_path])
+        # Remove from state
+        self.uploaded_files = [
+            f for f in self.uploaded_files if f.file_path != file_path
+        ]
+        logger.debug("Removed uploaded file: %s", file_path)
+
+    def _clear_uploaded_files(self) -> None:
+        """Clear all uploaded files from state and disk."""
+        if not self.uploaded_files:
+            return
+        count = len(self.uploaded_files)
+        file_paths = [f.file_path for f in self.uploaded_files]
+        file_manager.cleanup_uploaded_files(file_paths)
+        self.uploaded_files = []
+        logger.debug("Cleared %d uploaded files", count)
 
     # -------------------------------------------------------------------------
     # Message processing
     # -------------------------------------------------------------------------
 
+    @rx.event
+    def set_editing_mode(self, message_id: str, content: str) -> None:
+        """Enable editing mode for a message."""
+        self.editing_message_id = message_id
+        self.edited_message_content = content
+
+    @rx.event
+    def set_edited_message_content(self, content: str) -> None:
+        """Set the content of the message currently being edited."""
+        self.edited_message_content = content
+
+    @rx.event
+    def cancel_edit(self) -> None:
+        """Cancel editing mode."""
+        self.editing_message_id = None
+        self.edited_message_content = ""
+
+    @rx.event(background=True)
+    async def submit_edited_message(self) -> AsyncGenerator[Any, Any]:
+        """Submit edited message."""
+        async with self:
+            content = self.edited_message_content.strip()
+            if len(content) < 1:
+                yield rx.toast.error(
+                    "Nachricht darf nicht leer sein", position="top-right"
+                )
+                return
+
+            # Find message index
+            msg_index = -1
+            for i, m in enumerate(self.messages):
+                if m.id == self.editing_message_id:
+                    msg_index = i
+                    break
+
+            if msg_index == -1:
+                self.cancel_edit()
+                return
+
+            target_message = self.messages[msg_index]
+
+            # Update message
+            target_message.original_text = (
+                target_message.original_text or target_message.text
+            )
+            target_message.text = content
+
+            # Remove all messages AFTER this one
+            self.messages = self.messages[: msg_index + 1]
+
+            # Set prompt to bypass empty check in _begin_message_processing
+            self.prompt = content
+            self._skip_user_message = True
+
+            # Clear edit state
+            self.editing_message_id = None
+            self.edited_message_content = ""
+
+        # Trigger processing
+        await self._process_message()
+
     @rx.event(background=True)
     async def submit_message(self) -> AsyncGenerator[Any, Any]:
         """Submit a message and process the response."""
@@ -451,6 +590,76 @@ class ThreadState(rx.State):
             }
         """)
 
+    @rx.event(background=True)
+    async def delete_message(self, message_id: str) -> None:
+        """Delete a message from the conversation."""
+        async with self:
+            self.messages = [m for m in self.messages if m.id != message_id]
+            self._thread.messages = self.messages
+
+            if self._thread.state != ThreadStatus.NEW:
+                await self._thread_service.save_thread(
+                    self._thread, self.current_user_id
+                )
+
+    @rx.event
+    def copy_message(self, text: str) -> list[Any]:
+        """Copy message text to clipboard."""
+        return [
+            rx.set_clipboard(text),
+            rx.toast.success("Nachricht kopiert"),
+        ]
+
+    @rx.event
+    def download_message(self, text: str, message_id: str) -> Any:
+        """Download message as markdown file."""
+        timestamp = datetime.now(UTC).strftime("%Y%m%d_%H%M%S")
+        filename = (
+            f"message_{message_id}_{timestamp}.md"
+            if message_id
+            else f"message_{timestamp}.md"
+        )
+
+        # Use JavaScript to trigger download
+        return rx.call_script(f"""
+            const blob = new Blob([{json.dumps(text)}], {{type: 'text/markdown'}});
+            const url = URL.createObjectURL(blob);
+            const a = document.createElement('a');
+            a.href = url;
+            a.download = '{filename}';
+            document.body.appendChild(a);
+            a.click();
+            document.body.removeChild(a);
+            URL.revokeObjectURL(url);
+        """)
+
+    @rx.event(background=True)
+    async def retry_message(self, message_id: str) -> None:
+        """Retry generating a message."""
+        async with self:
+            # Find message index
+            index = -1
+            for i, msg in enumerate(self.messages):
+                if msg.id == message_id:
+                    index = i
+                    break
+
+            if index == -1:
+                return
+
+            # Keep context up to this message
+            # effectively removing this message and everything after
+            self.messages = self.messages[:index]
+
+            # Set prompt to bypass check (content checks)
+            self.prompt = "Regenerate"
+
+            # Flag to skip adding a new user message
+            self._skip_user_message = True
+
+        # Trigger processing directly
+        await self._process_message()
+
     async def _process_message(self) -> None:
         """Process the current message and stream the response."""
         logger.debug("Processing message: %s", self.prompt)
@@ -458,7 +667,7 @@ class ThreadState(rx.State):
         start = await self._begin_message_processing()
         if not start:
             return
-        current_prompt, selected_model, mcp_servers, is_new_thread = start
+        current_prompt, selected_model, mcp_servers, file_paths, is_new_thread = start
 
         processor = ModelManager().get_processor_for_model(selected_model)
         if not processor:
@@ -475,11 +684,16 @@ class ThreadState(rx.State):
             accumulator = ResponseAccumulator()
             accumulator.attach_messages_ref(self.messages)
 
+            # Clear uploaded files from UI and mark for cleanup after processing
+            self.uploaded_files = []
+            self._pending_file_cleanup = file_paths
+
         first_response_received = False
         try:
             async for chunk in processor.process(
                 self.messages,
                 selected_model,
+                files=file_paths or None,
                 mcp_servers=mcp_servers,
                 user_id=user_id,
             ):
@@ -506,7 +720,7 @@ class ThreadState(rx.State):
 
     async def _begin_message_processing(
         self,
-    ) -> tuple[str, str, list[MCPServer], bool] | None:
+    ) -> tuple[str, str, list[MCPServer], list[str], bool] | None:
         """Prepare state for sending a message. Returns None if no-op."""
         async with self:
             current_prompt = self.prompt.strip()
@@ -523,12 +737,21 @@ class ThreadState(rx.State):
 
             is_new_thread = self._thread.state == ThreadStatus.NEW
 
+            # Capture file paths before clearing
+            file_paths = [f.file_path for f in self.uploaded_files]
+            # Capture filenames for message display
+            attachment_names = [f.filename for f in self.uploaded_files]
+
             # Add user message unless skipped (e.g., OAuth resend)
             if self._skip_user_message:
                 self._skip_user_message = False
             else:
                 self.messages.append(
-                    Message(text=current_prompt, type=MessageType.HUMAN)
+                    Message(
+                        text=current_prompt,
+                        type=MessageType.HUMAN,
+                        attachments=attachment_names,
+                    )
                 )
             # Always add assistant placeholder
             self.messages.append(Message(text="", type=MessageType.ASSISTANT))
@@ -540,7 +763,13 @@ class ThreadState(rx.State):
                 return None
 
             mcp_servers = self.selected_mcp_servers
-            return current_prompt, selected_model, mcp_servers, is_new_thread
+            return (
+                current_prompt,
+                selected_model,
+                mcp_servers,
+                file_paths,
+                is_new_thread,
+            )
 
     async def _stop_processing_with_error(self, error_msg: str) -> None:
         """Stop processing and show an error message."""
@@ -650,6 +879,11 @@ class ThreadState(rx.State):
             self.processing = False
             self.current_activity = ""
 
+            # Clean up uploaded files from disk
+            if self._pending_file_cleanup:
+                file_manager.cleanup_uploaded_files(self._pending_file_cleanup)
+                self._pending_file_cleanup = []
+
     def _handle_auth_required_from_accumulator(
         self, accumulator: ResponseAccumulator
     ) -> None:

{appkit_assistant-0.16.3.dist-info → appkit_assistant-0.17.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: appkit-assistant
-Version: 0.16.3
+Version: 0.17.1
 Summary: Add your description here
 Project-URL: Homepage, https://github.com/jenreh/appkit
 Project-URL: Documentation, https://github.com/jenreh/appkit/tree/main/docs
@@ -16,9 +16,12 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Classifier: Topic :: Software Development :: User Interfaces
 Requires-Python: >=3.13
+Requires-Dist: anthropic>=0.40.0
 Requires-Dist: appkit-commons
 Requires-Dist: appkit-mantine
 Requires-Dist: appkit-ui
+Requires-Dist: google-genai>=1.52.0
+Requires-Dist: mcp>=1.0.0
 Requires-Dist: openai>=2.3.0
 Requires-Dist: reflex>=0.8.22
 Description-Content-Type: text/markdown

{appkit_assistant-0.16.3.dist-info → appkit_assistant-0.17.1.dist-info}/RECORD

@@ -1,34 +1,39 @@
-appkit_assistant/configuration.py,sha256=3nBL-dEGYsvSnRDNpxtikZn4QMMkMlNbb4VqGOPolJI,346
+appkit_assistant/configuration.py,sha256=Wpo3EuGWQrV0WIQnAhkj19PzgGkJoAde5ky-MA7kJwg,429
 appkit_assistant/pages.py,sha256=gDvBweUO2WjrhP1RE5AAkjL1_S-givWr3CkkGZKws_E,471
+appkit_assistant/backend/file_manager.py,sha256=54SYphu6FsxbEYuMx8ohQiSAeY2gGDV1q3S6RZuNku0,3153
 appkit_assistant/backend/mcp_auth_service.py,sha256=lYQEe4yOZ48ear6dvcuOXsaOc6RClIBMsOOkV7SG5Aw,27768
 appkit_assistant/backend/model_manager.py,sha256=fmv3yP63LxDnne4vjT7IzETTI2aSxViC2FSUfHQajlk,4382
-appkit_assistant/backend/models.py,sha256=jNhJ6z8LKN6geJ6KTU60d10MVhWFcof6yJIVP18gtOs,7983
+appkit_assistant/backend/models.py,sha256=GCFWUUPsVspcFEv9naQ8n2bMU8FzMNrmqm3VgW1Fqfw,8346
 appkit_assistant/backend/processor.py,sha256=LtkkC6v0idNtbuMm3Sw5J9zIjYqtNhtxYxjYefPSS9A,2135
 appkit_assistant/backend/repositories.py,sha256=R-7kYdxg4RWQrTEOU4tbcOEhJA_FlesWrt65UpItRSU,5547
+appkit_assistant/backend/response_accumulator.py,sha256=BCK-Ut_Wmo7rhEqOb7QlqXO8TrtANjL80FbX_AvMl1Q,10056
 appkit_assistant/backend/system_prompt_cache.py,sha256=83OIyixeTb3HKOy3XIzPyTAE-G2JyqrfcG8xVeTS2Ls,5514
-appkit_assistant/backend/processors/lorem_ipsum_processor.py,sha256=j-MZhzibrtabzbGB2Pf4Xcdlr1TlTYWNRdE22LsDp9Q,4635
-appkit_assistant/backend/processors/openai_base.py,sha256=IQS4m375BOD_K0PBFOk4i7wL1z5MEiPFxbSmC-HBNgU,4414
-appkit_assistant/backend/processors/openai_chat_completion_processor.py,sha256=nTxouoXDU6VcQr8UhA2KiMNt60KvIwM8cH9Z8lo4dXY,4218
-appkit_assistant/backend/processors/openai_responses_processor.py,sha256=z4FxW7A9ysC0scyZ4gpvpoAsweYXSlMFDzGhJO5Lu2U,29094
+appkit_assistant/backend/processors/claude_base.py,sha256=j0DhBn8EVAjW_bfCghXaEHyORO1raUNdQeemVWCKJlA,5376
+appkit_assistant/backend/processors/claude_responses_processor.py,sha256=_quuN5FRsPSpQjaNIceN-Z66Fb3RnhXtcsS4gE0NrPA,32980
+appkit_assistant/backend/processors/gemini_base.py,sha256=ijCa8-_xdddD6ms_pkUjCj5kHZOoQ1wfw-7AhhXu6vo,2286
+appkit_assistant/backend/processors/gemini_responses_processor.py,sha256=V5jZ83a3vi60aFTtRnbwP2q1PyEGw1iB6dYne-4CgI4,26484
+appkit_assistant/backend/processors/lorem_ipsum_processor.py,sha256=WR4s1RyOi5xgWVpre04xoxoMLIAINkpdvRCkSfNuoMc,4699
+appkit_assistant/backend/processors/openai_base.py,sha256=TBQCZW7RxaPMag_DX_tpLyMpZ_PZ_KiO8_sFN49OCpk,4424
+appkit_assistant/backend/processors/openai_chat_completion_processor.py,sha256=XFp_V9Yp4KcTmicbCe3fl_3G2tP5Ffm-xLVqW50HyQw,4629
+appkit_assistant/backend/processors/openai_responses_processor.py,sha256=hysK7mlsG2t8SfltQpCvKA6MxpeIomRMGX-tSMaUshc,29492
 appkit_assistant/backend/processors/perplexity_processor.py,sha256=weHukv78MSCF_uSCKGSMpNYHsET9OB8IhpvUiMfPQ8A,3355
 appkit_assistant/backend/services/thread_service.py,sha256=LpM8ZZHt1o4MYEKzH_XPURSi3qS6p3pAQA53tOE53MU,4663
-appkit_assistant/components/__init__.py,sha256=5tzK5VjX9FGKK-qTUHLjr8-ohT4ykb4a-zC-I3yeRLY,916
-appkit_assistant/components/composer.py,sha256=F4VPxWp4P6fvTW4rQ7S-YWn0eje5c3jGsWrpC1aewss,3885
+appkit_assistant/components/__init__.py,sha256=ptv0wShA4CHjgNpehlHqgoinCl-yyofjkV6rTTNIRHE,954
+appkit_assistant/components/composer.py,sha256=jQNZd8Y0eQgbfC3d1AV1osRNnwfztEpXtrFp7aagXaY,7190
 appkit_assistant/components/composer_key_handler.py,sha256=KyZYyhxzFR8DH_7F_DrvTFNT6v5kG6JihlGTmCv2wv0,1028
 appkit_assistant/components/mcp_oauth.py,sha256=puLwxAhmF25BjnZMdJbKIfC6bFXK2D8LybOX0kD7Ri4,1737
 appkit_assistant/components/mcp_server_dialogs.py,sha256=afIImmhfrNyLmxDZBpCxHxvD8HKpDanIloLEC8dJgro,23444
 appkit_assistant/components/mcp_server_table.py,sha256=1dziN7hDDvE8Y3XcdIs0wUPv1H64kP9gRAEjgH9Yvzo,2323
-appkit_assistant/components/message.py,sha256=clqw5ISSO10NbbpD3OCfGo2CuefFl5-EIJXmlQEHncY,13554
+appkit_assistant/components/message.py,sha256=SJcwWfSw5XxiSHtKirJUun9jkx3-ATpZ0AHLv82KwXc,19824
 appkit_assistant/components/system_prompt_editor.py,sha256=REl33zFmcpYRe9kxvFrBRYg40dV4L4FtVC_3ibLsmrU,2940
-appkit_assistant/components/thread.py,sha256=COl8PxMht4YIublIL-iOcvcJnWVpZfmLGifHC6dm7Rk,8421
+appkit_assistant/components/thread.py,sha256=-KK1vcEmITR_oDGLbQhIa2WxRCVxEqel12m4haS7y9w,8461
 appkit_assistant/components/threadlist.py,sha256=1xVakSTQYi5-wgED3fTJVggeIjL_fkthehce0wKUYtM,4896
 appkit_assistant/components/tools_modal.py,sha256=12iiAVahy3j4JwjGfRlegVEa4ePhGsEu7Bq92JLn1ZI,3353
-appkit_assistant/logic/response_accumulator.py,sha256=85NKtIxpWVsI4rDkNAy_U34uavY8NoUEp9n2kO0ief4,8060
 appkit_assistant/state/mcp_oauth_state.py,sha256=6MofExrbOOEl_YUcUOqcSTN3h7KAaERI5IdVfXdVUVs,7669
 appkit_assistant/state/mcp_server_state.py,sha256=3AFvy53xx_eLTxw-LfJklPTgq4Ohqu4xs1QlLs-kU4U,11387
 appkit_assistant/state/system_prompt_state.py,sha256=zdnYrTnl7EszALRiodu6pcuQUd2tmtPG1eJ10j_OotI,7705
 appkit_assistant/state/thread_list_state.py,sha256=DEOR5Nklj1qfYaxSRMXCZdZRv2iq2Jb37JSg739_wL4,10250
-appkit_assistant/state/thread_state.py,sha256=rfbcdVTGcuC2MRnkTLP7x9fdUAHMK6NeS8cnhf621kg,30781
-appkit_assistant-0.16.3.dist-info/METADATA,sha256=OTRn32g9bgRtaFxt5p9x7r1WnC2rBD-DK9GszfUD-io,9403
-appkit_assistant-0.16.3.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-appkit_assistant-0.16.3.dist-info/RECORD,,
+appkit_assistant/state/thread_state.py,sha256=Hz7vRsBJ2cTV_xENuMoZBVSo2FGbR4uovdwiXF1zixk,39292
+appkit_assistant-0.17.1.dist-info/METADATA,sha256=t0hdCrMbCpvbETpblS3bmLtJ0ltfMXhGI9MTOtnrzII,9498
+appkit_assistant-0.17.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+appkit_assistant-0.17.1.dist-info/RECORD,,