# import os
# from groq import Groq
# import gradio as gr

# # Initialize client
# client = Groq(api_key=os.environ.get("GROQ_API_KEY"))

# # System prompt
# SYSTEM_PROMPT = {
#     "role": "system",
#     "content": "You are a helpful, friendly AI assistant."
# }

# # Clean message (IMPORTANT FIX)
# def clean_message(msg):
#     return {
#         "role": msg.get("role"),
#         "content": msg.get("content")
#     }

# # Chatbot function
# def chatbot(message, history):
#     try:
#         messages = [SYSTEM_PROMPT]

#         # Limit history
#         history = history[-6:] if history else []

#         for item in history:
#             # Case 1: tuple (user, bot)
#             if isinstance(item, (list, tuple)) and len(item) == 2:
#                 user_msg, bot_msg = item

#                 if user_msg:
#                     messages.append({"role": "user", "content": str(user_msg)})
#                 if bot_msg:
#                     messages.append({"role": "assistant", "content": str(bot_msg)})

#             # Case 2: dict (REMOVE metadata)
#             elif isinstance(item, dict):
#                 if "role" in item and "content" in item:
#                     messages.append(clean_message(item))

#         # Add current message
#         messages.append({"role": "user", "content": str(message)})

#         # Call Groq
#         response = client.chat.completions.create(
#             model="llama-3.3-70b-versatile",
#             messages=messages,
#             temperature=0.7,
#             max_tokens=1024,
#         )

#         return response.choices[0].message.content.strip()

#     except Exception as e:
#         return f"⚠️ Error: {str(e)}"


# # UI
# demo = gr.ChatInterface(
#     fn=chatbot,
#     title="🚀 Groq AI Chatbot",
#     description="Fast chatbot powered by Groq",
# )

# # Launch
# if __name__ == "__main__":
#     demo.launch()
import os
from groq import Groq
import gradio as gr
from PyPDF2 import PdfReader

client = Groq(api_key=os.environ.get("GROQ_API_KEY"))

SYSTEM_PROMPT = {
    "role": "system",
    "content": "You are a helpful AI assistant."
}

# -------- FILE TEXT --------
def get_file_text(file):
    if not file:
        return ""

    try:
        if file.name.endswith(".pdf"):
            reader = PdfReader(file)
            text = ""
            for page in reader.pages:
                text += page.extract_text() or ""
            return text[:1500]

        elif file.name.endswith(".txt"):
            return file.read().decode("utf-8", errors="ignore")[:1500]

    except Exception as e:
        return f"(File error: {e})"

    return ""

# -------- VOICE (placeholder) --------
def get_voice_text(audio):
    if audio:
        return "User sent a voice message"
    return ""

# -------- MAIN FUNCTION --------
def respond(message, history, file, audio):

    # Combine everything into ONE message
    file_text = get_file_text(file)
    voice_text = get_voice_text(audio)

    full_message = message

    if file_text:
        full_message += f"\n\n📄 File:\n{file_text}"

    if voice_text:
        full_message += f"\n\n🎤 Voice:\n{voice_text}"

    # Build messages for Groq
    messages = [SYSTEM_PROMPT]

    for h in history:
        messages.append({"role": "user", "content": h[0]})
        messages.append({"role": "assistant", "content": h[1]})

    messages.append({"role": "user", "content": full_message})

    # Streaming response
    stream = client.chat.completions.create(
        model="llama-3.3-70b-versatile",
        messages=messages,
        stream=True,
    )

    response = ""

    for chunk in stream:
        if chunk.choices[0].delta.content:
            response += chunk.choices[0].delta.content
            yield response

# -------- UI --------
with gr.Blocks(css="""
.gradio-container {background: #0f172a; color: white;}
""") as demo:

    gr.Markdown("# 🚀 AI Chatbot (Fixed Version)")
    gr.Markdown("💬 Chat + 📄 PDF + 🎤 Voice")

    chatbot = gr.ChatInterface(
        fn=respond,
        additional_inputs=[
            gr.File(file_types=[".pdf", ".txt"]),
            gr.Audio(type="filepath")
        ],
    )

# Launch
if __name__ == "__main__":
    demo.launch()