UI4AI 0.1.0__tar.gz

ui4ai-0.1.0/PKG-INFO

@@ -0,0 +1,52 @@
+Metadata-Version: 2.2
+Name: UI4AI
+Version: 0.1.0
+Summary: Streamlit UI for LLM chat apps
+Home-page: https://github.com/DKethan/UI4AI/tree/dev-01
+Author: Kethan Dosapati
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: streamlit
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# UI4AI
+
+A lightweight, plug-and-play Streamlit-based UI for LLM chatbot applications.
+
+## Features
+- Add your own `generate_response` function
+- Sidebar history, session management
+- Optional: title generation, token counting
+
+## Usage
+
+```python
+from ui4ai import run_chat
+
+run_chat(generate_response=my_response_function)
+```
+
+## Install
+
+```
+pip install ui4ai
+```
+```
+
+---
+
+### ✅ Next Step?
+- Want me to generate the license file?
+- Ready to build and upload to PyPI?
+- Want me to zip this entire structure for you?
+
+Let’s roll!

ui4ai-0.1.0/README.md

@@ -0,0 +1,32 @@
+# UI4AI
+
+A lightweight, plug-and-play Streamlit-based UI for LLM chatbot applications.
+
+## Features
+- Add your own `generate_response` function
+- Sidebar history, session management
+- Optional: title generation, token counting
+
+## Usage
+
+```python
+from ui4ai import run_chat
+
+run_chat(generate_response=my_response_function)
+```
+
+## Install
+
+```
+pip install ui4ai
+```
+```
+
+---
+
+### ✅ Next Step?
+- Want me to generate the license file?
+- Ready to build and upload to PyPI?
+- Want me to zip this entire structure for you?
+
+Let’s roll!

ui4ai-0.1.0/UI4AI/__init__.py

@@ -0,0 +1 @@
+from .chat_ui import run_chat

ui4ai-0.1.0/UI4AI/chat_ui.py

@@ -0,0 +1,158 @@
+import streamlit as st
+import uuid
+from datetime import datetime
+from typing import List, Dict, Callable, Optional
+
+
+def run_chat(
+    generate_response: Optional[Callable[[List[Dict]], str]],
+    generate_title: Optional[Callable[[str], str]] = None,
+    count_tokens: Optional[Callable[[List[Dict]], int]] = None,
+    page_title: str = "AI Chat",
+    title: str = "Conversational bot",
+    layout: str = "wide",
+    new_conversation: str = "➕ New Chat",
+    chat_placeholder: str = "Ask me anything...",
+    sidebar_instructions: str = "Conversation History",
+    spinner_text: str = "Thinking...",
+    max_history_tokens: Optional[int] = None
+):
+    """Streamlit UI for LLM chat with flexible core"""
+
+    if not generate_response:
+        print("No generate_response function provided.")
+        st.set_page_config(page_title="Error", layout="wide")
+        st.error("No `generate_response` function provided.")
+        return
+
+    _init_session_state()
+
+    st.set_page_config(page_title=page_title, layout=layout)
+    st.title(title)
+
+    with st.sidebar:
+        _render_sidebar(generate_title, count_tokens, sidebar_instructions, new_conversation)
+
+    _render_chat_history()
+
+    _handle_user_input(
+        generate_response,
+        generate_title,
+        count_tokens,
+        chat_placeholder,
+        spinner_text,
+        max_history_tokens
+    )
+
+
+def _init_session_state():
+    defaults = {"conversations": {}, "current_convo_id": None, "messages": []}
+    for key, val in defaults.items():
+        if key not in st.session_state:
+            st.session_state[key] = val
+
+
+def _render_sidebar(
+    generate_title: Optional[Callable],
+    count_tokens: Optional[Callable],
+    instructions: str,
+    new_conversation: str
+):
+    st.markdown("### 📖 Instructions")
+    st.markdown(instructions)
+
+    if st.button(new_conversation):
+        print("🔄 New conversation started")
+        _reset_conversation()
+
+    if generate_title:
+        for convo_id, convo in st.session_state.conversations.items():
+            label = convo["title"]
+            if count_tokens:
+                label += f" ({convo.get('token_count', '?')} tokens)"
+            if st.button(label, key=convo_id):
+                st.session_state.current_convo_id = convo_id
+                st.session_state.messages = convo["messages"]
+                st.rerun()
+
+
+def _render_chat_history():
+    for msg in st.session_state.messages:
+        with st.chat_message(msg["role"]):
+            st.markdown(msg["content"])
+
+
+def _handle_user_input(
+    generate_response: Callable,
+    generate_title: Optional[Callable],
+    count_tokens: Optional[Callable],
+    placeholder: str,
+    spinner_text: str,
+    max_tokens: Optional[int]
+):
+    if prompt := st.chat_input(placeholder):
+        _create_conversation_if_needed(prompt, generate_title)
+
+        st.chat_message("user").markdown(prompt)
+        st.session_state.messages.append({"role": "user", "content": prompt})
+
+        try:
+            with st.spinner(spinner_text):
+                if count_tokens and max_tokens:
+                    messages_for_api = _truncate_messages(
+                        st.session_state.messages,
+                        count_tokens,
+                        max_tokens
+                    )
+                elif generate_title or count_tokens:
+                    messages_for_api = st.session_state.messages
+                else:
+                    messages_for_api = [st.session_state.messages[-1]]
+
+                response = generate_response(messages_for_api)
+
+                st.chat_message("assistant").markdown(response)
+                st.session_state.messages.append({"role": "assistant", "content": response})
+
+                if st.session_state.current_convo_id:
+                    convo = st.session_state.conversations[st.session_state.current_convo_id]
+                    convo["messages"] = st.session_state.messages
+                    if count_tokens:
+                        convo["token_count"] = count_tokens(st.session_state.messages)
+
+        except Exception as e:
+            print(f"Exception: {e}")
+            st.error(f"Error: {str(e)}")
+
+
+def _create_conversation_if_needed(prompt: str, generate_title: Optional[Callable]):
+    if not st.session_state.current_convo_id:
+        convo_id = str(uuid.uuid4())
+        title = generate_title(prompt) if generate_title else "Untitled Chat"
+        st.session_state.conversations[convo_id] = {
+            "id": convo_id,
+            "title": title,
+            "messages": [],
+            "token_count": 0,
+            "created_at": datetime.now().isoformat()
+        }
+        st.session_state.current_convo_id = convo_id
+
+
+def _reset_conversation():
+    st.session_state.current_convo_id = None
+    st.session_state.messages = []
+
+
+def _truncate_messages(messages: List[Dict], count_tokens: Callable, max_tokens: int) -> List[Dict]:
+    trimmed = []
+    total_tokens = 0
+
+    for msg in reversed(messages):
+        tokens = count_tokens([msg])
+        if total_tokens + tokens > max_tokens:
+            break
+        trimmed.insert(0, msg)
+        total_tokens += tokens
+
+    return trimmed

