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 |
|