Spaces:
Running
Running
Kuberwastaken
commited on
Commit
·
9e95cac
1
Parent(s):
6f07235
V3 Fully Revamped UI + IMSDB Support
Browse files- __pycache__/script_search_api.cpython-310.pyc +0 -0
- gradio_app.py +443 -176
- script_search_api.py +279 -0
__pycache__/script_search_api.cpython-310.pyc
ADDED
Binary file (8.41 kB). View file
|
|
gradio_app.py
CHANGED
@@ -2,216 +2,468 @@ import gradio as gr
|
|
2 |
from model.analyzer import analyze_content
|
3 |
import asyncio
|
4 |
import time
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
|
6 |
custom_css = """
|
7 |
* {
|
8 |
-
font-family: '
|
9 |
-
transition: all 0.3s
|
10 |
}
|
|
|
11 |
.gradio-container {
|
12 |
-
background: #
|
13 |
color: #fff !important;
|
|
|
|
|
14 |
overflow: hidden;
|
15 |
-
transition: all 0.5s ease;
|
16 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
.treat-title {
|
18 |
text-align: center;
|
19 |
-
padding:
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
}
|
27 |
-
.treat-title:hover {
|
28 |
-
transform: translateY(-5px);
|
29 |
-
box-shadow: 0 8px 16px rgba(106, 13, 173);
|
30 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
31 |
.treat-title h1 {
|
32 |
-
font-size:
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
38 |
}
|
39 |
-
|
40 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
41 |
}
|
|
|
42 |
.treat-title p {
|
43 |
-
font-size: 1.
|
44 |
-
color:
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
52 |
}
|
53 |
-
|
54 |
-
|
55 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
56 |
}
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
|
|
|
|
|
66 |
}
|
67 |
-
|
68 |
-
|
69 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
70 |
}
|
|
|
|
|
71 |
.gradio-textbox textarea {
|
72 |
-
background
|
73 |
-
|
74 |
-
border: 1px solid rgb(106, 13, 173) !important;
|
75 |
border-radius: 8px !important;
|
76 |
-
|
77 |
-
font-size:
|
|
|
|
|
78 |
transition: all 0.3s ease;
|
|
|
|
|
79 |
}
|
80 |
-
|
81 |
-
border-color: rgb(106, 13, 173) !important;
|
82 |
-
box-shadow: 0 0 10px rgb(106, 13, 173) !important;
|
83 |
-
}
|
84 |
.gradio-textbox textarea:focus {
|
85 |
-
border-color:
|
86 |
-
box-shadow: 0 0
|
|
|
87 |
transform: translateY(-2px);
|
88 |
}
|
|
|
|
|
89 |
.gradio-button {
|
90 |
-
background
|
91 |
-
|
|
|
|
|
|
|
92 |
border: none !important;
|
93 |
-
border-radius:
|
94 |
-
|
95 |
-
font-
|
96 |
-
|
97 |
-
|
|
|
98 |
position: relative;
|
99 |
overflow: hidden;
|
|
|
|
|
100 |
}
|
|
|
101 |
.gradio-button::before {
|
102 |
content: '';
|
103 |
position: absolute;
|
104 |
-
top: 50%;
|
105 |
-
left: 50%;
|
106 |
-
width:
|
107 |
-
height:
|
108 |
-
background:
|
109 |
-
|
110 |
-
|
111 |
-
|
|
|
112 |
}
|
|
|
113 |
.gradio-button:hover {
|
114 |
-
transform:
|
115 |
-
|
116 |
-
box-shadow: 0 6px 15px rgb(106, 13, 173);
|
117 |
}
|
|
|
118 |
.gradio-button:hover::before {
|
119 |
-
|
120 |
-
height: 300px;
|
121 |
}
|
122 |
-
.gradio-button:active {
|
123 |
-
transform: scale(0.98) translateY(1px);
|
124 |
-
background-color: rgb(106, 13, 173) !important;
|
125 |
|
126 |
-
/*
|
127 |
-
.
|
128 |
-
background
|
129 |
-
border
|
|
|
|
|
|
|
|
|
|
|
|
|
130 |
}
|
131 |
|
132 |
-
}
|
133 |
-
label {
|
134 |
-
color: rgb(106, 13, 173) !important;
|
135 |
-
font-weight: 500 !important;
|
136 |
-
margin-bottom: 10px !important;
|
137 |
-
transition: color 0.3s ease;
|
138 |
-
}
|
139 |
-
label:hover {
|
140 |
-
color: #fff !important;
|
141 |
-
}
|
142 |
-
.center-row {
|
143 |
-
display: flex;
|
144 |
-
justify-content: center;
|
145 |
-
align-items: center;
|
146 |
-
}
|
147 |
.footer {
|
148 |
text-align: center;
|
149 |
-
|
150 |
-
|
151 |
-
|
152 |
-
|
153 |
-
|
154 |
-
transition: all 0.3s ease;
|
155 |
-
}
|
156 |
-
.footer:hover {
|
157 |
-
transform: translateY(-3px);
|
158 |
}
|
|
|
159 |
.footer p {
|
160 |
-
color:
|
161 |
-
|
|
|
|
|
|
|
162 |
}
|
|
|
163 |
.footer .heart {
|
|
|
164 |
display: inline-block;
|
165 |
-
|
166 |
-
|
|
|
|
|
167 |
}
|
168 |
-
|
169 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
170 |
}
|
171 |
-
|
172 |
-
|
|
|
173 |
text-decoration: none;
|
174 |
position: relative;
|
175 |
transition: all 0.3s ease;
|
|
|
176 |
}
|
177 |
-
|
178 |
-
|
179 |
-
|
180 |
-
.footer a::after {
|
181 |
-
content: '';
|
182 |
-
position: absolute;
|
183 |
-
width: 0;
|
184 |
-
height: 2px;
|
185 |
-
bottom: -2px;
|
186 |
-
left: 0;
|
187 |
-
background-color: rgb(106, 13, 173);
|
188 |
-
transition: width 0.3s ease;
|
189 |
-
}
|
190 |
-
.footer a:hover::after {
|
191 |
-
width: 100%;
|
192 |
}
|
|
|
193 |
footer {
|
194 |
visibility: hidden;
|
195 |
-
}
|
196 |
|
197 |
-
|
198 |
-
|
199 |
-
|
|
|
|
|
200 |
}
|
201 |
-
|
202 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
203 |
100% { opacity: 1; transform: translateY(0); }
|
204 |
}
|
|
|
205 |
@keyframes fadeInUp {
|
206 |
-
0% { opacity: 0; transform: translateY(
|
207 |
100% { opacity: 1; transform: translateY(0); }
|
208 |
}
|
209 |
-
|
|
|
210 |
0% { transform: scale(1); }
|
211 |
-
|
|
|
|
|
|
|
|
|
212 |
100% { transform: scale(1); }
|
213 |
}
|
|
|
|
|
|
|
|
|
|
|
|
|
214 |
"""
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
215 |
|
216 |
def analyze_with_loading(text, progress=gr.Progress()):
|
217 |
"""
|
@@ -256,55 +508,70 @@ def analyze_with_loading(text, progress=gr.Progress()):
|
|
256 |
trigger_list = "\n".join([f"• {trigger}" for trigger in triggers])
|
257 |
return f"⚠ Triggers Detected:\n{trigger_list}"
|
258 |
|
259 |
-
#
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
260 |
with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as iface:
|
261 |
# Title section
|
262 |
gr.HTML("""
|
263 |
<div class="treat-title">
|
264 |
-
<h1>TREAT</h1>
|
265 |
-
<p
|
266 |
-
<span class="highlight">R</span>ecognition for
|
267 |
-
<span class="highlight">E</span>njoyable and
|
268 |
-
<span class="highlight">A</span>ppropriate
|
269 |
-
<span class="highlight">T</span>elevision</p>
|
270 |
</div>
|
271 |
""")
|
272 |
|
273 |
-
|
274 |
-
|
275 |
-
|
276 |
-
|
277 |
-
|
278 |
-
|
279 |
-
|
280 |
-
|
281 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
282 |
|
283 |
-
|
284 |
-
|
285 |
-
|
286 |
-
|
287 |
-
|
288 |
-
)
|
289 |
|
290 |
-
|
291 |
-
|
292 |
-
|
293 |
-
output_text = gr.Textbox(
|
294 |
-
label="Analysis Results",
|
295 |
-
lines=5,
|
296 |
-
interactive=False
|
297 |
-
)
|
298 |
|
299 |
-
#
|
300 |
analyze_btn.click(
|
301 |
fn=analyze_with_loading,
|
302 |
-
inputs=
|
303 |
-
outputs=
|
304 |
-
|
|
|
|
|
|
|
|
|
|
|
305 |
)
|
306 |
|
307 |
-
# Footer section
|
308 |
gr.HTML("""
|
309 |
<div class="footer">
|
310 |
<p>Made with <span class="heart">💖</span> by <a href="https://www.linkedin.com/in/kubermehta/" target="_blank">Kuber Mehta</a></p>
|
|
|
2 |
from model.analyzer import analyze_content
|
3 |
import asyncio
|
4 |
import time
|
5 |
+
import httpx
|
6 |
+
import subprocess
|
7 |
+
import atexit
|
8 |
+
|
9 |
+
# Start the API server
|
10 |
+
def start_api_server():
|
11 |
+
# Start uvicorn in a subprocess
|
12 |
+
process = subprocess.Popen(["uvicorn", "script_search_api:app", "--reload"])
|
13 |
+
return process
|
14 |
+
|
15 |
+
# Stop the API server
|
16 |
+
def stop_api_server(process):
|
17 |
+
process.terminate()
|
18 |
+
|
19 |
+
# Register the exit handler
|
20 |
+
api_process = start_api_server()
|
21 |
+
atexit.register(stop_api_server, api_process)
|
22 |
+
|
23 |
|
24 |
custom_css = """
|
25 |
* {
|
26 |
+
font-family: 'Inter', system-ui, sans-serif;
|
27 |
+
transition: all 0.3s cubic-bezier(0.4, 0, 0.2, 1);
|
28 |
}
|
29 |
+
|
30 |
.gradio-container {
|
31 |
+
background: #0a0a0f !important;
|
32 |
color: #fff !important;
|
33 |
+
min-height: 100vh;
|
34 |
+
position: relative;
|
35 |
overflow: hidden;
|
|
|
36 |
}
|
37 |
+
|
38 |
+
/* Animated Background */
|
39 |
+
.gradio-container::before {
|
40 |
+
content: '';
|
41 |
+
position: fixed;
|
42 |
+
top: 0;
|
43 |
+
left: 0;
|
44 |
+
right: 0;
|
45 |
+
bottom: 0;
|
46 |
+
background:
|
47 |
+
linear-gradient(125deg,
|
48 |
+
#0a0a0f 0%,
|
49 |
+
rgba(99, 102, 241, 0.05) 30%,
|
50 |
+
rgba(99, 102, 241, 0.1) 50%,
|
51 |
+
rgba(99, 102, 241, 0.05) 70%,
|
52 |
+
#0a0a0f 100%);
|
53 |
+
animation: gradientMove 15s ease infinite;
|
54 |
+
background-size: 400% 400%;
|
55 |
+
z-index: 0;
|
56 |
+
}
|
57 |
+
|
58 |
+
/* Floating Particles */
|
59 |
+
.gradio-container::after {
|
60 |
+
content: '';
|
61 |
+
position: fixed;
|
62 |
+
top: 0;
|
63 |
+
left: 0;
|
64 |
+
width: 100%;
|
65 |
+
height: 100%;
|
66 |
+
background: radial-gradient(circle at center, transparent 0%, #0a0a0f 70%),
|
67 |
+
url("data:image/svg+xml,%3Csvg width='100' height='100' viewBox='0 0 100 100' xmlns='http://www.w3.org/2000/svg'%3E%3Ccircle cx='50' cy='50' r='1' fill='rgba(99, 102, 241, 0.15)'/%3E%3C/svg%3E");
|
68 |
+
opacity: 0.5;
|
69 |
+
animation: floatingParticles 20s linear infinite;
|
70 |
+
z-index: 1;
|
71 |
+
}
|
72 |
+
|
73 |
+
/* Futuristic Header */
|
74 |
.treat-title {
|
75 |
text-align: center;
|
76 |
+
padding: 3rem 1rem;
|
77 |
+
position: relative;
|
78 |
+
overflow: hidden;
|
79 |
+
z-index: 2;
|
80 |
+
background: linear-gradient(180deg,
|
81 |
+
rgba(99, 102, 241, 0.1),
|
82 |
+
transparent 70%);
|
|
|
|
|
|
|
|
|
83 |
}
|
84 |
+
|
85 |
+
.treat-title::before {
|
86 |
+
content: '';
|
87 |
+
position: absolute;
|
88 |
+
top: 0;
|
89 |
+
left: 50%;
|
90 |
+
width: 80%;
|
91 |
+
height: 1px;
|
92 |
+
background: linear-gradient(90deg,
|
93 |
+
transparent,
|
94 |
+
rgba(99, 102, 241, 0.5),
|
95 |
+
transparent);
|
96 |
+
transform: translateX(-50%);
|
97 |
+
animation: scanline 3s ease-in-out infinite;
|
98 |
+
}
|
99 |
+
|
100 |
.treat-title h1 {
|
101 |
+
font-size: 4.5rem;
|
102 |
+
font-weight: 800;
|
103 |
+
background: linear-gradient(135deg,
|
104 |
+
#2a2b55 0%,
|
105 |
+
#6366f1 50%,
|
106 |
+
#2a2b55 100%);
|
107 |
+
background-size: 200% auto;
|
108 |
+
-webkit-background-clip: text;
|
109 |
+
-webkit-text-fill-color: transparent;
|
110 |
+
margin-bottom: 0.5rem;
|
111 |
+
letter-spacing: -0.05em;
|
112 |
+
animation: gradientFlow 8s ease infinite;
|
113 |
+
position: relative;
|
114 |
}
|
115 |
+
|
116 |
+
.treat-title h1::after {
|
117 |
+
content: attr(data-text);
|
118 |
+
position: absolute;
|
119 |
+
left: 0;
|
120 |
+
top: 0;
|
121 |
+
width: 100%;
|
122 |
+
height: 100%;
|
123 |
+
background: linear-gradient(135deg,
|
124 |
+
transparent 0%,
|
125 |
+
rgba(99, 102, 241, 0.4) 50%,
|
126 |
+
transparent 100%);
|
127 |
+
background-size: 200% auto;
|
128 |
+
-webkit-background-clip: text;
|
129 |
+
-webkit-text-fill-color: transparent;
|
130 |
+
opacity: 0.5;
|
131 |
+
animation: textGlow 4s ease-in-out infinite;
|
132 |
}
|
133 |
+
|
134 |
.treat-title p {
|
135 |
+
font-size: 1.1rem;
|
136 |
+
color: rgba(255, 255, 255, 0.7);
|
137 |
+
max-width: 600px;
|
138 |
+
margin: 0 auto;
|
139 |
+
position: relative;
|
140 |
+
animation: fadeInUp 1s ease-out;
|
141 |
+
}
|
142 |
+
|
143 |
+
/* Tabs Styling */
|
144 |
+
.tabs {
|
145 |
+
background: rgba(17, 17, 27, 0.7);
|
146 |
+
border: 1px solid rgba(99, 102, 241, 0.2);
|
147 |
+
border-radius: 16px;
|
148 |
+
padding: 1rem;
|
149 |
+
margin: 0 1rem 2rem 1rem;
|
150 |
+
position: relative;
|
151 |
+
z-index: 2;
|
152 |
+
backdrop-filter: blur(10px);
|
153 |
+
box-shadow: 0 0 30px rgba(99, 102, 241, 0.1);
|
154 |
+
animation: floatIn 1s ease-out;
|
155 |
}
|
156 |
+
|
157 |
+
.tabs::before {
|
158 |
+
content: '';
|
159 |
+
position: absolute;
|
160 |
+
top: -1px;
|
161 |
+
left: -1px;
|
162 |
+
right: -1px;
|
163 |
+
bottom: -1px;
|
164 |
+
background: linear-gradient(45deg,
|
165 |
+
rgba(99, 102, 241, 0.1),
|
166 |
+
transparent,
|
167 |
+
rgba(99, 102, 241, 0.1));
|
168 |
+
border-radius: 16px;
|
169 |
+
z-index: -1;
|
170 |
+
animation: borderGlow 4s ease-in-out infinite;
|
171 |
}
|
172 |
+
|
173 |
+
/* Content Area */
|
174 |
+
.content-area {
|
175 |
+
background: rgba(17, 17, 27, 0.7) !important;
|
176 |
+
border: 1px solid rgba(99, 102, 241, 0.2) !important;
|
177 |
+
border-radius: 12px !important;
|
178 |
+
padding: 1.5rem !important;
|
179 |
+
backdrop-filter: blur(10px);
|
180 |
+
position: relative;
|
181 |
+
overflow: hidden;
|
182 |
+
animation: fadeScale 0.5s ease-out;
|
183 |
}
|
184 |
+
|
185 |
+
.content-area::before {
|
186 |
+
content: '';
|
187 |
+
position: absolute;
|
188 |
+
top: -50%;
|
189 |
+
left: -50%;
|
190 |
+
width: 200%;
|
191 |
+
height: 200%;
|
192 |
+
background: radial-gradient(circle at center,
|
193 |
+
rgba(99, 102, 241, 0.1) 0%,
|
194 |
+
transparent 70%);
|
195 |
+
animation: rotateGradient 10s linear infinite;
|
196 |
}
|
197 |
+
|
198 |
+
/* Input Fields */
|
199 |
.gradio-textbox textarea {
|
200 |
+
background: rgba(17, 17, 27, 0.6) !important;
|
201 |
+
border: 1px solid rgba(99, 102, 241, 0.3) !important;
|
|
|
202 |
border-radius: 8px !important;
|
203 |
+
color: rgba(255, 255, 255, 0.9) !important;
|
204 |
+
font-size: 0.95rem !important;
|
205 |
+
line-height: 1.6 !important;
|
206 |
+
padding: 1rem !important;
|
207 |
transition: all 0.3s ease;
|
208 |
+
position: relative;
|
209 |
+
z-index: 2;
|
210 |
}
|
211 |
+
|
|
|
|
|
|
|
212 |
.gradio-textbox textarea:focus {
|
213 |
+
border-color: #6366f1 !important;
|
214 |
+
box-shadow: 0 0 20px rgba(99, 102, 241, 0.2) !important;
|
215 |
+
background: rgba(17, 17, 27, 0.8) !important;
|
216 |
transform: translateY(-2px);
|
217 |
}
|
218 |
+
|
219 |
+
/* Buttons */
|
220 |
.gradio-button {
|
221 |
+
background: linear-gradient(45deg,
|
222 |
+
#6366f1,
|
223 |
+
#818cf8,
|
224 |
+
#6366f1) !important;
|
225 |
+
background-size: 200% auto !important;
|
226 |
border: none !important;
|
227 |
+
border-radius: 8px !important;
|
228 |
+
color: white !important;
|
229 |
+
font-weight: 600 !important;
|
230 |
+
font-size: 0.95rem !important;
|
231 |
+
padding: 0.75rem 1.5rem !important;
|
232 |
+
letter-spacing: 0.025em !important;
|
233 |
position: relative;
|
234 |
overflow: hidden;
|
235 |
+
transition: all 0.3s ease !important;
|
236 |
+
animation: gradientFlow 3s ease infinite;
|
237 |
}
|
238 |
+
|
239 |
.gradio-button::before {
|
240 |
content: '';
|
241 |
position: absolute;
|
242 |
+
top: -50%;
|
243 |
+
left: -50%;
|
244 |
+
width: 200%;
|
245 |
+
height: 200%;
|
246 |
+
background: radial-gradient(circle at center,
|
247 |
+
rgba(255, 255, 255, 0.2) 0%,
|
248 |
+
transparent 70%);
|
249 |
+
transform: scale(0);
|
250 |
+
transition: transform 0.5s ease;
|
251 |
}
|
252 |
+
|
253 |
.gradio-button:hover {
|
254 |
+
transform: translateY(-2px);
|
255 |
+
box-shadow: 0 5px 20px rgba(99, 102, 241, 0.4) !important;
|
|
|
256 |
}
|
257 |
+
|
258 |
.gradio-button:hover::before {
|
259 |
+
transform: scale(1);
|
|
|
260 |
}
|
|
|
|
|
|
|
261 |
|
262 |
+
/* Results Area */
|
263 |
+
.results-area {
|
264 |
+
background: rgba(17, 17, 27, 0.7) !important;
|
265 |
+
border: 1px solid rgba(99, 102, 241, 0.2) !important;
|
266 |
+
border-radius: 12px !important;
|
267 |
+
margin-top: 2rem !important;
|
268 |
+
backdrop-filter: blur(10px);
|
269 |
+
animation: slideUp 0.5s ease-out;
|
270 |
+
position: relative;
|
271 |
+
overflow: hidden;
|
272 |
}
|
273 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
274 |
.footer {
|
275 |
text-align: center;
|
276 |
+
padding: 2rem 0;
|
277 |
+
margin-top: 3rem;
|
278 |
+
font-size: 1.0rem;
|
279 |
+
position: relative;
|
280 |
+
z-index: 2;
|
|
|
|
|
|
|
|
|
281 |
}
|
282 |
+
|
283 |
.footer p {
|
284 |
+
color: rgba(255, 255, 255, 0.8);
|
285 |
+
display: flex;
|
286 |
+
align-items: center;
|
287 |
+
justify-content: center;
|
288 |
+
gap: 0.5rem;
|
289 |
}
|
290 |
+
|
291 |
.footer .heart {
|
292 |
+
color: #6366f1;
|
293 |
display: inline-block;
|
294 |
+
position: relative;
|
295 |
+
font-size: 1.0rem;
|
296 |
+
transform-origin: center;
|
297 |
+
animation: heartbeat 1.5s ease infinite;
|
298 |
}
|
299 |
+
|
300 |
+
.footer .heart::before,
|
301 |
+
.footer .heart::after {
|
302 |
+
content: '✦';
|
303 |
+
position: absolute;
|
304 |
+
opacity: 0;
|
305 |
+
font-size: 0.6rem;
|
306 |
+
animation: sparkle 1.5s ease infinite;
|
307 |
+
}
|
308 |
+
|
309 |
+
.footer .heart::before {
|
310 |
+
top: -8px;
|
311 |
+
left: -8px;
|
312 |
+
animation-delay: 0.2s;
|
313 |
+
}
|
314 |
+
|
315 |
+
.footer .heart::after {
|
316 |
+
top: -8px;
|
317 |
+
right: -8px;
|
318 |
+
animation-delay: 0.4s;
|
319 |
}
|
320 |
+
|
321 |
+
.footer .name {
|
322 |
+
color: #6366f1;
|
323 |
text-decoration: none;
|
324 |
position: relative;
|
325 |
transition: all 0.3s ease;
|
326 |
+
padding: 0 4px;
|
327 |
}
|
328 |
+
|
329 |
+
.footer .name:hover {
|
330 |
+
color: #818cf8;
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
331 |
}
|
332 |
+
|
333 |
footer {
|
334 |
visibility: hidden;
|
335 |
+
}
|
336 |
|
337 |
+
/* Animations */
|
338 |
+
@keyframes gradientMove {
|
339 |
+
0% { background-position: 0% 50%; }
|
340 |
+
50% { background-position: 100% 50%; }
|
341 |
+
100% { background-position: 0% 50%; }
|
342 |
}
|
343 |
+
|
344 |
+
@keyframes floatingParticles {
|
345 |
+
0% { transform: translateY(0); }
|
346 |
+
100% { transform: translateY(-100%); }
|
347 |
+
}
|
348 |
+
|
349 |
+
@keyframes scanline {
|
350 |
+
0% { transform: translateX(-150%) scaleX(0.5); opacity: 0; }
|
351 |
+
50% { transform: translateX(-50%) scaleX(1); opacity: 1; }
|
352 |
+
100% { transform: translateX(50%) scaleX(0.5); opacity: 0; }
|
353 |
+
}
|
354 |
+
|
355 |
+
@keyframes gradientFlow {
|
356 |
+
0% { background-position: 0% 50%; }
|
357 |
+
50% { background-position: 100% 50%; }
|
358 |
+
100% { background-position: 0% 50%; }
|
359 |
+
}
|
360 |
+
|
361 |
+
@keyframes textGlow {
|
362 |
+
0% { opacity: 0.3; transform: scale(1); }
|
363 |
+
50% { opacity: 0.5; transform: scale(1.02); }
|
364 |
+
100% { opacity: 0.3; transform: scale(1); }
|
365 |
+
}
|
366 |
+
|
367 |
+
@keyframes borderGlow {
|
368 |
+
0% { opacity: 0.5; }
|
369 |
+
50% { opacity: 1; }
|
370 |
+
100% { opacity: 0.5; }
|
371 |
+
}
|
372 |
+
|
373 |
+
@keyframes rotateGradient {
|
374 |
+
0% { transform: rotate(0deg); }
|
375 |
+
100% { transform: rotate(360deg); }
|
376 |
+
}
|
377 |
+
|
378 |
+
@keyframes fadeScale {
|
379 |
+
0% { opacity: 0; transform: scale(0.95); }
|
380 |
+
100% { opacity: 1; transform: scale(1); }
|
381 |
+
}
|
382 |
+
|
383 |
+
@keyframes slideUp {
|
384 |
+
0% { opacity: 0; transform: translateY(20px); }
|
385 |
+
100% { opacity: 1; transform: translateY(0); }
|
386 |
+
}
|
387 |
+
|
388 |
+
@keyframes floatIn {
|
389 |
+
0% { opacity: 0; transform: translateY(20px); }
|
390 |
100% { opacity: 1; transform: translateY(0); }
|
391 |
}
|
392 |
+
|
393 |
@keyframes fadeInUp {
|
394 |
+
0% { opacity: 0; transform: translateY(10px); }
|
395 |
100% { opacity: 1; transform: translateY(0); }
|
396 |
}
|
397 |
+
|
398 |
+
@keyframes heartbeat {
|
399 |
0% { transform: scale(1); }
|
400 |
+
10% { transform: scale(1.2); }
|
401 |
+
20% { transform: scale(0.9); }
|
402 |
+
30% { transform: scale(1.1); }
|
403 |
+
40% { transform: scale(0.95); }
|
404 |
+
50% { transform: scale(1); }
|
405 |
100% { transform: scale(1); }
|
406 |
}
|
407 |
+
|
408 |
+
@keyframes sparkle {
|
409 |
+
0% { transform: scale(0); opacity: 0; }
|
410 |
+
50% { transform: scale(1.2); opacity: 1; }
|
411 |
+
100% { transform: scale(0); opacity: 0; }
|
412 |
+
}
|
413 |
"""
|
414 |
+
# Start the API server
|
415 |
+
def start_api_server():
|
416 |
+
# Start uvicorn in a subprocess
|
417 |
+
process = subprocess.Popen(["uvicorn", "script_search_api:app", "--reload"])
|
418 |
+
return process
|
419 |
+
|
420 |
+
# Stop the API server
|
421 |
+
def stop_api_server(process):
|
422 |
+
process.terminate()
|
423 |
+
|
424 |
+
# Register the exit handler
|
425 |
+
api_process = start_api_server()
|
426 |
+
atexit.register(stop_api_server, api_process)
|
427 |
+
|
428 |
+
async def analyze_with_progress(movie_name, progress=gr.Progress()):
|
429 |
+
"""Handle analysis with progress updates in Gradio"""
|
430 |
+
try:
|
431 |
+
async with httpx.AsyncClient(timeout=60.0) as client:
|
432 |
+
# Start the analysis
|
433 |
+
response = await client.get(
|
434 |
+
"http://localhost:8000/api/start_analysis",
|
435 |
+
params={"movie_name": movie_name}
|
436 |
+
)
|
437 |
+
response.raise_for_status()
|
438 |
+
task_id = response.json()["task_id"]
|
439 |
+
|
440 |
+
# Poll for progress
|
441 |
+
while True:
|
442 |
+
progress_response = await client.get(
|
443 |
+
f"http://localhost:8000/api/progress/{task_id}"
|
444 |
+
)
|
445 |
+
progress_response.raise_for_status()
|
446 |
+
status = progress_response.json()
|
447 |
+
|
448 |
+
# Update Gradio progress
|
449 |
+
progress(status["progress"], desc=status["status"])
|
450 |
+
|
451 |
+
if status["is_complete"]:
|
452 |
+
if status["error"]:
|
453 |
+
return f"Error: {status['error']}"
|
454 |
+
elif status["result"]:
|
455 |
+
triggers = status["result"].get("detected_triggers", [])
|
456 |
+
if not triggers or triggers == ["None"]:
|
457 |
+
return "✓ No triggers detected in the content."
|
458 |
+
else:
|
459 |
+
trigger_list = "\n".join([f"• {trigger}" for trigger in triggers])
|
460 |
+
return f"⚠ Triggers Detected:\n{trigger_list}"
|
461 |
+
break
|
462 |
+
|
463 |
+
await asyncio.sleep(0.5)
|
464 |
+
|
465 |
+
except Exception as e:
|
466 |
+
return f"Error: {str(e)}"
|
467 |
|
468 |
def analyze_with_loading(text, progress=gr.Progress()):
|
469 |
"""
|
|
|
508 |
trigger_list = "\n".join([f"• {trigger}" for trigger in triggers])
|
509 |
return f"⚠ Triggers Detected:\n{trigger_list}"
|
510 |
|
511 |
+
# Update the Gradio interface with new styling
|
512 |
+
import gradio as gr
|
513 |
+
from model.analyzer import analyze_content
|
514 |
+
import asyncio
|
515 |
+
import time
|
516 |
+
import httpx
|
517 |
+
import subprocess
|
518 |
+
import atexit
|
519 |
+
|
520 |
+
# Keep your existing CSS and server setup code...
|
521 |
+
# [Previous code until the interface definition remains the same]
|
522 |
+
|
523 |
+
# Update the Gradio interface with fixed button handling
|
524 |
with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as iface:
|
525 |
# Title section
|
526 |
gr.HTML("""
|
527 |
<div class="treat-title">
|
528 |
+
<h1 data-text="TREAT">TREAT</h1>
|
529 |
+
<p>Trigger Recognition for Enjoyable and Appropriate Television</p>
|
|
|
|
|
|
|
|
|
530 |
</div>
|
531 |
""")
|
532 |
|
533 |
+
with gr.Tabs() as tabs:
|
534 |
+
with gr.Tab("Content Analysis"): # Changed from TabItem to Tab
|
535 |
+
with gr.Column():
|
536 |
+
input_text = gr.Textbox(
|
537 |
+
label="ANALYZE CONTENT",
|
538 |
+
placeholder="Enter the content you want to analyze...",
|
539 |
+
lines=8
|
540 |
+
)
|
541 |
+
analyze_btn = gr.Button("✨ Analyze")
|
542 |
+
|
543 |
+
with gr.Tab("Movie Search"): # Changed from TabItem to Tab
|
544 |
+
with gr.Column():
|
545 |
+
search_query = gr.Textbox(
|
546 |
+
label="SEARCH MOVIES",
|
547 |
+
placeholder="Type a movie title to search...",
|
548 |
+
lines=1
|
549 |
+
)
|
550 |
+
search_button = gr.Button("🔍 Search")
|
551 |
|
552 |
+
output_text = gr.Textbox(
|
553 |
+
label="ANALYSIS RESULTS",
|
554 |
+
lines=5,
|
555 |
+
interactive=False
|
556 |
+
)
|
|
|
557 |
|
558 |
+
status_text = gr.Markdown(
|
559 |
+
value=""
|
560 |
+
)
|
|
|
|
|
|
|
|
|
|
|
561 |
|
562 |
+
# Define click events
|
563 |
analyze_btn.click(
|
564 |
fn=analyze_with_loading,
|
565 |
+
inputs=input_text,
|
566 |
+
outputs=output_text
|
567 |
+
)
|
568 |
+
|
569 |
+
search_button.click(
|
570 |
+
fn=analyze_with_progress,
|
571 |
+
inputs=search_query,
|
572 |
+
outputs=output_text
|
573 |
)
|
574 |
|
|
|
575 |
gr.HTML("""
|
576 |
<div class="footer">
|
577 |
<p>Made with <span class="heart">💖</span> by <a href="https://www.linkedin.com/in/kubermehta/" target="_blank">Kuber Mehta</a></p>
|
script_search_api.py
ADDED
@@ -0,0 +1,279 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# script_search_api.py
|
2 |
+
from fastapi import FastAPI, HTTPException
|
3 |
+
from fastapi.middleware.cors import CORSMiddleware
|
4 |
+
import asyncio
|
5 |
+
from datetime import datetime, timedelta
|
6 |
+
from typing import Dict, Optional
|
7 |
+
from pydantic import BaseModel
|
8 |
+
from dataclasses import dataclass
|
9 |
+
import logging
|
10 |
+
import requests
|
11 |
+
from bs4 import BeautifulSoup
|
12 |
+
from difflib import get_close_matches
|
13 |
+
from model.analyzer import analyze_content
|
14 |
+
|
15 |
+
logging.basicConfig(level=logging.INFO)
|
16 |
+
logger = logging.getLogger(__name__)
|
17 |
+
|
18 |
+
app = FastAPI()
|
19 |
+
|
20 |
+
app.add_middleware(
|
21 |
+
CORSMiddleware,
|
22 |
+
allow_origins=["*"],
|
23 |
+
allow_credentials=True,
|
24 |
+
allow_methods=["*"],
|
25 |
+
allow_headers=["*"],
|
26 |
+
)
|
27 |
+
|
28 |
+
@dataclass
|
29 |
+
class ProgressState:
|
30 |
+
progress: float
|
31 |
+
status: str
|
32 |
+
timestamp: datetime
|
33 |
+
task_id: str
|
34 |
+
is_complete: bool = False
|
35 |
+
result: Optional[dict] = None
|
36 |
+
error: Optional[str] = None
|
37 |
+
|
38 |
+
class ProgressResponse(BaseModel):
|
39 |
+
progress: float
|
40 |
+
status: str
|
41 |
+
is_complete: bool
|
42 |
+
result: Optional[dict] = None
|
43 |
+
error: Optional[str] = None
|
44 |
+
|
45 |
+
# Global progress tracker
|
46 |
+
progress_tracker: Dict[str, ProgressState] = {}
|
47 |
+
|
48 |
+
BASE_URL = "https://imsdb.com"
|
49 |
+
ALL_SCRIPTS_URL = f"{BASE_URL}/all-scripts.html"
|
50 |
+
|
51 |
+
def create_task_id(movie_name: str) -> str:
|
52 |
+
"""Create a unique task ID for a movie analysis request"""
|
53 |
+
return f"{movie_name}-{datetime.now().timestamp()}"
|
54 |
+
|
55 |
+
async def cleanup_old_tasks():
|
56 |
+
"""Remove tasks older than 1 hour"""
|
57 |
+
while True:
|
58 |
+
current_time = datetime.now()
|
59 |
+
expired_tasks = [
|
60 |
+
task_id for task_id, state in progress_tracker.items()
|
61 |
+
if current_time - state.timestamp > timedelta(hours=1)
|
62 |
+
]
|
63 |
+
for task_id in expired_tasks:
|
64 |
+
del progress_tracker[task_id]
|
65 |
+
await asyncio.sleep(300) # Cleanup every 5 minutes
|
66 |
+
|
67 |
+
@app.on_event("startup")
|
68 |
+
async def startup_event():
|
69 |
+
"""Initialize the server and start cleanup task"""
|
70 |
+
progress_tracker.clear()
|
71 |
+
asyncio.create_task(cleanup_old_tasks())
|
72 |
+
logger.info("Server started, progress tracker initialized")
|
73 |
+
|
74 |
+
def update_progress(task_id: str, progress: float, status: str, result: Optional[dict] = None, error: Optional[str] = None):
|
75 |
+
"""Update progress state for a task"""
|
76 |
+
is_complete = progress >= 1.0
|
77 |
+
progress_tracker[task_id] = ProgressState(
|
78 |
+
progress=progress,
|
79 |
+
status=status,
|
80 |
+
timestamp=datetime.now(),
|
81 |
+
task_id=task_id,
|
82 |
+
is_complete=is_complete,
|
83 |
+
result=result,
|
84 |
+
error=error
|
85 |
+
)
|
86 |
+
logger.info(f"Task {task_id}: {status} (Progress: {progress * 100:.0f}%)")
|
87 |
+
|
88 |
+
@app.get("/api/start_analysis")
|
89 |
+
async def start_analysis(movie_name: str):
|
90 |
+
"""Start a new analysis task"""
|
91 |
+
task_id = create_task_id(movie_name)
|
92 |
+
update_progress(task_id, 0.0, "Starting analysis...")
|
93 |
+
|
94 |
+
# Start the analysis task in the background
|
95 |
+
asyncio.create_task(run_analysis(task_id, movie_name))
|
96 |
+
|
97 |
+
return {"task_id": task_id}
|
98 |
+
|
99 |
+
@app.get("/api/progress/{task_id}")
|
100 |
+
async def get_progress(task_id: str) -> ProgressResponse:
|
101 |
+
"""Get current progress for a task"""
|
102 |
+
if task_id not in progress_tracker:
|
103 |
+
raise HTTPException(status_code=404, detail="Task not found")
|
104 |
+
|
105 |
+
state = progress_tracker[task_id]
|
106 |
+
return ProgressResponse(
|
107 |
+
progress=state.progress,
|
108 |
+
status=state.status,
|
109 |
+
is_complete=state.is_complete,
|
110 |
+
result=state.result,
|
111 |
+
error=state.error
|
112 |
+
)
|
113 |
+
|
114 |
+
def find_movie_link(movie_name: str, soup: BeautifulSoup) -> str | None:
|
115 |
+
"""Find the closest matching movie link from the script database."""
|
116 |
+
movie_links = {link.text.strip().lower(): link['href'] for link in soup.find_all('a', href=True)}
|
117 |
+
close_matches = get_close_matches(movie_name.lower(), movie_links.keys(), n=1, cutoff=0.6)
|
118 |
+
|
119 |
+
if close_matches:
|
120 |
+
logger.info(f"Close match found: {close_matches[0]}")
|
121 |
+
return BASE_URL + movie_links[close_matches[0]]
|
122 |
+
|
123 |
+
logger.info("No close match found.")
|
124 |
+
return None
|
125 |
+
|
126 |
+
def find_script_link(soup: BeautifulSoup, movie_name: str) -> str | None:
|
127 |
+
"""Find the script download link for a given movie."""
|
128 |
+
patterns = [
|
129 |
+
f'Read "{movie_name}" Script',
|
130 |
+
f'Read "{movie_name.title()}" Script',
|
131 |
+
f'Read "{movie_name.upper()}" Script',
|
132 |
+
f'Read "{movie_name.lower()}" Script'
|
133 |
+
]
|
134 |
+
|
135 |
+
for link in soup.find_all('a', href=True):
|
136 |
+
link_text = link.text.strip()
|
137 |
+
if any(pattern.lower() in link_text.lower() for pattern in patterns):
|
138 |
+
return link['href']
|
139 |
+
elif all(word.lower() in link_text.lower() for word in ["Read", "Script", movie_name]):
|
140 |
+
return link['href']
|
141 |
+
return None
|
142 |
+
|
143 |
+
def fetch_script(movie_name: str) -> str | None:
|
144 |
+
"""Fetch and extract the script content for a given movie."""
|
145 |
+
# Initial page load
|
146 |
+
update_progress(movie_name, 0.1, "Fetching the script database...")
|
147 |
+
try:
|
148 |
+
response = requests.get(ALL_SCRIPTS_URL)
|
149 |
+
response.raise_for_status()
|
150 |
+
except requests.RequestException as e:
|
151 |
+
logger.error(f"Failed to load the main page: {str(e)}")
|
152 |
+
return None
|
153 |
+
|
154 |
+
# Search for movie
|
155 |
+
update_progress(movie_name, 0.2, "Searching for the movie...")
|
156 |
+
soup = BeautifulSoup(response.text, 'html.parser')
|
157 |
+
movie_link = find_movie_link(movie_name, soup)
|
158 |
+
|
159 |
+
if not movie_link:
|
160 |
+
logger.error(f"Script for '{movie_name}' not found.")
|
161 |
+
return None
|
162 |
+
|
163 |
+
# Fetch movie page
|
164 |
+
update_progress(movie_name, 0.3, "Loading movie details...")
|
165 |
+
try:
|
166 |
+
response = requests.get(movie_link)
|
167 |
+
response.raise_for_status()
|
168 |
+
except requests.RequestException as e:
|
169 |
+
logger.error(f"Failed to load the movie page: {str(e)}")
|
170 |
+
return None
|
171 |
+
|
172 |
+
# Find script link
|
173 |
+
update_progress(movie_name, 0.4, "Locating script download...")
|
174 |
+
soup = BeautifulSoup(response.text, 'html.parser')
|
175 |
+
script_link = find_script_link(soup, movie_name)
|
176 |
+
|
177 |
+
if not script_link:
|
178 |
+
logger.error(f"Unable to find script link for '{movie_name}'.")
|
179 |
+
return None
|
180 |
+
|
181 |
+
# Fetch script content
|
182 |
+
script_page_url = BASE_URL + script_link
|
183 |
+
update_progress(movie_name, 0.5, "Downloading script content...")
|
184 |
+
|
185 |
+
try:
|
186 |
+
response = requests.get(script_page_url)
|
187 |
+
response.raise_for_status()
|
188 |
+
except requests.RequestException as e:
|
189 |
+
logger.error(f"Failed to load the script: {str(e)}")
|
190 |
+
return None
|
191 |
+
|
192 |
+
# Extract script text
|
193 |
+
update_progress(movie_name, 0.6, "Extracting script text...")
|
194 |
+
soup = BeautifulSoup(response.text, 'html.parser')
|
195 |
+
script_content = soup.find('pre')
|
196 |
+
|
197 |
+
if script_content:
|
198 |
+
update_progress(movie_name, 0.7, "Script extracted successfully")
|
199 |
+
return script_content.get_text()
|
200 |
+
else:
|
201 |
+
logger.error("Failed to extract script content.")
|
202 |
+
return None
|
203 |
+
|
204 |
+
async def run_analysis(task_id: str, movie_name: str):
|
205 |
+
"""Run the actual analysis task"""
|
206 |
+
try:
|
207 |
+
# Fetch script
|
208 |
+
update_progress(task_id, 0.2, "Fetching script...")
|
209 |
+
script_text = fetch_script(movie_name)
|
210 |
+
if not script_text:
|
211 |
+
raise Exception("Script not found")
|
212 |
+
|
213 |
+
# Analyze content
|
214 |
+
update_progress(task_id, 0.6, "Analyzing content...")
|
215 |
+
result = await analyze_content(script_text)
|
216 |
+
|
217 |
+
# Complete
|
218 |
+
update_progress(task_id, 1.0, "Analysis complete", result=result)
|
219 |
+
|
220 |
+
except Exception as e:
|
221 |
+
logger.error(f"Error in analysis: {str(e)}", exc_info=True)
|
222 |
+
update_progress(task_id, 1.0, "Error occurred", error=str(e))
|
223 |
+
|
224 |
+
@app.get("/api/fetch_and_analyze")
|
225 |
+
async def fetch_and_analyze(movie_name: str):
|
226 |
+
"""Fetch and analyze a movie script, with progress tracking."""
|
227 |
+
try:
|
228 |
+
# Initialize progress
|
229 |
+
task_id = create_task_id(movie_name)
|
230 |
+
update_progress(task_id, 0.0, "Starting script search...")
|
231 |
+
|
232 |
+
# Fetch script
|
233 |
+
script_text = fetch_script(movie_name)
|
234 |
+
if not script_text:
|
235 |
+
raise HTTPException(status_code=404, detail="Script not found or error occurred")
|
236 |
+
|
237 |
+
# Analyze content
|
238 |
+
update_progress(task_id, 0.8, "Analyzing script content...")
|
239 |
+
result = await analyze_content(script_text)
|
240 |
+
|
241 |
+
# Finalize
|
242 |
+
update_progress(task_id, 1.0, "Analysis complete!")
|
243 |
+
return result
|
244 |
+
|
245 |
+
except Exception as e:
|
246 |
+
logger.error(f"Error in fetch_and_analyze: {str(e)}", exc_info=True)
|
247 |
+
# Clean up progress tracker in case of error
|
248 |
+
if movie_name in progress_tracker:
|
249 |
+
del progress_tracker[movie_name]
|
250 |
+
raise HTTPException(status_code=500, detail=f"Internal Server Error: {str(e)}")
|
251 |
+
|
252 |
+
@app.get("/api/progress")
|
253 |
+
def get_progress(movie_name: str):
|
254 |
+
"""Get the current progress and status for a movie analysis."""
|
255 |
+
if movie_name not in progress_tracker:
|
256 |
+
return {
|
257 |
+
"progress": 0,
|
258 |
+
"status": "Waiting to start..."
|
259 |
+
}
|
260 |
+
|
261 |
+
progress_info = progress_tracker[movie_name]
|
262 |
+
|
263 |
+
# Clean up old entries (optional)
|
264 |
+
current_time = datetime.now()
|
265 |
+
if (current_time - progress_info.timestamp).total_seconds() > 3600: # 1 hour timeout
|
266 |
+
del progress_tracker[movie_name]
|
267 |
+
return {
|
268 |
+
"progress": 0,
|
269 |
+
"status": "Session expired. Please try again."
|
270 |
+
}
|
271 |
+
|
272 |
+
return {
|
273 |
+
"progress": progress_info.progress,
|
274 |
+
"status": progress_info.status
|
275 |
+
}
|
276 |
+
|
277 |
+
if __name__ == "__main__":
|
278 |
+
import uvicorn
|
279 |
+
uvicorn.run(app, host="0.0.0.0", port=8000)
|