appkit-assistant 0.16.2__py3-none-any.whl → 0.17.0__py3-none-any.whl

appkit_assistant/state/thread_state.py

@@ -18,6 +18,7 @@ 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 +32,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 +71,9 @@ class ThreadState(rx.State):
     thinking_expanded: bool = False
     current_activity: str = ""
 
+    # File upload state
+    uploaded_files: list[UploadedFile] = []
+
     # Internal logic helper (not reactive)
     @property
     def _thread_service(self) -> ThreadService:
@@ -97,6 +102,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 +141,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,6 +446,65 @@ 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
@@ -458,7 +531,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 +548,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 +584,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 +601,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 +627,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 +743,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.2.dist-info → appkit_assistant-0.17.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: appkit-assistant
-Version: 0.16.2
+Version: 0.17.0
 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.2.dist-info → appkit_assistant-0.17.0.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=vymgJ7npF-L-4WGXWwVUdHqLo3MAypPPaaD_SyDcAIw,8198
 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=dFJZiXcDn4V2POSk7Us6gQLsrUIRBMLdoOP61LwutdI,9766
 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/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=pzbczuw7MsoABdkuJGNxlebV8POA2D72XMzLoMhj9k4,26429
+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=nTxouoXDU6VcQr8UhA2KiMNt60KvIwM8cH9Z8lo4dXY,4218
-appkit_assistant/backend/processors/openai_responses_processor.py,sha256=z4FxW7A9ysC0scyZ4gpvpoAsweYXSlMFDzGhJO5Lu2U,29094
+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=xFMcrSMErHbbgZGl0rujNwhL7nRGwIp-S-a47cYwaJ4,14605
 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.2.dist-info/METADATA,sha256=kX6nu1t95GqH3szeSq0-2XQ2otiqLR2x_r0N_p3uDEA,9403
-appkit_assistant-0.16.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-appkit_assistant-0.16.2.dist-info/RECORD,,
+appkit_assistant/state/thread_state.py,sha256=xLydHA5YDFj3J1fGBQoAZsWg3zlcKOh2-9SzvlGvyU0,34742
+appkit_assistant-0.17.0.dist-info/METADATA,sha256=jzEwclPQN3VXn5APgK9EQMLiGKnug-wI-JRxPu44Kao,9498
+appkit_assistant-0.17.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+appkit_assistant-0.17.0.dist-info/RECORD,,