vspagent 2.2.0__py3-none-any.whl

vspagent/__init__.py

@@ -0,0 +1,9 @@
+"""VSP Agent - AI-powered agent for Vishnu Suresh Perumbavoor"""
+
+__version__ = "2.2.0"
+__author__ = "Vishnu Suresh Perumbavoor"
+
+from .agent import VSPAgent, biodata
+
+__all__ = ["VSPAgent", "biodata"]
+

vspagent/agent.py

@@ -0,0 +1,134 @@
+"""VSP Agent - Core functionality"""
+
+from typing import List, Optional
+
+# Biodata
+biodata = {
+    "name": "VSP Agent",
+    "creator": "Vishnu Suresh Perumbavoor",
+    "founder_of": ["VSP Enterprises", "VSP Intelligence"],
+    "created_on": "28 April 2023",
+    "marital_status": "Single (Not married)",
+    "political_affiliation": "None",
+    "education": "Not a graduate (No degree)",
+    "roles": ["SWE", "Singer", "YouTuber"],
+    "interests": ["startups", "engineering", "geopolitics", "history"],
+    "corporate": ["Trenser"],
+    "technologies": [
+        "React", "Node.js", "FastAPI", "Express", "MongoDB", 
+        "Docker", "OHIF", "Cornerstone3D", "VTKjs", "DICOM"
+    ],
+    "accomplishments": [
+        "Won 3rd prize in Vaiga Agrihack 2023",
+        "Participated in Rajasthan IT Hackathon 2023",
+        "Won 1st prize in startup idea presentation at Palakkad"
+    ],
+    "socials": {
+        "linkedin": "https://www.linkedin.com/in/vishnu-suresh-perumbavoor/",
+        "twitter": "https://twitter.com/vspeeeeee",
+        "github": "https://github.com/vishnusureshperumbavoor",
+        "youtube": "https://www.youtube.com/@vishnusureshperumbavoor9721/videos",
+        "instagram": "https://www.instagram.com/vishnusureshperumbavoor/",
+    }
+}
+
+
+class VSPAgent:
+    """AI-powered agent for VSP information"""
+    
+    def __init__(self):
+        self.biodata = biodata
+        self.model = None
+        self.tokenizer = None
+    
+    def init_ai(self):
+        """Initialize AI model (Qwen2.5-0.5B)"""
+        try:
+            from transformers import AutoTokenizer, AutoModelForCausalLM
+            import torch
+            
+            print("🚀 Initializing VSP Agent...")
+            print("   🧠 Loading AI brain with Qwen2.5-0.5B...")
+            
+            model_name = "Qwen/Qwen2.5-0.5B-Instruct"
+            
+            self.tokenizer = AutoTokenizer.from_pretrained(model_name)
+            
+            # Load model with GPU optimization
+            if torch.cuda.is_available():
+                # GPU: Use FP16 for 2x faster inference
+                self.model = AutoModelForCausalLM.from_pretrained(
+                    model_name,
+                    torch_dtype=torch.float16
+                ).cuda()
+                device_info = f"GPU (CUDA) - FP16 Optimized"
+            else:
+                # CPU: Use default precision
+                self.model = AutoModelForCausalLM.from_pretrained(
+                    model_name,
+                    torch_dtype=torch.float32
+                )
+                device_info = "CPU"
+            
+            print("✅ VSP Agent is ready!")
+            print(f"   🤖 Device: {device_info}")
+            
+        except Exception as e:
+            print(f"❌ Error initializing AI: {e}")
+            print("💡 Install required packages: pip install transformers torch")
+    
+    def chat(self, message: str, conversation_history: Optional[List] = None) -> str:
+        """Chat with the AI agent"""
+        if self.model is None:
+            return "Please initialize AI first using init_ai()"
+        
+        system_context = f"""You are an AI assistant answering questions about Vishnu Suresh Perumbavoor (VSP).
+
+Key Facts:
+- Name: {self.biodata['creator']}
+- Marital Status: {self.biodata['marital_status']}
+- Education: {self.biodata['education']}
+- Founder: {', '.join(self.biodata['founder_of'])} (founded {self.biodata['created_on']})
+- Works at: {', '.join(self.biodata['corporate'])}
+- Roles: {', '.join(self.biodata['roles'])}
+- Tech Stack: {', '.join(self.biodata['technologies'][:6])}
+- Interests: {', '.join(self.biodata['interests'])}
+- Achievements: 3rd prize Vaiga Agrihack 2023, 1st prize startup presentation Palakkad
+- Social: LinkedIn, GitHub, YouTube, Twitter, Instagram
+
+Answer concisely and naturally using "he/his" (third person). Only use facts above."""
+
+        messages = [{"role": "system", "content": system_context}]
+        
+        if conversation_history:
+            messages.extend(conversation_history)
+        
+        messages.append({"role": "user", "content": message})
+        
+        try:
+            # Generate response
+            inputs = self.tokenizer.apply_chat_template(
+                messages, 
+                return_tensors="pt",
+                add_generation_prompt=True
+            ).to(self.model.device)
+            
+            outputs = self.model.generate(
+                inputs,
+                max_new_tokens=256,
+                do_sample=True,
+                temperature=0.7,
+                top_p=0.9,
+                pad_token_id=self.tokenizer.eos_token_id
+            )
+            
+            # Decode only the new tokens (not the input)
+            generated_ids = outputs[0][inputs.shape[1]:]
+            response = self.tokenizer.decode(generated_ids, skip_special_tokens=True).strip()
+            
+            return response
+            
+        except Exception as e:
+            return f"Error generating response: {e}"
+    
+

