Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -6,6 +6,7 @@ import re
|
|
6 |
from dotenv import load_dotenv
|
7 |
import os
|
8 |
from fastapi import FastAPI
|
|
|
9 |
from pydantic import BaseModel
|
10 |
from fastapi.middleware.cors import CORSMiddleware
|
11 |
|
@@ -64,7 +65,7 @@ app.add_middleware(
|
|
64 |
|
65 |
# Create a Pydantic model to handle the input data
|
66 |
class TopicInput(BaseModel):
|
67 |
-
user_input: str
|
68 |
num_topics: int = Query(default=5, description="Number of subtopics to generate (default: 5)")
|
69 |
previous_query: str = Query(default="", description="Previous query for context (default: empty string)")
|
70 |
|
|
|
6 |
from dotenv import load_dotenv
|
7 |
import os
|
8 |
from fastapi import FastAPI
|
9 |
+
from fastapi import Query
|
10 |
from pydantic import BaseModel
|
11 |
from fastapi.middleware.cors import CORSMiddleware
|
12 |
|
|
|
65 |
|
66 |
# Create a Pydantic model to handle the input data
|
67 |
class TopicInput(BaseModel):
|
68 |
+
user_input: str = Query(default="", description="input query to generate subtopics")
|
69 |
num_topics: int = Query(default=5, description="Number of subtopics to generate (default: 5)")
|
70 |
previous_query: str = Query(default="", description="Previous query for context (default: empty string)")
|
71 |
|