Initial commit

Author: MaxMLang

.idea/.gitignore

@@ -0,0 +1,67 @@
+# Lightning ⚡️ AI Style Guide Assistant
+
+<p align="center">
+  <img src="assets/logo.png" alt="AI Style Guide Assistant Logo" width="200" height="200">
+</p>
+
+[![GitHub Stars](https://img.shields.io/github/stars/MaxMLang/lightningAI-styleguide-assistant?style=social)](https://github.com/MaxMLang/lightningAI-styleguide-assistant/stargazers)
+[![GitHub Issues](https://img.shields.io/github/issues/MaxMLang/lightningAI-styleguide-assistant)](https://github.com/MaxMLang/lightningAI-styleguide-assistant/issues)
+[![GitHub Pull Requests](https://img.shields.io/github/issues-pr/MaxMLang/lightningAI-styleguide-assistant)](https://github.com/MaxMLang/lightningAI-styleguide-assistant/pulls)
+[![License](https://img.shields.io/github/license/MaxMLang/lightningAI-styleguide-assistant)](https://github.com/MaxMLang/lightningAI-styleguide-assistant/blob/main/LICENSE)
+
+The Lightning ⚡️ AI Style Guide Assistant is a specialized adaptation of the [Lightning Chatbot](https://github.com/MaxMLang/lightningfast-ai-chat), designed to assist in maintaining consistency and accuracy in style guides. Leveraging Groq LPUs, it provides real-time style and grammar recommendations, making it an essential tool for editors and writers. This AI-powered assistant enhances productivity by automating the enforcement of writing standards and style consistency.
+
+## Features
+
+- ⚡ Real-time style and grammar recommendations powered by Groq LPUs.
+- 📘 Supports custom style guide integration for personalized recommendations.
+- 💬 Interactive editing experience with instant feedback.
+- 🌐 Deploy easily using the Streamlit web framework.
+
+## Installation
+
+1. Clone the repository:
+   ```
+   git clone https://github.com/MaxMLang/lightningAI-styleguide-assistant.git
+   ```
+2. Navigate to the project directory:
+   ```
+   cd lightningAI-styleguide-assistant
+   ```
+3. Install the required dependencies:
+   ```
+   pip install -r requirements.txt
+   ```
+4. Configure the Groq API key:
+   - Create a `.env` file in the project directory.
+   - Add the following line, replacing `your_api_key` with your actual Groq API key:
+     ```
+     GROQ_API_KEY=your_api_key
+     ```
+
+5. Launch the assistant:
+   ```
+   streamlit run app.py
+   ```
+
+6. Open the provided URL in your browser to access the AI Style Guide Assistant interface.
+
+## Usage
+
+1. Select your custom style guide settings from the sidebar.
+2. Type a block of text into the input field and submit.
+3. Receive style and grammar recommendations based on your settings.
+4. Apply or dismiss suggestions and see the edits in real-time.
+5. Continue refining your text with further input and revisions.
+
+## License
+
+This project is licensed under the [MIT License](LICENSE).
+
+## Acknowledgements
+
+- [Groq](https://groq.com) for the cutting-edge LPUs.
+- [Langchain](https://github.com/hwchase17/langchain) for robust language modeling tools.
+- [Streamlit](https://streamlit.io) for the user-friendly web application framework.
+
+

README.md

@@ -0,0 +1,129 @@
+import os
+
+import streamlit as st
+from langchain.chains import ConversationChain
+from langchain.chains.conversation.memory import ConversationBufferWindowMemory
+from langchain_groq import ChatGroq
+import dotenv
+
+dotenv.load_dotenv(dotenv.find_dotenv())
+
+
+def initialize_session_state():
+    """
+    Initialize the session state variables if they don't exist.
+    """
+    if 'chat_history' not in st.session_state:
+        st.session_state.chat_history = []
+    if 'model' not in st.session_state:
+        st.session_state.model = 'mixtral-8x7b-32768'
+    if 'programming_language' not in st.session_state:
+        st.session_state.programming_language = 'Python'
+    if 'style_guide' not in st.session_state:
+        st.session_state.style_guide = 'PEP 8'
+
+
+def display_customization_options():
+    """
+    Add customization options to the sidebar for model selection, memory length, programming language, and style guide.
+    """
+    st.sidebar.title('Customization')
+    model = st.sidebar.selectbox(
+        'Choose a model',
+        ['mixtral-8x7b-32768', 'llama2-70b-4096'],
+        key='model_selectbox'
+    )
+    programming_language = st.sidebar.selectbox(
+        'Programming Language',
+        ['Python', 'R', 'Java', 'JavaScript']
+    )
+
+    style_guide_options = {
+        'Python': ['PEP 8', 'Google Python Style Guide'],
+        'R': ['Google R Style Guide', 'Tidyverse Style Guide'],
+        'Java': ['Google Java Style Guide', 'Oracle Java Code Conventions'],
+        'JavaScript': ['Airbnb JavaScript Style Guide', 'Google JavaScript Style Guide']
+    }
+    style_guide = st.sidebar.selectbox(
+        'Style Guide',
+        style_guide_options[programming_language]
+    )
+
+    return model, programming_language, style_guide
+
+
+def initialize_groq_chat(groq_api_key, model):
+    """
+    Initialize the Groq Langchain chat object.
+    """
+    return ChatGroq(
+        groq_api_key=groq_api_key,
+        model_name=model
+    )
+
+
+def initialize_conversation(groq_chat, memory):
+    """
+    Initialize the conversation chain with the Groq chat object and memory.
+    """
+    return ConversationChain(
+        llm=groq_chat,
+        memory=memory
+    )
+
+
+def process_user_code(user_code, conversation, programming_language, style_guide):
+    """
+    Process the user's code and generate a response using the conversation chain.
+    """
+    prompt = f"Please rewrite the provided {programming_language} code to adhere strictly to the {style_guide} standards. Ensure the output consists solely of the revised code, ready for copy-paste:\n\n{user_code}. Please ensure to that the output only consists of the revised code this is very important!"
+    response = conversation(prompt)
+    message = {'human': prompt, 'AI': response['response']}
+    st.session_state.chat_history.append(message)
+
+
+def main():
+    """
+    The main entry point of the application.
+    """
+    groq_api_key = os.environ['GROQ_API_KEY']
+    initialize_session_state()
+    st.title("Lightning ⚡️ Code Style Guide Assistant")
+    st.markdown("Get your code rewritten according to popular style guides by Lightning, an ultra-fast AI chatbot powered by Groq LPUs!!")
+
+    model, programming_language, style_guide = display_customization_options()
+
+    if st.session_state.model != model or st.session_state.programming_language != programming_language or st.session_state.style_guide != style_guide:
+        # Reset chat history and session state when the model, programming language, or style guide is switched
+        st.session_state.chat_history = []
+        st.session_state.model = model
+        st.session_state.programming_language = programming_language
+        st.session_state.style_guide = style_guide
+        st.rerun()
+
+    memory = ConversationBufferWindowMemory(k=1)
+
+    st.divider()
+
+    user_code = st.text_area("Enter your code:")
+    if user_code:
+        st.session_state.chat_history.append({"human": user_code, "AI": ""})
+
+        with st.expander("Original Code"):
+            st.code(user_code, language=programming_language.lower())
+
+        for message in st.session_state.chat_history:
+            memory.save_context({'input': message['human']}, {'output': message['AI']})
+
+        groq_chat = initialize_groq_chat(groq_api_key, model)
+        conversation = initialize_conversation(groq_chat, memory)
+        process_user_code(user_code, conversation, programming_language, style_guide)
+
+        with st.expander("Rewritten Code"):
+            response = conversation(user_code)
+            st.code(response['response'], language=programming_language.lower())
+            st.session_state.chat_history[-1]["AI"] = response['response']
+
+
+if __name__ == "__main__":
+    main()

app.py

@@ -0,0 +1,7 @@
+langchain~=0.1.12
+langchain-community==0.0.28
+langchain-core==0.1.32
+langchain-openai==0.0.3
+langchain-groq==0.1.2
+langchain-text-splitters==0.0.1
+python-dotenv==1.0.1