Build an AI Web App
with Python
& Free APIs.

# Copy this file to .env and paste your real key from
# https://aistudio.google.com/app/apikey
GEMINI_API_KEY=your_api_key_here

cp .env.example .env
# Then open .env in your editor and replace "your_api_key_here"
# with your real key starting with AIza...

# Virtual environment
venv/
.env

# Python cache
__pycache__/
*.pyc

# Streamlit secrets (used in cloud deploy)
.streamlit/secrets.toml

SUMMARIZE_PROMPT = """Summarize the following text in {n} bullet points.
Be concise and factual. Do not add information that is not in the source.

TEXT:
{text}

SUMMARY:"""

SENTIMENT_PROMPT = """Analyze the sentiment of the following text.
Return your answer in this exact format:
- Overall sentiment: Positive / Negative / Neutral
- Confidence: 0-100%
- Key emotions: (comma-separated list)
- One-sentence explanation

TEXT:
{text}

ANALYSIS:"""

QA_PROMPT = """Use only the context below to answer the question.
If the answer is not contained in the context, reply exactly:
"I cannot answer this from the given context."

CONTEXT:
{context}

QUESTION:
{question}

ANSWER:"""

GENERATE_PROMPT = """You are a helpful writing assistant.
Write {n} words of {tone} content about: {topic}
Do not add a title or headings unless asked.

CONTENT:"""

TRANSLATE_PROMPT = """Translate the following text from {source_lang} to {target_lang}.
Preserve the original tone and formatting. Return only the translation.

TEXT:
{text}

TRANSLATION:"""

"""Thin wrapper around the Google Gemini API with friendly error handling."""
import os
import google.generativeai as genai
from google.api_core.exceptions import (
    ResourceExhausted, PermissionDenied, GoogleAPICallError, InvalidArgument,
)
from .prompts import (
    SUMMARIZE_PROMPT, SENTIMENT_PROMPT, QA_PROMPT,
    GENERATE_PROMPT, TRANSLATE_PROMPT,
)

# Load the API key from the environment
API_KEY = os.getenv("GEMINI_API_KEY")
if not API_KEY:
    raise RuntimeError(
        "GEMINI_API_KEY is missing. Copy .env.example to .env "
        "and paste your key from https://aistudio.google.com/app/apikey"
    )

genai.configure(api_key=API_KEY)
_MODEL = genai.GenerativeModel("gemini-1.5-flash")

class AIClientError(Exception):
    """Raised when the Gemini API call fails. The message is user-facing."""

def _safe_generate(prompt: str) -> str:
    try:
        response = _MODEL.generate_content(prompt)
        return response.text.strip()
    except ResourceExhausted:
        raise AIClientError(
            "Rate limit reached. The free tier allows 15 requests/minute "
            "and 1,500/day. Wait 60 seconds and try again."
        )
    except PermissionDenied:
        raise AIClientError(
            "API key rejected. Check that GEMINI_API_KEY in your .env "
            "matches a key from https://aistudio.google.com/app/apikey"
        )
    except InvalidArgument as e:
        raise AIClientError(f"Invalid request: {e}")
    except GoogleAPICallError as e:
        raise AIClientError(f"Gemini API error: {e}")

def summarize_text(text: str, n: int = 5) -> str:
    if len(text.strip()) < 30:
        raise AIClientError("Please paste at least 30 characters to summarize.")
    return _safe_generate(SUMMARIZE_PROMPT.format(n=n, text=text))

def analyze_sentiment(text: str) -> str:
    if len(text.strip()) < 10:
        raise AIClientError("Please paste at least 10 characters to analyze.")
    return _safe_generate(SENTIMENT_PROMPT.format(text=text))

def answer_question(context: str, question: str) -> str:
    if len(context.strip()) < 30:
        raise AIClientError("Context must be at least 30 characters.")
    if len(question.strip()) < 3:
        raise AIClientError("Question must be at least 3 characters.")
    return _safe_generate(QA_PROMPT.format(context=context, question=question))

def generate_content(topic: str, tone: str = "informative", n: int = 200) -> str:
    if len(topic.strip()) < 3:
        raise AIClientError("Please enter a topic of at least 3 characters.")
    return _safe_generate(GENERATE_PROMPT.format(topic=topic, tone=tone, n=n))

def translate_text(text: str, source_lang: str, target_lang: str) -> str:
    if len(text.strip()) < 5:
        raise AIClientError("Please enter at least 5 characters to translate.")
    return _safe_generate(
        TRANSLATE_PROMPT.format(
            text=text, source_lang=source_lang, target_lang=target_lang
        )
    )

"""Small helpers shared across the Streamlit UI."""
import streamlit as st

