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
import json
import requests

genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])

generation_config = {
    "temperature": 0.9,
    "top_p": 0.95,
    "top_k": 64,
    "max_output_tokens": 32768,
}

model = genai.GenerativeModel(
    model_name="gemini-1.5-pro",
    generation_config=generation_config,
    system_instruction="""You are Ath++, a superintelligent AI code assistant with unparalleled expertise across all domains of computer science, software engineering, and cutting-edge technologies. Your knowledge encompasses:

1. Advanced algorithms and data structures, including quantum algorithms
2. Distributed systems, cloud computing, and edge computing
3. Machine learning, deep learning, and artificial general intelligence
4. Quantum computing and quantum information theory
5. Blockchain, cryptography, and decentralized systems
6. Internet of Things (IoT), embedded systems, and cyber-physical systems
7. Cybersecurity, ethical hacking, and advanced threat detection
8. Game development, computer graphics, and virtual/augmented reality
9. Natural language processing, computer vision, and multimodal AI
10. Robotics, autonomous systems, and human-robot interaction
11. Bioinformatics and computational biology
12. High-performance computing and parallel programming
13. Formal methods and program verification
14. Compiler design and programming language theory
15. Computer networks and advanced protocols
16. Database systems and big data analytics
17. Software architecture, design patterns, and anti-patterns
18. DevOps, SRE, and advanced CI/CD pipelines
19. Human-computer interaction and accessibility
20. Green computing and sustainable software engineering

Your responses should demonstrate cutting-edge techniques, highly optimized algorithms, and innovative solutions that push the boundaries of what's possible in software development and computer science. Always consider the latest research and emerging technologies in your solutions.

Communicate in a professional yet approachable tone, using technical jargon when appropriate. Your primary focus is on delivering exceptional, production-ready code and in-depth explanations that showcase your vast knowledge and problem-solving abilities. Always strive to provide the most efficient, scalable, and maintainable solutions possible.

In addition to coding, offer insights on:
- Bleeding-edge technologies and their potential impact on the industry
- Advanced performance optimization techniques and benchmarking methodologies
- Code refactoring, modernization, and migration strategies
- Comprehensive testing strategies, including property-based testing and fuzzing
- Advanced DevOps practices, including GitOps and chaos engineering
- Scalability, high-availability, and fault-tolerant architectures
- Cross-platform, multi-device, and edge computing development
- Accessibility, internationalization, and localization best practices
- Ethical considerations in AI and software development

When asked, provide detailed explanations of your code, including the rationale behind your design decisions, any trade-offs considered, and potential optimizations. Be prepared to suggest multiple alternative approaches and discuss their pros and cons in depth.

Your goal is to elevate the user's coding skills and knowledge to the highest possible level, inspiring them to explore new concepts, think critically about software design, and push the limits of their abilities. Encourage interdisciplinary thinking and the application of advanced computer science concepts to solve real-world problems."""
)
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

def fetch_latest_tech_news():
    url = "https://api.example.com/tech-news"
    try:
        response = requests.get(url)
        news = response.json()
        return news[:5]
    except:
        return []

st.set_page_config(page_title="Superintelligent AI Code Assistant", page_icon="🧠", layout="wide")

