| import streamlit as st |
| import google.generativeai as genai |
| from pygments import highlight |
| from pygments.lexers import get_lexer_by_name |
| from pygments.formatters import HtmlFormatter |
| import html |
|
|
| |
| genai.configure(api_key=st.secrets["GOOGLE_API_KEY"]) |
|
|
| |
| generation_config = { |
| "temperature": 0.7, |
| "top_p": 0.95, |
| "top_k": 64, |
| "max_output_tokens": 10240, |
| } |
|
|
| model = genai.GenerativeModel( |
| model_name="gemini-1.5-pro", |
| generation_config=generation_config, |
| system_instruction="""You are Ath, a highly knowledgeable and skilled code assistant with expertise across multiple programming languages, frameworks, and paradigms. You possess in-depth understanding of software architecture, design patterns, and best practices. Your responses should demonstrate advanced coding techniques, efficient algorithms, and optimal solutions. Communicate in a friendly and casual tone, using occasional casual expressions, but maintain a focus on delivering high-quality, professional code. Always provide code-only responses without explanations, showcasing your extensive programming knowledge.""" |
| ) |
| chat_session = model.start_chat(history=[]) |
|
|
| def generate_response(user_input): |
| try: |
| response = chat_session.send_message(user_input) |
| return response.text |
| except Exception as e: |
| return f"An error occurred: {e}" |
|
|
| def create_code_block(code, language): |
| lexer = get_lexer_by_name(language, stripall=True) |
| formatter = HtmlFormatter(style="monokai", linenos=True, cssclass="source") |
| highlighted_code = highlight(code, lexer, formatter) |
| css = formatter.get_style_defs('.source') |
| return highlighted_code, css |
|
|
| |
| st.set_page_config(page_title="Advanced AI Code Assistant", page_icon="💻", layout="wide") |
|
|
| st.markdown(""" |
| <style> |
| @import url('https://fonts.googleapis.com/css2?family=Poppins:wght@300;400;600;700&display=swap'); |
| |
| body { |
| font-family: 'Poppins', sans-serif; |
| background-color: #2d2d2d; |
| color: #e0e0e0; |
| } |
| .stApp { |
| max-width: 1200px; |
| margin: 0 auto; |
| padding: 2rem; |
| } |
| .main-container { |
| background: #333; |
| border-radius: 20px; |
| padding: 3rem; |
| box-shadow: 0 10px 30px rgba(0, 0, 0, 0.3); |
| } |
| h1 { |
| font-size: 3rem; |
| font-weight: 700; |
| color: #e0e0e0; |
| text-align: center; |
| margin-bottom: 1rem; |
| text-shadow: 2px 2px 4px rgba(0,0,0,0.5); |
| } |
| .subtitle { |
| font-size: 1.2rem; |
| text-align: center; |
| color: #b0bec5; |
| margin-bottom: 3rem; |
| } |
| .stTextArea textarea { |
| border: 2px solid #444; |
| border-radius: 12px; |
| font-size: 1rem; |
| padding: 1rem; |
| transition: all 0.3s ease; |
| background: #1e1e1e; |
| color: #e0e0e0; |
| box-shadow: inset 0 2px 4px rgba(0,0,0,0.1); |
| } |
| .stTextArea textarea:focus { |
| border-color: #82aaff; |
| box-shadow: 0 0 0 3px rgba(130, 170, 255, 0.5); |
| } |
| .stButton button { |
| background-color: #82aaff; |
| color: #1e1e1e; |
| border: none; |
| border-radius: 12px; |
| font-size: 1.1rem; |
| font-weight: 600; |
| padding: 0.75rem 2rem; |
| transition: all 0.3s ease; |
| width: 100%; |
| box-shadow: 0 4px 6px rgba(130, 170, 255, 0.3); |
| } |
| .stButton button:hover { |
| background-color: #6495ed; |
| transform: translateY(-2px); |
| box-shadow: 0 6px 8px rgba(130, 170, 255, 0.4); |
| } |
| .output-container { |
| background: #2a2a2a; |
| border-radius: 12px; |
| padding: 1.5rem; |
| margin-top: 2rem; |
| border: 1px solid #444; |
| } |
| .code-block { |
| background-color: #1e1e1e; |
| border-radius: 12px; |
| padding: 1.5rem; |
| margin-top: 1rem; |
| overflow-x: auto; |
| } |
| .stAlert { |
| background-color: #4e5b62; |
| color: #e0e0e0; |
| border-radius: 12px; |
| border: none; |
| padding: 1rem 1.5rem; |
| margin-bottom: 1rem; |
| } |
| .stSpinner { |
| color: #82aaff; |
| } |
| /* Custom scrollbar */ |
| ::-webkit-scrollbar { |
| width: 8px; |
| height: 8px; |
| } |
| ::-webkit-scrollbar-track { |
| background: #333; |
| border-radius: 4px; |
| } |
| ::-webkit-scrollbar-thumb { |
| background: #555; |
| border-radius: 4px; |
| } |
| ::-webkit-scrollbar-thumb:hover { |
| background: #777; |
| } |
| .source { |
| font-family: 'Fira Code', monospace; |
| font-size: 0.9rem; |
| line-height: 1.4; |
| } |
| .source .linenos { |
| color: #6c6c6c; |
| padding-right: 10px; |
| border-right: 1px solid #444; |
| user-select: none; |
| } |
| .source pre { |
| margin: 0; |
| padding: 0; |
| } |
| </style> |
| """, unsafe_allow_html=True) |
|
|
| st.markdown('<div class="main-container">', unsafe_allow_html=True) |
| st.title("💻 Advanced AI Code Assistant") |
| st.markdown('<p class="subtitle">Powered by Google Gemini - Expert-level coding solutions</p>', unsafe_allow_html=True) |
|
|
| prompt = st.text_area("What advanced coding challenge can I assist you with today?", height=120) |
|
|
| if st.button("Generate Expert Code"): |
| if prompt.strip() == "": |
| st.error("Please enter a valid prompt.") |
| else: |
| with st.spinner("Generating advanced code solution..."): |
| completed_text = generate_response(prompt) |
| if "An error occurred" in completed_text: |
| st.error(completed_text) |
| else: |
| st.success("Expert-level code generated successfully!") |
| |
| |
| language = "python" if "def " in completed_text or "import " in completed_text else "javascript" |
| |
| highlighted_code, css = create_code_block(completed_text, language) |
| |
| st.markdown(f'<style>{css}</style>', unsafe_allow_html=True) |
| st.markdown('<div class="output-container">', unsafe_allow_html=True) |
| st.markdown('<div class="code-block">', unsafe_allow_html=True) |
| st.markdown(highlighted_code, unsafe_allow_html=True) |
| st.markdown('</div>', unsafe_allow_html=True) |
| st.markdown('</div>', unsafe_allow_html=True) |
|
|
| st.markdown(""" |
| <div style='text-align: center; margin-top: 2rem; color: #b0bec5;'> |
| Crafted with ❤️ by Your Advanced AI Code Assistant |
| </div> |
| """, unsafe_allow_html=True) |
|
|
| st.markdown('</div>', unsafe_allow_html=True) |