def show_error(message: str) -> None:
    st.error(f"⚠ {message}")

def show_success(message: str) -> None:
    st.success(f"✅ {message}")

def render_markdown(content: str) -> None:
    st.markdown(content)

"""AI Content Studio — a Streamlit web app powered by Google Gemini."""
import os
from dotenv import load_dotenv
import streamlit as st

# Load .env before importing ai_client (which reads the key at import time)
load_dotenv()

from src.ai_client import (
    AIClientError, summarize_text, analyze_sentiment,
    answer_question, generate_content, translate_text,
)
from src.utils import show_error, show_success, render_markdown

st.set_page_config(
    page_title="AI Content Studio",
    page_icon="🤖",
    layout="wide",
    initial_sidebar_state="expanded",
)

# ---------- Header ----------
st.title("🤖 AI Content Studio")
st.caption("Powered by Google Gemini 1.5 Flash · Free tier · No credit card required")

# ---------- Sidebar ----------
with st.sidebar:
    st.header("Choose a tool")
    tool = st.radio(
        "Available tools",
        ["📝 Summarize", "😊 Sentiment", "❓ Q&A", "✍️ Generate", "🌐 Translate"],
        label_visibility="collapsed",
    )
    st.divider()
    st.markdown(
        "**Free tier limits**\n"
        "- 15 requests / minute\n"
        "- 1,500 requests / day\n"
        "- 1 million tokens / minute\n\n"
        "_No credit card required. Get a key at "
        "[Google AI Studio](https://aistudio.google.com/app/apikey)._\""
    )

# ---------- Tool layouts ----------
left, right = st.columns([1, 1.2])

if tool == "📝 Summarize":
    with left:
        st.subheader("Paste your text")
        text = st.text_area("Text to summarize", height=300, key="sum_text")
        n = st.slider("Number of bullet points", 3, 10, 5)
        if st.button("Summarize", type="primary"):
            try:
                with st.spinner("Summarizing..."):
                    out = summarize_text(text, n=n)
                show_success("Done!")
                render_markdown(out)
            except AIClientError as e:
                show_error(str(e))

elif tool == "😊 Sentiment":
    with left:
        st.subheader("Paste text to analyze")
        text = st.text_area("Text", height=300, key="sent_text")
        if st.button("Analyze", type="primary"):
            try:
                with st.spinner("Analyzing..."):
                    out = analyze_sentiment(text)
                show_success("Done!")
                render_markdown(out)
            except AIClientError as e:
                show_error(str(e))

elif tool == "❓ Q&A":
    with left:
        st.subheader("Provide context and a question")
        context = st.text_area("Context (the source text)", height=220, key="qa_ctx")
        question = st.text_input("Your question", key="qa_q")
        if st.button("Answer", type="primary"):
            try:
                with st.spinner("Answering..."):
                    out = answer_question(context, question)
                show_success("Done!")
                render_markdown(out)
            except AIClientError as e:
                show_error(str(e))

elif tool == "✍️ Generate":
    with left:
        st.subheader("Generate new content")
        topic = st.text_input("Topic", key="gen_topic")
        tone = st.selectbox("Tone", ["informative", "casual", "professional", "funny", "persuasive"])
        n = st.slider("Word count", 50, 500, 200, step=50)
        if st.button("Generate", type="primary"):
            try:
                with st.spinner("Generating..."):
                    out = generate_content(topic, tone=tone, n=n)
                show_success("Done!")
                render_markdown(out)
            except AIClientError as e:
                show_error(str(e))

elif tool == "🌐 Translate":
    with left:
        st.subheader("Translate text")
        text = st.text_area("Text to translate", height=220, key="tr_text")
        col1, col2 = st.columns(2)
        with col1:
            source_lang = st.text_input("From", value="English")
        with col2:
            target_lang = st.text_input("To", value="Spanish")
        if st.button("Translate", type="primary"):
            try:
                with st.spinner("Translating..."):
                    out = translate_text(text, source_lang, target_lang)
                show_success("Done!")
                render_markdown(out)
            except AIClientError as e:
                show_error(str(e))

with right:
    st.subheader("Result")
    st.info("Your AI output will appear on the left after you click the action button.")

# ---------- Footer ----------
st.divider()
st.caption(
    "Built with [Streamlit](https://streamlit.io) + "
    "[Google Gemini](https://ai.google.dev). "
    "Source: [Darekdari.com tutorial](https://darekdari.com/build-ai-web-app-python-free-apis/)"
)

[server]
maxUploadSize = 5

[theme]
base = "dark"
primaryColor = "#22d3ee"
backgroundColor = "#04060d"
secondaryBackgroundColor = "#0a1128"
textColor = "#f5f8ff"