Генератор стартап-питчей на базе AI: пошаговое руководство по созданию с использованием LiteLLM, Gradio и FPDF

Генератор Питчей для Стартапов на Основе AI

В этом руководстве мы создадим мощное и интерактивное AI-приложение для генерации идей питчей стартапов, используя модель Gemini Pro от Google через универсальный фреймворк LiteLLM.

Обзор Проекта

LiteLLM предоставляет единый интерфейс для взаимодействия с более чем 100 провайдерами языковых моделей, упрощая работу с API. Мы интегрируем возможности Gemini для креативного мышления и оформляем результаты в удобный интерфейс Gradio. Также мы используем FPDF для генерации качественных PDF-документов с полным питчем стартапа.

Установка Библиотек

Для начала установим необходимые библиотеки:

        !pip install litellm gradio fpdf --quiet

Эти библиотеки необходимы для работы с API Gemini, создания веб-интерфейса и экспорта в PDF.

Импорт Библиотек

Импортируем основные библиотеки Python:

        
import os
import gradio as gr
import uuid
from fpdf import FPDF
from litellm import completion

api_key = "Ваш API ключ"

Подготовка Шрифта

Убедимся, что шрифт DejaVuS доступен для создания PDF-документов:

        
if not os.path.exists("DejaVuS"):
    print("Скачивание DejaVuS...")
    # Код для скачивания и извлечения шрифта

Вызов Модели Gemini

Создаем функцию для взаимодействия с моделью Gemini:

        
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"Создайте инновационную идею стартапа в области {theme}."
        tagline_prompt = "Сгенерируйте короткий слоган для стартапа."
        pitch_prompt = "Напишите краткий питч, включая проблему, решение и бизнес-модель."
        
        idea = call_gemini("Вы стратег инноваций.", idea_prompt)
        tagline = call_gemini("Вы эксперт по брендингу.", tagline_prompt)
        pitch = call_gemini("Вы ментор стартапов.", pitch_prompt)

        filename = f"startup_pitch_{uuid.uuid4().hex[:8]}.pdf"
        pdf = FPDF()
        pdf.add_page()
        pdf.set_font("DejaVu", size=12)
        full_text = f"Идея стартапа:\n{idea}\n\nСлоган:\n{tagline}\n\nПитч:\n{pitch}"
        pdf.multi_cell(0, 10, full_text)
        pdf.output(filename)

        return idea, tagline, pitch, filename
    except Exception as e:
        return f"Ошибка: {e}", "", "", None

Создание Интерфейса Пользователя

Определяем интерфейс Gradio для генератора питчей:

        
with gr.Blocks() as demo:
    gr.Markdown("# Генератор Питчей Стартапов (с экспортом в PDF)")
    theme_input = gr.Textbox(label="Введите тему или отрасль", placeholder="например, психическое здоровье, финтех")
    generate_button = gr.Button("Сгенерировать Питч")
    idea_output = gr.Textbox(label="Идея стартапа")
    tagline_output = gr.Textbox(label="Слоган")
    pitch_output = gr.Textbox(label="Краткое содержание питча", lines=10)
    pdf_output = gr.File(label="Скачать питч в PDF")

    def wrapper(theme):
        idea, tagline, pitch, pdf_path = generate_startup_pitch(theme)
        return idea, tagline, pitch, pdf_path

    generate_button.click(fn=wrapper, inputs=theme_input, outputs=[idea_output, tagline_output, pitch_output, pdf_output])
    demo.launch(share=True)

Заключение

Сочетая возможности LiteLLM и креативный интеллект Gemini Pro, мы показали, как разработчики могут быстро создавать интеллектуальные приложения. LiteLLM упрощает работу с API языковых моделей, а Gradio обеспечивает интуитивно понятный интерфейс для пользователей.