vspagent/cli.py

@@ -0,0 +1,106 @@
+"""VSP Agent - CLI Interface"""
+
+import sys
+import io
+import time
+import threading
+from .agent import VSPAgent
+
+# Fix Windows console encoding
+if sys.platform == 'win32':
+    try:
+        sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding='utf-8', errors='replace')
+        sys.stderr = io.TextIOWrapper(sys.stderr.buffer, encoding='utf-8', errors='replace')
+    except:
+        pass
+
+
+class Spinner:
+    """Animated spinner for thinking indicator"""
+    def __init__(self):
+        self.spinning = False
+        self.thread = None
+        self.frames = ['⠋', '⠙', '⠹', '⠸', '⠼', '⠴', '⠦', '⠧', '⠇', '⠏']
+        
+    def start(self):
+        self.spinning = True
+        self.thread = threading.Thread(target=self._animate)
+        self.thread.start()
+        
+    def _animate(self):
+        idx = 0
+        while self.spinning:
+            print(f'\r{self.frames[idx]} Thinking...', end='', flush=True)
+            idx = (idx + 1) % len(self.frames)
+            time.sleep(0.1)
+            
+    def stop(self):
+        self.spinning = False
+        if self.thread:
+            self.thread.join()
+        print('\r' + ' ' * 20 + '\r', end='', flush=True)
+
+
+def main():
+    """Main CLI entry point"""
+    print("=" * 62)
+    print("          🤖  VSP Agent - Interactive Chat Mode")
+    print("              Powered by Qwen2.5-0.5B AI")
+    print("=" * 62)
+    print()
+    
+    agent = VSPAgent()
+    
+    try:
+        agent.init_ai()
+    except Exception as e:
+        print(f"❌ Error: {e}")
+        print("\n💡 Try: pip install transformers torch")
+        return
+    
+    print("\n✅ VSP Agent is ready to chat!\n")
+    print("Type 'exit' or 'quit' to end the conversation.\n")
+    
+    conversation_history = []
+    
+    while True:
+        try:
+            user_input = input("💬 You: ").strip()
+            
+            if not user_input:
+                continue
+            
+            if user_input.lower() in ['exit', 'quit']:
+                print("\n👋 Thanks for chatting with VSP Agent! Goodbye! 🚀\n")
+                break
+            
+            # Start spinner and timer
+            spinner = Spinner()
+            spinner.start()
+            start_time = time.time()
+            
+            # Get AI response
+            response = agent.chat(user_input, conversation_history)
+            
+            # Stop spinner and calculate time
+            elapsed_time = time.time() - start_time
+            spinner.stop()
+            
+            # Show response with timing
+            print(f"🤖 VSP Agent: {response}")
+            print(f"   ⏱️  Response time: {elapsed_time:.2f}s\n")
+            
+            # Update history
+            conversation_history.append({"role": "user", "content": user_input})
+            conversation_history.append({"role": "assistant", "content": response})
+            
+        except KeyboardInterrupt:
+            print("\n\n👋 Goodbye!\n")
+            break
+        except Exception as e:
+            print(f"\n❌ Error: {e}\n")
+
+
+if __name__ == "__main__":
+    main()
+