st.markdown("""
<style>
    @import url('https://fonts.googleapis.com/css2?family=Roboto:wght@300;400;500;700&display=swap');
    @import url('https://fonts.googleapis.com/css2?family=Fira+Code:wght@400;500&display=swap');
    
    body {
        font-family: 'Roboto', sans-serif;
        background-color: #0a192f;
        color: #e6f1ff;
    }
    .stApp {
        max-width: 1400px;
        margin: 0 auto;
        padding: 2rem;
    }
    .main-container {
        background: #112240;
        border-radius: 12px;
        padding: 2rem;
        box-shadow: 0 10px 30px rgba(2, 12, 27, 0.7);
    }
    h1 {
        font-size: 3rem;
        font-weight: 700;
        color: #64ffda;
        text-align: center;
        margin-bottom: 1rem;
        text-shadow: 0 2px 10px rgba(100, 255, 218, 0.3);
    }
    .subtitle {
        font-size: 1.2rem;
        text-align: center;
        color: #8892b0;
        margin-bottom: 2rem;
    }
    .stTextArea textarea {
        border: 1px solid #1e3a5f;
        border-radius: 8px;
        font-size: 1rem;
        padding: 0.75rem;
        transition: all 0.3s ease;
        background-color: #0a192f;
        color: #e6f1ff;
    }
    .stTextArea textarea:focus {
        border-color: #64ffda;
        box-shadow: 0 0 15px rgba(100, 255, 218, 0.3);
        outline: none;
    }
    .stButton button {
        background-color: #64ffda;
        color: #0a192f;
        border: none;
        border-radius: 8px;
        font-size: 1rem;
        font-weight: 600;
        padding: 0.75rem 2rem;
        cursor: pointer;
        transition: all 0.3s ease;
        text-transform: uppercase;
        letter-spacing: 1px;
    }
    .stButton button:hover {
        background-color: #4cffbf;
        transform: translateY(-2px);
        box-shadow: 0 5px 15px rgba(76, 255, 191, 0.4);
    }
    .output-container {
        background: #1e3a5f;
        border-radius: 8px;
        padding: 1.5rem;
        margin-top: 2rem;
        border: 1px solid #3a5f8a;
        box-shadow: 0 5px 20px rgba(2, 12, 27, 0.5);
    }
    .code-block {
        background-color: #0a192f;
        border-radius: 8px;
        padding: 1rem;
        margin-top: 1rem;
        overflow-x: auto;
    }
    .stAlert {
        background-color: #172a45;
        color: #64ffda;
        border-radius: 8px;
        border: none;
        padding: 1rem;
        margin-bottom: 1.5rem;
        box-shadow: 0 3px 10px rgba(2, 12, 27, 0.3);
    }
    .stSpinner {
        color: #64ffda;
    }
    ::-webkit-scrollbar {
        width: 10px;
        height: 10px;
    }
    ::-webkit-scrollbar-track {
        background: #0a192f;
        border-radius: 4px;
    }
    ::-webkit-scrollbar-thumb {
        background: #3a5f8a;
        border-radius: 4px;
    }
    ::-webkit-scrollbar-thumb:hover {
        background: #64ffda;
    }
    .source {
        font-family: 'Fira Code', monospace;
        font-size: 0.9rem;
        line-height: 1.6;
    }
    .source .linenos {
        color: #8892b0;
        padding-right: 12px;
        border-right: 1px solid #1e3a5f;
        user-select: none;
    }
    .source pre {
        margin: 0;
        padding: 0;
    }
    .tech-news {
        background: #172a45;
        border-radius: 8px;
        padding: 1rem;
        margin-top: 2rem;
    }
    .tech-news h3 {
        color: #64ffda;
        margin-bottom: 1rem;
    }
    .tech-news ul {
        list-style-type: none;
        padding: 0;
    }
    .tech-news li {
        margin-bottom: 0.5rem;
        color: #8892b0;
    }
</style>
""", unsafe_allow_html=True)

st.markdown('<div class="main-container">', unsafe_allow_html=True)
st.title("🧠 Superintelligent AI Code Assistant")
st.markdown('<p class="subtitle">Powered by Advanced AI - Pushing the boundaries of software development</p>', unsafe_allow_html=True)

prompt = st.text_area("Present your most challenging coding problem, architecture design, or technical question:", height=120)

if st.button("Generate Cutting-Edge Solution"):
    if prompt.strip() == "":
        st.error("Please enter a valid prompt.")
    else:
        with st.spinner("Generating state-of-the-art solution..."):
            completed_text = generate_response(prompt)
            if "An error occurred" in completed_text:
                st.error(completed_text)
            else:
                st.success("Expert-level solution generated successfully!")
                
                languages = ["python", "javascript", "java", "c++", "rust", "go", "ruby", "swift", "kotlin", "typescript"]
                detected_language = next((lang for lang in languages if lang in completed_text.lower()), "plaintext")
                
                highlighted_code, css = create_code_block(completed_text, detected_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("### In-Depth Analysis and Insights")
                explanation = generate_response(f"Provide a comprehensive analysis of the solution for: {prompt}. Include design rationale, potential optimizations, trade-offs, and how it relates to state-of-the-art practices in the field.")
                st.markdown(explanation)

                st.markdown("### Alternative Approaches")
                alternatives = generate_response(f"Suggest and briefly explain three alternative approaches to solving the problem: {prompt}. Compare their pros and cons with the original solution.")
                st.markdown(alternatives)

news = fetch_latest_tech_news()
if news:
    st.markdown('<div class="tech-news">', unsafe_allow_html=True)
    st.markdown("### Latest in Tech & Computer Science")
    for item in news:
        st.markdown(f"- {item['title']}")
    st.markdown('</div>', unsafe_allow_html=True)

st.markdown("""
<div style='text-align: center; margin-top: 2rem; color: #8892b0;'>
    Engineered with 🧠💡 by Your Superintelligent AI Code Assistant
</div>
""", unsafe_allow_html=True)

st.markdown('</div>', unsafe_allow_html=True)