A Step-by-Step Coding Guide to Building a Gemini-Powered AI Startup Pitch Generator Using LiteLLM Framework, Gradio, and FPDF in Google Colab with PDF Export Support

!pip install litellm gradio fpdf --quiet

import os
import gradio as gr
import uuid
import urllib.request
from fpdf import FPDF
from litellm import completion


api_key = "Your API Key"

import urllib.request
import zipfile
import os
import shutil


if not os.path.exists("DejaVuSans.ttf"):
    print(" Downloading DejaVuSans.ttf...")
    font_zip_url = "https://downloads.sourceforge.net/project/dejavu/dejavu/2.37/dejavu-fonts-ttf-2.37.zip"
    font_zip_path = "dejavu-fonts.zip"


    urllib.request.urlretrieve(font_zip_url, font_zip_path)


    with zipfile.ZipFile(font_zip_path, 'r') as zip_ref:
        zip_ref.extractall("dejavu-extracted")


    for root, dirs, files in os.walk("dejavu-extracted"):
        for file in files:
            if file == "DejaVuSans.ttf":
                ttf_path = os.path.join(root, file)
                shutil.copy(ttf_path, "DejaVuSans.ttf")
                print(" Font extracted and ready.")
                break

def call_gemini(system_prompt, user_prompt):
    messages = [
        {"role": "system", "content": system_prompt},
        {"role": "user", "content": user_prompt}
    ]
    response = completion(
        model="gemini/gemini-2.0-flash-lite",
        messages=messages,
        api_key=api_key
    )
    return response["choices"][0]["message"]["content"]

def generate_startup_pitch(theme):
    try:
        idea_prompt = f"Generate an innovative startup idea in the field of {theme}. Focus on solving real problems using modern technology."
        tagline_prompt = "Based on the idea you just gave, generate a short, catchy tagline for the startup."
        pitch_prompt = """
        Based on the previous startup idea, write a concise pitch deck covering:
        1. Problem
        2. Solution
        3. Market Opportunity
        4. Team Description
        5. Business Model
        6. Traction or Future Plan
        Format it in a way that looks like slide notes for a VC pitch.
        """


        idea = call_gemini("You are an innovation strategist.", idea_prompt)
        tagline = call_gemini("You are a branding expert.", tagline_prompt)
        pitch = call_gemini("You are a startup mentor writing a pitch deck.", pitch_prompt)


        filename = f"startup_pitch_{uuid.uuid4().hex[:8]}.pdf"
        pdf = FPDF()
        pdf.add_page()
        pdf.add_font("DejaVu", "", font_path, uni=True)
        pdf.set_font("DejaVu", size=12)


        full_text = f"Startup Idea:\n{idea}\n\nTagline:\n{tagline}\n\nPitch Deck:\n{pitch}"
        pdf.multi_cell(0, 10, full_text)
        pdf.output(filename)


        return idea, tagline, pitch, filename
    except Exception as e:
        return f" Error: {e}", "", "", None

with gr.Blocks() as demo:
    gr.Markdown("# 
                            
                                Read More