ui4ai-0.1.0/UI4AI.egg-info/PKG-INFO

@@ -0,0 +1,52 @@
+Metadata-Version: 2.2
+Name: UI4AI
+Version: 0.1.0
+Summary: Streamlit UI for LLM chat apps
+Home-page: https://github.com/DKethan/UI4AI/tree/dev-01
+Author: Kethan Dosapati
+Classifier: Programming Language :: Python :: 3
+Classifier: Operating System :: OS Independent
+Requires-Python: >=3.7
+Description-Content-Type: text/markdown
+Requires-Dist: streamlit
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# UI4AI
+
+A lightweight, plug-and-play Streamlit-based UI for LLM chatbot applications.
+
+## Features
+- Add your own `generate_response` function
+- Sidebar history, session management
+- Optional: title generation, token counting
+
+## Usage
+
+```python
+from ui4ai import run_chat
+
+run_chat(generate_response=my_response_function)
+```
+
+## Install
+
+```
+pip install ui4ai
+```
+```
+
+---
+
+### ✅ Next Step?
+- Want me to generate the license file?
+- Ready to build and upload to PyPI?
+- Want me to zip this entire structure for you?
+
+Let’s roll!

ui4ai-0.1.0/UI4AI.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+README.md
+setup.py
+UI4AI/__init__.py
+UI4AI/chat_ui.py
+UI4AI.egg-info/PKG-INFO
+UI4AI.egg-info/SOURCES.txt
+UI4AI.egg-info/dependency_links.txt
+UI4AI.egg-info/requires.txt
+UI4AI.egg-info/top_level.txt
+tests/test_apr_10_441.py

ui4ai-0.1.0/UI4AI.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

ui4ai-0.1.0/UI4AI.egg-info/requires.txt

@@ -0,0 +1 @@
+streamlit

ui4ai-0.1.0/UI4AI.egg-info/top_level.txt

@@ -0,0 +1 @@
+UI4AI

ui4ai-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

ui4ai-0.1.0/setup.py

@@ -0,0 +1,18 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="UI4AI",
+    version="0.1.0",
+    author="Kethan Dosapati",
+    description="Streamlit UI for LLM chat apps",
+    long_description=open("README.md").read(),
+    long_description_content_type="text/markdown",
+    url="https://github.com/DKethan/UI4AI/tree/dev-01",
+    packages=find_packages(),
+    install_requires=["streamlit"],
+    classifiers=[
+        "Programming Language :: Python :: 3",
+        "Operating System :: OS Independent",
+    ],
+    python_requires='>=3.7',
+)

ui4ai-0.1.0/tests/test_apr_10_441.py

@@ -0,0 +1,24 @@
+from UI4AI import run_chat
+import openai
+
+openai.api_key = "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"
+
+def generate_response(messages) -> str:
+    """Generate response with history management"""
+    try:
+        response = openai.chat.completions.create(
+            model="gpt-4",
+            messages=messages,
+            temperature=0.7
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        raise RuntimeError(f"Response generation failed: {str(e)}")
+
+run_chat(
+    generate_response=generate_response,
+    page_title="GPT-4 Chat",
+    chat_placeholder="Ask me anything...",
+    sidebar_instructions="Powered by GPT-4",
+    spinner_text="Generating response...",
+)