openai-sdk-helpers 0.1.2__tar.gz → 0.1.4__tar.gz

{openai_sdk_helpers-0.1.2 → openai_sdk_helpers-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: openai-sdk-helpers
-Version: 0.1.2
+Version: 0.1.4
 Summary: Composable helpers for OpenAI SDK agents, prompts, and storage
 Author: openai-sdk-helpers maintainers
 License: MIT

{openai_sdk_helpers-0.1.2 → openai_sdk_helpers-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "openai-sdk-helpers"
-version = "0.1.2"
+version = "0.1.4"
 requires-python = ">=3.10"
 readme = "README.md"
 description = "Composable helpers for OpenAI SDK agents, prompts, and storage"

{openai_sdk_helpers-0.1.2 → openai_sdk_helpers-0.1.4}/src/openai_sdk_helpers/streamlit_app/app.py

@@ -8,6 +8,7 @@ rendering, response execution, and resource cleanup.
 from __future__ import annotations
 
 import json
+import os
 import tempfile
 from pathlib import Path
 from typing import Any
@@ -30,22 +31,30 @@ from openai_sdk_helpers.utils import (
     log,
 )
 
-# Supported file extensions for OpenAI Assistants file search
+# Supported file extensions for OpenAI Assistants file search and vision
 SUPPORTED_FILE_EXTENSIONS = (
     ".csv",
     ".docx",
+    ".gif",
     ".html",
     ".json",
+    ".jpeg",
+    ".jpg",
     ".md",
     ".pdf",
+    ".png",
     ".pptx",
     ".txt",
+    ".webp",
     ".xlsx",
 )
 
 
 def _validate_file_type(filename: str) -> bool:
-    """Check if a file has a supported extension for vector storage.
+    """Check if a file has a supported extension for upload.
+
+    Supports both document formats (for file search) and image formats
+    (for vision analysis).
 
     Parameters
     ----------
@@ -61,6 +70,32 @@ def _validate_file_type(filename: str) -> bool:
     return file_ext in SUPPORTED_FILE_EXTENSIONS
 
 
+def _cleanup_temp_files(file_paths: list[str] | None = None) -> None:
+    """Delete temporary files that were created for uploads.
+
+    Parameters
+    ----------
+    file_paths : list[str] or None, default None
+        Specific file paths to delete. If None, deletes all tracked
+        temporary files from session state.
+
+    Notes
+    -----
+    Silently ignores errors when deleting files that may have already
+    been removed or are inaccessible.
+    """
+    paths_to_delete = file_paths or st.session_state.get("temp_file_paths", [])
+    for path in paths_to_delete:
+        try:
+            if os.path.exists(path):
+                os.remove(path)
+        except (OSError, IOError):
+            pass  # Silently ignore if file already deleted or inaccessible
+
+    if file_paths is None:
+        st.session_state["temp_file_paths"] = []
+
+
 def _extract_assistant_text(response: BaseResponse[Any]) -> str:
     """Extract the latest assistant message as readable text.
 
@@ -86,15 +121,33 @@ def _extract_assistant_text(response: BaseResponse[Any]) -> str:
     if message is None:
         return ""
 
+    # Check if the message content has output_text attribute
+    output_text = getattr(message.content, "output_text", None)
+    if output_text:
+        return str(output_text)
+
     content = getattr(message.content, "content", None)
     if content is None:
         return ""
 
     text_parts: list[str] = []
     for part in ensure_list(content):
-        text_value = getattr(getattr(part, "text", None), "value", None)
-        if text_value:
-            text_parts.append(text_value)
+        # Handle both dict-like parts and object-like parts
+        text_content = None
+        if hasattr(part, "text"):
+            text_content = getattr(part, "text", None)
+        elif isinstance(part, dict):
+            text_content = part.get("text")
+
+        if text_content:
+            # If text_content is a string, use it directly (dict-style)
+            if isinstance(text_content, str):
+                text_parts.append(text_content)
+            # If text_content is an object with a value attribute, extract that value (object-style)
+            else:
+                text_value = getattr(text_content, "value", None)
+                if text_value:
+                    text_parts.append(text_value)
     if text_parts:
         return "\n\n".join(text_parts)
     return ""
@@ -223,7 +276,8 @@ def _reset_chat(close_response: bool = True) -> None:
     """Clear conversation and optionally close the response session.
 
     Saves the current conversation to disk, closes the response to clean
-    up resources, and clears the chat history from session state.
+    up resources, and clears the chat history from session state. Also
+    cleans up any temporary files that were created for uploads.
 
     Parameters
     ----------
@@ -234,13 +288,17 @@ def _reset_chat(close_response: bool = True) -> None:
     Notes
     -----
     This function mutates st.session_state in-place, clearing the
-    chat_history and response_instance keys.
+    chat_history, response_instance, and temp_file_paths keys.
     """
     response = st.session_state.get("response_instance")
     if close_response and isinstance(response, BaseResponse):
         filepath = f"./data/{response.name}.{response.uuid}.json"
         response.save(filepath)
         response.close()
+
+    # Clean up temporary files
+    _cleanup_temp_files()
+
     st.session_state["chat_history"] = []
     st.session_state.pop("response_instance", None)
 
@@ -249,7 +307,8 @@ def _init_session_state() -> None:
     """Initialize Streamlit session state for chat functionality.
 
     Creates the chat_history list in session state if it doesn't exist,
-    enabling conversation persistence across Streamlit reruns.
+    enabling conversation persistence across Streamlit reruns. Also
+    initializes a list for tracking temporary file paths that need cleanup.
 
     Notes
     -----
@@ -258,8 +317,12 @@ def _init_session_state() -> None:
     """
     if "chat_history" not in st.session_state:
         st.session_state["chat_history"] = []
-    if "uploaded_files" not in st.session_state:
-        st.session_state["uploaded_files"] = []
+    if "temp_file_paths" not in st.session_state:
+        st.session_state["temp_file_paths"] = []
+    if "current_attachments" not in st.session_state:
+        st.session_state["current_attachments"] = []
+    if "attachment_names" not in st.session_state:
+        st.session_state["attachment_names"] = []
 
 
 def _render_chat_history() -> None:
@@ -293,7 +356,10 @@ def _render_chat_history() -> None:
 
 
 def _handle_user_message(
-    prompt: str, config: StreamlitAppConfig, attachment_paths: list[str] | None = None
+    prompt: str,
+    config: StreamlitAppConfig,
+    attachment_paths: list[str] | None = None,
+    attachment_names: list[str] | None = None,
 ) -> None:
     """Process user input and generate assistant response.
 
@@ -309,6 +375,8 @@ def _handle_user_message(
         Loaded configuration with response handler definition.
     attachment_paths : list[str] or None, default None
         Optional list of file paths to attach to the message.
+    attachment_names : list[str] or None, default None
+        Optional list of original filenames for display purposes.
 
     Notes
     -----
@@ -316,11 +384,15 @@ def _handle_user_message(
     chat transcript rather than crashing the application. The function
     triggers a Streamlit rerun after successful response generation.
     """
-    attachment_names = (
-        [Path(p).name for p in attachment_paths] if attachment_paths else []
+    # Use provided display names or fall back to extracting from paths
+    display_names = (
+        attachment_names
+        if attachment_names
+        else [Path(p).name for p in attachment_paths] if attachment_paths else []
     )
+
     st.session_state["chat_history"].append(
-        {"role": "user", "content": prompt, "attachments": attachment_names}
+        {"role": "user", "content": prompt, "attachments": display_names}
     )
     try:
         response = _get_response_instance(config)
@@ -388,40 +460,63 @@ def main(config_path: Path) -> None:
 
     _render_chat_history()
 
-    # File uploader for attachments
-    uploaded_files = st.file_uploader(
-        "Attach files (optional)",
-        accept_multiple_files=True,
-        key="file_uploader",
-        help=f"Supported formats: {', '.join(sorted(SUPPORTED_FILE_EXTENSIONS))}",
-    )
+    # File uploader form - auto-clears on submit
+    with st.form("file_upload_form", clear_on_submit=True):
+        uploaded_files = st.file_uploader(
+            "Attach files (optional)",
+            accept_multiple_files=True,
+            help=f"Supported formats: {', '.join(sorted(SUPPORTED_FILE_EXTENSIONS))}",
+        )
+        submit_files = st.form_submit_button("Attach files")
 
-    # Save uploaded files to temporary directory and track paths
+    # Process uploaded files if form was submitted
     attachment_paths: list[str] = []
-    if uploaded_files:
+    original_filenames: list[str] = []
+    if submit_files and uploaded_files:
         invalid_files = []
         for uploaded_file in uploaded_files:
             if not _validate_file_type(uploaded_file.name):
                 invalid_files.append(uploaded_file.name)
                 continue
+
+            # Create temporary file with the uploaded content
             with tempfile.NamedTemporaryFile(
                 delete=False, suffix=Path(uploaded_file.name).suffix
             ) as tmp_file:
                 tmp_file.write(uploaded_file.getbuffer())
+                tmp_file.flush()
                 attachment_paths.append(tmp_file.name)
+                original_filenames.append(uploaded_file.name)
+                # Track for cleanup
+                if tmp_file.name not in st.session_state.get("temp_file_paths", []):
+                    st.session_state["temp_file_paths"].append(tmp_file.name)
 
         if invalid_files:
             st.warning(
-                f"⚠️ Unsupported file types skipped: {', '.join(invalid_files)}. "
-                f"Supported formats: {', '.join(sorted(SUPPORTED_FILE_EXTENSIONS))}"
+                f"⚠️ Unsupported file types: {', '.join(invalid_files)}. "
+                f"Supported: {', '.join(sorted(SUPPORTED_FILE_EXTENSIONS))}"
             )
         if attachment_paths:
-            st.caption(f"📎 {len(attachment_paths)} file(s) ready to attach")
+            st.session_state["current_attachments"] = attachment_paths
+            st.session_state["attachment_names"] = original_filenames
+            st.info(f"📎 {len(attachment_paths)} file(s) attached")
+
+    # Get attachment paths from session state if they were previously attached
+    attachment_paths = st.session_state.get("current_attachments", [])
+    attachment_display_names = st.session_state.get("attachment_names", [])
+    if attachment_paths:
+        st.caption(f"Ready to send: {', '.join(attachment_display_names)}")
 
     prompt = st.chat_input("Message the assistant")
     if prompt:
+        # Clear attachments before rerun to prevent them from being sent again
+        st.session_state["current_attachments"] = []
+        st.session_state["attachment_names"] = []
         _handle_user_message(
-            prompt, config, attachment_paths if attachment_paths else None
+            prompt,
+            config,
+            attachment_paths or None,
+            attachment_display_names or None,
         )