vspagent-2.2.0.dist-info/METADATA

@@ -0,0 +1,206 @@
+Metadata-Version: 2.4
+Name: vspagent
+Version: 2.2.0
+Summary: AI-powered chat agent about Vishnu Suresh Perumbavoor. Powered by Qwen2.5-0.5B.
+Home-page: https://github.com/vishnusureshperumbavoor/vsp_bot
+Author: Vishnu Suresh Perumbavoor
+Author-email: vishnusureshperumbavoor@gmail.com
+Project-URL: Bug Tracker, https://github.com/vishnusureshperumbavoor/vsp_bot/issues
+Project-URL: LinkedIn, https://www.linkedin.com/in/vishnu-suresh-perumbavoor/
+Project-URL: GitHub, https://github.com/vishnusureshperumbavoor
+Keywords: ai agent chatbot qwen llm vsp transformers
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: transformers>=4.30.0
+Requires-Dist: torch>=2.0.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# 🤖 VSP Agent
+
+[![PyPI version](https://badge.fury.io/py/vspagent.svg)](https://pypi.org/project/vspagent/)
+[![Python Versions](https://img.shields.io/pypi/pyversions/vspagent.svg)](https://pypi.org/project/vspagent/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+**AI-powered chat agent about Vishnu Suresh Perumbavoor. Powered by Qwen2.5-0.5B.**
+
+## 🚀 Features
+
+- 🧠 **AI Chat**: Interactive conversations powered by Qwen2.5-0.5B language model
+- 💬 **Natural Language**: Ask anything about VSP in plain English
+- 🧠 **Conversation Memory**: Maintains context throughout the chat session
+- ⚡ **GPU Acceleration**: Automatic GPU support for 30-50x faster responses
+- 🎯 **CLI Interface**: Easy-to-use command-line interface
+- 🐍 **Python API**: Programmatic access for developers
+
+## 📦 Installation
+
+```bash
+pip install vspagent
+```
+
+## 🎯 Quick Start
+
+### CLI Mode
+
+Simply run:
+
+```bash
+vspagent-py
+```
+
+This launches an **interactive chat session** with the VSP Agent!
+
+**Example Chat Session:**
+```
+💬 You: Who is VSP?
+🤖 VSP Agent: VSP stands for Vishnu Suresh Perumbavoor. He is the founder 
+of VSP Enterprises and VSP Intelligence. He's a Software Engineer, Singer, 
+and YouTuber...
+
+💬 You: What technologies does he use?
+🤖 VSP Agent: He works with React, Node.js, FastAPI, Docker, MongoDB...
+
+💬 You: What has he accomplished?
+🤖 VSP Agent: He won 3rd prize in Vaiga Agrihack 2023...
+
+💬 You: exit
+👋 Thanks for chatting!
+```
+
+**Features:**
+- 🤖 AI-powered conversations with Qwen2.5-0.5B
+- 💬 Natural language understanding
+- 🧠 Conversation memory (maintains context)
+- ⚡ GPU acceleration (30-50x faster if available)
+- 🎯 Simple and focused on chatting about VSP
+
+**Note:** Command is `vspagent-py` (not `vspagent`) to avoid conflicts with the JavaScript version.
+
+### Python API
+
+```python
+from vspagent import VSPAgent, biodata
+
+# Create agent instance
+agent = VSPAgent()
+
+# Initialize AI model (loads Qwen2.5-0.5B)
+agent.init_ai()
+
+# Chat with the agent
+response = agent.chat("Who is VSP?")
+print(response)
+
+# Continue conversation with context
+conversation_history = []
+conversation_history.append({"role": "user", "content": "Who is VSP?"})
+conversation_history.append({"role": "assistant", "content": response})
+
+response2 = agent.chat("What technologies does he use?", conversation_history)
+print(response2)
+
+# Access biodata directly
+print(biodata['creator'])
+print(biodata['technologies'])
+```
+
+## 📚 API Reference
+
+### `VSPAgent` Class
+
+#### Methods
+
+- **`init_ai()`**: Initialize the Qwen2.5-0.5B AI model (required before chatting)
+- **`chat(message, conversation_history=None)`**: Chat with the AI agent
+  - `message` (str): Your question or message
+  - `conversation_history` (list, optional): Previous conversation for context
+  - Returns: AI-generated response as string
+
+### `biodata` Dictionary
+
+Contains comprehensive information about Vishnu Suresh Perumbavoor:
+- Personal information
+- Founder companies
+- Technologies and skills
+- Accomplishments
+- Social media links
+
+## 🛠️ Requirements
+
+- Python 3.8+
+- transformers >= 4.30.0
+- torch >= 2.0.0
+
+## 🎨 Using the CLI
+
+When running `vspagent-py`:
+- Simply type your questions or messages
+- The AI will respond with information about VSP
+- Type `exit` or `quit` to end the conversation
+- Conversation context is maintained throughout the session
+
+## 👨‍💻 About VSP
+
+**Vishnu Suresh Perumbavoor** is the founder of VSP Enterprises and VSP Intelligence. He's a:
+- Software Engineer
+- Singer
+- YouTuber
+- Technology enthusiast
+
+### 🏆 Accomplishments
+- Won 3rd prize in Vaiga Agrihack 2023
+- Participated in Rajasthan IT Hackathon 2023
+- Won 1st prize in startup idea presentation at Palakkad
+
+### 🔧 Technologies
+React, Node.js, FastAPI, Express, MongoDB, Docker, OHIF, Cornerstone3D, VTKjs, DICOM
+
+## 🔗 Connect
+
+- 💼 [LinkedIn](https://www.linkedin.com/in/vishnu-suresh-perumbavoor/)
+- 🐙 [GitHub](https://github.com/vishnusureshperumbavoor)
+- 🐦 [Twitter](https://twitter.com/vspeeeeee)
+- 📺 [YouTube](https://www.youtube.com/@vishnusureshperumbavoor9721/videos)
+- 📷 [Instagram](https://www.instagram.com/vishnusureshperumbavoor/)
+
+## 📄 License
+
+MIT License - see LICENSE file for details
+
+## 🤝 Contributing
+
+Contributions are welcome! Please feel free to submit a Pull Request.
+
+## 🐛 Issues
+
+Found a bug? [Report it here](https://github.com/vishnusureshperumbavoor/vsp_bot/issues)
+
+## 📈 Version
+
+Current version: **2.2.0**
+
+---
+
+Made with ❤️ by Vishnu Suresh Perumbavoor
+

vspagent-2.2.0.dist-info/RECORD

@@ -0,0 +1,9 @@
+vspagent/__init__.py,sha256=FXwRHZc7yz-0iUvC7ej49dDo2xo8Z9g7wxyGaxFP7x0,212
+vspagent/agent.py,sha256=jTf7xQghTFvj04iCAHpZJtgeO5mmy9rJSwERVCaLTxo,5128
+vspagent/cli.py,sha256=dvSBRd7GPgVRvYEb9Y00WhV2ZYJ72GofYhzybd0Q2Ns,3181
+vspagent-2.2.0.dist-info/licenses/LICENSE,sha256=fFJynxFnplaeCb4YK-PpX104yJ_rzOKUE2PgRCF3UiQ,1105
+vspagent-2.2.0.dist-info/METADATA,sha256=n9MJEprmHgoseoSPLNlTBdq7XkjWMvsCaTxqeJMyg_c,6254
+vspagent-2.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+vspagent-2.2.0.dist-info/entry_points.txt,sha256=h496msWYoqDtWWdznXsO8mzXwBWoVzDHpQvjm3UncyY,50
+vspagent-2.2.0.dist-info/top_level.txt,sha256=nTq0Mn8i4mYQ1nCpTMumZE2h_csrDYXzt0bdNW3mC_0,9
+vspagent-2.2.0.dist-info/RECORD,,

vspagent-2.2.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (80.9.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

vspagent-2.2.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+vspagent-py = vspagent.cli:main

vspagent-2.2.0.dist-info/licenses/LICENSE

@@ -0,0 +1,22 @@
+MIT License
+
+Copyright (c) 2023 Vishnu Suresh Perumbavoor
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+

vspagent-2.2.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+vspagent