Update app.py
Browse files
app.py
CHANGED
|
@@ -1209,68 +1209,148 @@ async def translate_caption_interface(base_caption, selected_languages):
|
|
| 1209 |
return result
|
| 1210 |
|
| 1211 |
|
| 1212 |
-
# Replace the CSS section in your create_gradio_app() function with this fixed version:
|
| 1213 |
-
|
| 1214 |
-
# Replace the problematic dropdowns with radio buttons - this completely avoids the positioning issue
|
| 1215 |
-
|
| 1216 |
def create_gradio_app():
|
| 1217 |
-
"""Create the Gradio app with
|
| 1218 |
|
| 1219 |
# Status indicators
|
| 1220 |
hf_status = "โ
Connected" if generator and generator.hf_client_working else "โ ๏ธ Fallback Mode"
|
| 1221 |
sambanova_status = "โ
Connected" if generator and generator.sambanova_client_working else "โ ๏ธ Fallback Mode"
|
| 1222 |
|
| 1223 |
-
#
|
| 1224 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1225 |
|
| 1226 |
-
#
|
| 1227 |
gr.HTML(f"""
|
| 1228 |
-
<
|
| 1229 |
-
|
| 1230 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1231 |
""")
|
| 1232 |
|
| 1233 |
# Main Interface
|
| 1234 |
with gr.Tab("๐ฏ Caption Generator"):
|
| 1235 |
with gr.Row():
|
| 1236 |
# Left Column - Controls
|
| 1237 |
-
with gr.Column(scale=2):
|
| 1238 |
gr.Markdown("### ๐ผ๏ธ Upload Images")
|
|
|
|
| 1239 |
|
| 1240 |
images = gr.File(
|
| 1241 |
label="๐ธ Upload Images (Max 3)",
|
| 1242 |
file_count="multiple",
|
| 1243 |
-
file_types=["image"]
|
|
|
|
| 1244 |
)
|
| 1245 |
|
| 1246 |
gr.Markdown("### โ๏ธ Configuration")
|
| 1247 |
|
| 1248 |
-
|
| 1249 |
-
|
| 1250 |
-
|
| 1251 |
-
|
| 1252 |
-
|
| 1253 |
-
|
| 1254 |
-
|
| 1255 |
-
|
| 1256 |
-
|
| 1257 |
-
|
| 1258 |
-
|
| 1259 |
-
|
| 1260 |
-
|
| 1261 |
-
|
| 1262 |
-
|
| 1263 |
-
|
| 1264 |
-
|
| 1265 |
-
|
| 1266 |
-
|
| 1267 |
-
|
| 1268 |
-
|
| 1269 |
-
|
| 1270 |
-
|
| 1271 |
-
|
| 1272 |
-
|
| 1273 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1274 |
|
| 1275 |
custom_prompt = gr.Textbox(
|
| 1276 |
label="๐ฌ Additional Instructions",
|
|
@@ -1280,37 +1360,42 @@ def create_gradio_app():
|
|
| 1280 |
|
| 1281 |
generate_btn = gr.Button(
|
| 1282 |
"๐ Generate Caption",
|
| 1283 |
-
variant="primary"
|
|
|
|
| 1284 |
)
|
| 1285 |
|
| 1286 |
# Right Column - Results
|
| 1287 |
-
with gr.Column(scale=3):
|
| 1288 |
gr.Markdown("### ๐ Generated Content")
|
| 1289 |
|
| 1290 |
output = gr.Textbox(
|
| 1291 |
label="๐ฏ Generated Caption",
|
| 1292 |
lines=15,
|
|
|
|
| 1293 |
show_copy_button=True,
|
| 1294 |
placeholder="Upload images and generate your Instagram content..."
|
| 1295 |
)
|
| 1296 |
|
| 1297 |
-
|
| 1298 |
-
|
| 1299 |
-
|
| 1300 |
-
|
|
|
|
|
|
|
| 1301 |
|
| 1302 |
alternatives_output = gr.Textbox(
|
| 1303 |
label="โจ Alternative Captions",
|
| 1304 |
lines=15,
|
| 1305 |
show_copy_button=True,
|
| 1306 |
-
placeholder="Generate 3 different caption alternatives..."
|
| 1307 |
)
|
| 1308 |
|
| 1309 |
# Multi-Language Tab
|
| 1310 |
with gr.Tab("๐ Multi-Language"):
|
| 1311 |
with gr.Row():
|
| 1312 |
-
with gr.Column():
|
| 1313 |
gr.Markdown("### ๐ฃ๏ธ Global Content Creation")
|
|
|
|
| 1314 |
|
| 1315 |
base_caption_input = gr.Textbox(
|
| 1316 |
label="๐ Base Caption",
|
|
@@ -1334,7 +1419,7 @@ def create_gradio_app():
|
|
| 1334 |
variant="primary"
|
| 1335 |
)
|
| 1336 |
|
| 1337 |
-
with gr.Column():
|
| 1338 |
multilingual_output = gr.Textbox(
|
| 1339 |
label="๐บ๏ธ Multi-Language Captions",
|
| 1340 |
lines=20,
|
|
@@ -1342,6 +1427,161 @@ def create_gradio_app():
|
|
| 1342 |
placeholder="Culturally adapted captions for global audiences..."
|
| 1343 |
)
|
| 1344 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1345 |
# Event Handlers
|
| 1346 |
generate_btn.click(
|
| 1347 |
fn=generate_advanced_caption_interface,
|
|
@@ -1349,12 +1589,14 @@ def create_gradio_app():
|
|
| 1349 |
outputs=[output, base_caption_input]
|
| 1350 |
)
|
| 1351 |
|
|
|
|
| 1352 |
alternatives_btn.click(
|
| 1353 |
fn=generate_multiple_captions_interface,
|
| 1354 |
inputs=[images, caption_style, target_audience, custom_prompt],
|
| 1355 |
outputs=alternatives_output
|
| 1356 |
)
|
| 1357 |
|
|
|
|
| 1358 |
translate_btn.click(
|
| 1359 |
fn=translate_caption_interface,
|
| 1360 |
inputs=[base_caption_input, language_selector],
|
|
@@ -1363,6 +1605,7 @@ def create_gradio_app():
|
|
| 1363 |
|
| 1364 |
return app
|
| 1365 |
|
|
|
|
| 1366 |
def main():
|
| 1367 |
"""Main function to launch the Instagram Caption Generator"""
|
| 1368 |
print("๐ Starting Instagram Caption Generator...")
|
|
|
|
| 1209 |
return result
|
| 1210 |
|
| 1211 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1212 |
def create_gradio_app():
|
| 1213 |
+
"""Create the Gradio app with good UI styling"""
|
| 1214 |
|
| 1215 |
# Status indicators
|
| 1216 |
hf_status = "โ
Connected" if generator and generator.hf_client_working else "โ ๏ธ Fallback Mode"
|
| 1217 |
sambanova_status = "โ
Connected" if generator and generator.sambanova_client_working else "โ ๏ธ Fallback Mode"
|
| 1218 |
|
| 1219 |
+
# Enhanced CSS for better UI
|
| 1220 |
+
css = """
|
| 1221 |
+
@import url('https://fonts.googleapis.com/css2?family=Space+Grotesk:wght@300;400;500;600;700&display=swap');
|
| 1222 |
+
|
| 1223 |
+
.gradio-container {
|
| 1224 |
+
background: linear-gradient(135deg, #667eea 0%, #764ba2 25%, #f093fb 50%, #f5576c 75%, #4facfe 100%);
|
| 1225 |
+
font-family: 'Space Grotesk', 'Inter', system-ui, sans-serif;
|
| 1226 |
+
min-height: 100vh;
|
| 1227 |
+
}
|
| 1228 |
+
|
| 1229 |
+
.main-header {
|
| 1230 |
+
text-align: center;
|
| 1231 |
+
color: white;
|
| 1232 |
+
margin-bottom: 30px;
|
| 1233 |
+
padding: 30px;
|
| 1234 |
+
background: rgba(255,255,255,0.1);
|
| 1235 |
+
border-radius: 20px;
|
| 1236 |
+
backdrop-filter: blur(20px);
|
| 1237 |
+
border: 1px solid rgba(255,255,255,0.2);
|
| 1238 |
+
box-shadow: 0 20px 40px rgba(0,0,0,0.1);
|
| 1239 |
+
}
|
| 1240 |
+
|
| 1241 |
+
.feature-card {
|
| 1242 |
+
background: rgba(255,255,255,0.1);
|
| 1243 |
+
border-radius: 15px;
|
| 1244 |
+
padding: 20px;
|
| 1245 |
+
backdrop-filter: blur(15px);
|
| 1246 |
+
border: 1px solid rgba(255,255,255,0.2);
|
| 1247 |
+
box-shadow: 0 15px 30px rgba(0,0,0,0.1);
|
| 1248 |
+
margin: 10px 0;
|
| 1249 |
+
}
|
| 1250 |
+
|
| 1251 |
+
.gradio-button-primary {
|
| 1252 |
+
background: linear-gradient(45deg, #ff6b6b, #ee5a24, #ff9ff3, #54a0ff) !important;
|
| 1253 |
+
border: none !important;
|
| 1254 |
+
border-radius: 15px !important;
|
| 1255 |
+
padding: 15px 25px !important;
|
| 1256 |
+
font-weight: 600 !important;
|
| 1257 |
+
color: white !important;
|
| 1258 |
+
box-shadow: 0 10px 20px rgba(255, 107, 107, 0.3) !important;
|
| 1259 |
+
transition: all 0.3s ease !important;
|
| 1260 |
+
}
|
| 1261 |
+
|
| 1262 |
+
.gradio-button-primary:hover {
|
| 1263 |
+
transform: translateY(-2px) !important;
|
| 1264 |
+
box-shadow: 0 15px 30px rgba(255, 107, 107, 0.5) !important;
|
| 1265 |
+
}
|
| 1266 |
+
|
| 1267 |
+
.gradio-button-secondary {
|
| 1268 |
+
background: linear-gradient(45deg, #feca57, #ff9ff3, #54a0ff, #5f27cd) !important;
|
| 1269 |
+
border: none !important;
|
| 1270 |
+
border-radius: 12px !important;
|
| 1271 |
+
padding: 12px 20px !important;
|
| 1272 |
+
font-weight: 600 !important;
|
| 1273 |
+
color: white !important;
|
| 1274 |
+
box-shadow: 0 8px 16px rgba(254, 202, 87, 0.3) !important;
|
| 1275 |
+
}
|
| 1276 |
+
|
| 1277 |
+
.status-badge {
|
| 1278 |
+
background: linear-gradient(90deg, #2ecc71, #27ae60);
|
| 1279 |
+
color: white;
|
| 1280 |
+
padding: 10px 20px;
|
| 1281 |
+
border-radius: 10px;
|
| 1282 |
+
text-align: center;
|
| 1283 |
+
font-weight: 600;
|
| 1284 |
+
box-shadow: 0 8px 16px rgba(46, 204, 113, 0.3);
|
| 1285 |
+
margin: 10px 0;
|
| 1286 |
+
}
|
| 1287 |
+
"""
|
| 1288 |
+
|
| 1289 |
+
with gr.Blocks(css=css, title="๐ฑ Instagram Generator", theme=gr.themes.Glass()) as app:
|
| 1290 |
|
| 1291 |
+
# Main Header
|
| 1292 |
gr.HTML(f"""
|
| 1293 |
+
<div class="main-header">
|
| 1294 |
+
<h1 style="font-size: 2.5rem; margin-bottom: 15px; font-weight: 800; background: linear-gradient(45deg, #ff6b6b, #feca57, #ff9ff3, #54a0ff); -webkit-background-clip: text; -webkit-text-fill-color: transparent;">
|
| 1295 |
+
๐ฑ INSTAGRAM CAPTION GENERATOR
|
| 1296 |
+
</h1>
|
| 1297 |
+
<h2 style="font-size: 1.2rem; margin-bottom: 20px; opacity: 0.9;">
|
| 1298 |
+
๐ AI-Powered Content Creation โข SambaNova + Hugging Face
|
| 1299 |
+
</h2>
|
| 1300 |
+
<div style="display: flex; justify-content: center; gap: 20px; margin-top: 15px;">
|
| 1301 |
+
<span style="background: rgba(255,255,255,0.2); padding: 8px 16px; border-radius: 15px;">๐ค SambaNova: {sambanova_status}</span>
|
| 1302 |
+
<span style="background: rgba(255,255,255,0.2); padding: 8px 16px; border-radius: 15px;">๐ค Hugging Face: {hf_status}</span>
|
| 1303 |
+
</div>
|
| 1304 |
+
</div>
|
| 1305 |
""")
|
| 1306 |
|
| 1307 |
# Main Interface
|
| 1308 |
with gr.Tab("๐ฏ Caption Generator"):
|
| 1309 |
with gr.Row():
|
| 1310 |
# Left Column - Controls
|
| 1311 |
+
with gr.Column(scale=2, elem_classes=["feature-card"]):
|
| 1312 |
gr.Markdown("### ๐ผ๏ธ Upload Images")
|
| 1313 |
+
gr.Markdown("*SambaNova AI vision analysis with quality scoring*")
|
| 1314 |
|
| 1315 |
images = gr.File(
|
| 1316 |
label="๐ธ Upload Images (Max 3)",
|
| 1317 |
file_count="multiple",
|
| 1318 |
+
file_types=["image"],
|
| 1319 |
+
height=200
|
| 1320 |
)
|
| 1321 |
|
| 1322 |
gr.Markdown("### โ๏ธ Configuration")
|
| 1323 |
|
| 1324 |
+
with gr.Row():
|
| 1325 |
+
caption_style = gr.Dropdown(
|
| 1326 |
+
choices=[
|
| 1327 |
+
"๐ฏ Viral Engagement",
|
| 1328 |
+
"๐ผ Professional Brand",
|
| 1329 |
+
"๐ Casual Fun",
|
| 1330 |
+
"๐ Humor & Memes",
|
| 1331 |
+
"๐ช Motivational",
|
| 1332 |
+
"๐ Storytelling",
|
| 1333 |
+
"๐ Luxury Lifestyle",
|
| 1334 |
+
"๐ฅ Trending Culture"
|
| 1335 |
+
],
|
| 1336 |
+
value="๐ฏ Viral Engagement",
|
| 1337 |
+
label="๐จ Caption Style"
|
| 1338 |
+
)
|
| 1339 |
+
|
| 1340 |
+
target_audience = gr.Dropdown(
|
| 1341 |
+
choices=[
|
| 1342 |
+
"๐ General Audience",
|
| 1343 |
+
"๐ผ Business Professionals",
|
| 1344 |
+
"โ๏ธ Travel Enthusiasts",
|
| 1345 |
+
"๐ Food Lovers",
|
| 1346 |
+
"๐ช Fitness Community",
|
| 1347 |
+
"๐ Fashion Forward",
|
| 1348 |
+
"๐ป Tech Innovators",
|
| 1349 |
+
"๐จ Creative Artists"
|
| 1350 |
+
],
|
| 1351 |
+
value="๐ General Audience",
|
| 1352 |
+
label="๐ฅ Target Audience"
|
| 1353 |
+
)
|
| 1354 |
|
| 1355 |
custom_prompt = gr.Textbox(
|
| 1356 |
label="๐ฌ Additional Instructions",
|
|
|
|
| 1360 |
|
| 1361 |
generate_btn = gr.Button(
|
| 1362 |
"๐ Generate Caption",
|
| 1363 |
+
variant="primary",
|
| 1364 |
+
size="lg"
|
| 1365 |
)
|
| 1366 |
|
| 1367 |
# Right Column - Results
|
| 1368 |
+
with gr.Column(scale=3, elem_classes=["feature-card"]):
|
| 1369 |
gr.Markdown("### ๐ Generated Content")
|
| 1370 |
|
| 1371 |
output = gr.Textbox(
|
| 1372 |
label="๐ฏ Generated Caption",
|
| 1373 |
lines=15,
|
| 1374 |
+
max_lines=20,
|
| 1375 |
show_copy_button=True,
|
| 1376 |
placeholder="Upload images and generate your Instagram content..."
|
| 1377 |
)
|
| 1378 |
|
| 1379 |
+
with gr.Row():
|
| 1380 |
+
alternatives_btn = gr.Button(
|
| 1381 |
+
"โจ Generate 3 Alternative Captions",
|
| 1382 |
+
variant="secondary",
|
| 1383 |
+
scale=1
|
| 1384 |
+
)
|
| 1385 |
|
| 1386 |
alternatives_output = gr.Textbox(
|
| 1387 |
label="โจ Alternative Captions",
|
| 1388 |
lines=15,
|
| 1389 |
show_copy_button=True,
|
| 1390 |
+
placeholder="Generate 3 different caption alternatives using Meta-Llama-3.2-3B-Instruct..."
|
| 1391 |
)
|
| 1392 |
|
| 1393 |
# Multi-Language Tab
|
| 1394 |
with gr.Tab("๐ Multi-Language"):
|
| 1395 |
with gr.Row():
|
| 1396 |
+
with gr.Column(elem_classes=["feature-card"]):
|
| 1397 |
gr.Markdown("### ๐ฃ๏ธ Global Content Creation")
|
| 1398 |
+
gr.Markdown("*Powered by Hugging Face Translation Models*")
|
| 1399 |
|
| 1400 |
base_caption_input = gr.Textbox(
|
| 1401 |
label="๐ Base Caption",
|
|
|
|
| 1419 |
variant="primary"
|
| 1420 |
)
|
| 1421 |
|
| 1422 |
+
with gr.Column(elem_classes=["feature-card"]):
|
| 1423 |
multilingual_output = gr.Textbox(
|
| 1424 |
label="๐บ๏ธ Multi-Language Captions",
|
| 1425 |
lines=20,
|
|
|
|
| 1427 |
placeholder="Culturally adapted captions for global audiences..."
|
| 1428 |
)
|
| 1429 |
|
| 1430 |
+
# SambaNova Features Tab
|
| 1431 |
+
with gr.Tab("๐ค SambaNova Features"):
|
| 1432 |
+
gr.HTML(f"""
|
| 1433 |
+
<div class="status-badge">
|
| 1434 |
+
๐ SambaNova Status: {sambanova_status} โข ๐ฆ Llama-4-Maverick + Llama-3.2-3B
|
| 1435 |
+
</div>
|
| 1436 |
+
""")
|
| 1437 |
+
|
| 1438 |
+
with gr.Row():
|
| 1439 |
+
with gr.Column(elem_classes=["feature-card"]):
|
| 1440 |
+
gr.HTML("""
|
| 1441 |
+
<div style="text-align: center; padding: 20px;">
|
| 1442 |
+
<h3>โก SambaNova AI Capabilities</h3>
|
| 1443 |
+
<div style="display: grid; grid-template-columns: repeat(2, 1fr); gap: 15px; margin-top: 20px;">
|
| 1444 |
+
<div style="padding: 15px; background: rgba(46, 204, 113, 0.2); border-radius: 10px; border: 2px solid #2ecc71;">
|
| 1445 |
+
<h4 style="color: #ffffff; margin: 0; text-shadow: 0 0 10px #2ecc71;">Llama-4-Maverick</h4>
|
| 1446 |
+
<p style="margin: 5px 0; color: #e8f8f5; font-weight: 500;">Main Caption Generation</p>
|
| 1447 |
+
</div>
|
| 1448 |
+
<div style="padding: 15px; background: rgba(52, 152, 219, 0.2); border-radius: 10px; border: 2px solid #3498db;">
|
| 1449 |
+
<h4 style="color: #ffffff; margin: 0; text-shadow: 0 0 10px #3498db;">Llama-3.2-3B</h4>
|
| 1450 |
+
<p style="margin: 5px 0; color: #ebf3fd; font-weight: 500;">Caption Variations</p>
|
| 1451 |
+
</div>
|
| 1452 |
+
<div style="padding: 15px; background: rgba(231, 76, 60, 0.2); border-radius: 10px; border: 2px solid #e74c3c;">
|
| 1453 |
+
<h4 style="color: #ffffff; margin: 0; text-shadow: 0 0 10px #e74c3c;">Multi-Modal</h4>
|
| 1454 |
+
<p style="margin: 5px 0; color: #fdedec; font-weight: 500;">Vision + Text Analysis</p>
|
| 1455 |
+
</div>
|
| 1456 |
+
<div style="padding: 15px; background: rgba(155, 89, 182, 0.2); border-radius: 10px; border: 2px solid #9b59b6;">
|
| 1457 |
+
<h4 style="color: #ffffff; margin: 0; text-shadow: 0 0 10px #9b59b6;">Advanced</h4>
|
| 1458 |
+
<p style="margin: 5px 0; color: #f4ecf7; font-weight: 500;">Style & Audience Targeting</p>
|
| 1459 |
+
</div>
|
| 1460 |
+
</div>
|
| 1461 |
+
</div>
|
| 1462 |
+
""")
|
| 1463 |
+
|
| 1464 |
+
with gr.Column(elem_classes=["feature-card"]):
|
| 1465 |
+
gr.Code(
|
| 1466 |
+
value="""
|
| 1467 |
+
# SambaNova API Integration:
|
| 1468 |
+
|
| 1469 |
+
from openai import OpenAI
|
| 1470 |
+
|
| 1471 |
+
client = OpenAI(
|
| 1472 |
+
api_key=os.environ["SAMBANOVA_API_KEY"],
|
| 1473 |
+
base_url="https://api.sambanova.ai/v1",
|
| 1474 |
+
)
|
| 1475 |
+
|
| 1476 |
+
# Main caption generation
|
| 1477 |
+
response = client.chat.completions.create(
|
| 1478 |
+
model="Llama-4-Maverick-17B-128E-Instruct",
|
| 1479 |
+
messages=[{
|
| 1480 |
+
"role": "user",
|
| 1481 |
+
"content": [
|
| 1482 |
+
{"type": "text", "text": "Create Instagram caption"},
|
| 1483 |
+
{"type": "image_url", "image_url": {"url": "data:image/jpeg;base64,..."}}
|
| 1484 |
+
]
|
| 1485 |
+
}],
|
| 1486 |
+
temperature=0.1,
|
| 1487 |
+
top_p=0.1
|
| 1488 |
+
)
|
| 1489 |
+
|
| 1490 |
+
# Caption variations
|
| 1491 |
+
variations = client.chat.completions.create(
|
| 1492 |
+
model="Meta-Llama-3.2-3B-Instruct",
|
| 1493 |
+
messages=[{
|
| 1494 |
+
"role": "user",
|
| 1495 |
+
"content": "Create different version of this caption..."
|
| 1496 |
+
}],
|
| 1497 |
+
temperature=0.9,
|
| 1498 |
+
top_p=0.95
|
| 1499 |
+
)
|
| 1500 |
+
""",
|
| 1501 |
+
language="python",
|
| 1502 |
+
label="๐ง SambaNova Integration Code"
|
| 1503 |
+
)
|
| 1504 |
+
|
| 1505 |
+
# Hugging Face Features Tab
|
| 1506 |
+
with gr.Tab("๐ค Hugging Face Features"):
|
| 1507 |
+
gr.HTML(f"""
|
| 1508 |
+
<div class="status-badge">
|
| 1509 |
+
๐ค Hugging Face Status: {hf_status} โข ๐ Multi-Language Translation
|
| 1510 |
+
</div>
|
| 1511 |
+
""")
|
| 1512 |
+
|
| 1513 |
+
with gr.Row():
|
| 1514 |
+
with gr.Column(elem_classes=["feature-card"]):
|
| 1515 |
+
gr.HTML("""
|
| 1516 |
+
<div style="text-align: center; padding: 20px;">
|
| 1517 |
+
<h3>๐ Translation Models</h3>
|
| 1518 |
+
<div style="display: grid; grid-template-columns: repeat(2, 1fr); gap: 15px; margin-top: 20px;">
|
| 1519 |
+
<div style="padding: 15px; background: rgba(46, 204, 113, 0.2); border-radius: 10px; border: 2px solid #2ecc71;">
|
| 1520 |
+
<h4 style="color: #ffffff; margin: 0; text-shadow: 0 0 10px #2ecc71;">๐ฉ๐ช German</h4>
|
| 1521 |
+
<p style="margin: 5px 0; color: #e8f8f5; font-weight: 500;">google-t5/t5-small</p>
|
| 1522 |
+
</div>
|
| 1523 |
+
<div style="padding: 15px; background: rgba(52, 152, 219, 0.2); border-radius: 10px; border: 2px solid #3498db;">
|
| 1524 |
+
<h4 style="color: #ffffff; margin: 0; text-shadow: 0 0 10px #3498db;">๐จ๐ณ Chinese</h4>
|
| 1525 |
+
<p style="margin: 5px 0; color: #ebf3fd; font-weight: 500;">chence08/mt5-small-iwslt2017-zh-en</p>
|
| 1526 |
+
</div>
|
| 1527 |
+
<div style="padding: 15px; background: rgba(231, 76, 60, 0.2); border-radius: 10px; border: 2px solid #e74c3c;">
|
| 1528 |
+
<h4 style="color: #ffffff; margin: 0; text-shadow: 0 0 10px #e74c3c;">๐ฎ๐ณ Hindi</h4>
|
| 1529 |
+
<p style="margin: 5px 0; color: #fdedec; font-weight: 500;">Helsinki-NLP/opus-mt-en-hi</p>
|
| 1530 |
+
</div>
|
| 1531 |
+
<div style="padding: 15px; background: rgba(155, 89, 182, 0.2); border-radius: 10px; border: 2px solid #9b59b6;">
|
| 1532 |
+
<h4 style="color: #ffffff; margin: 0; text-shadow: 0 0 10px #9b59b6;">๐ธ๐ฆ Arabic</h4>
|
| 1533 |
+
<p style="margin: 5px 0; color: #f4ecf7; font-weight: 500;">marefa-nlp/marefa-mt-en-ar</p>
|
| 1534 |
+
</div>
|
| 1535 |
+
</div>
|
| 1536 |
+
</div>
|
| 1537 |
+
""")
|
| 1538 |
+
|
| 1539 |
+
with gr.Column(elem_classes=["feature-card"]):
|
| 1540 |
+
gr.Code(
|
| 1541 |
+
value="""
|
| 1542 |
+
# Hugging Face Translation Integration:
|
| 1543 |
+
|
| 1544 |
+
from huggingface_hub import InferenceClient
|
| 1545 |
+
|
| 1546 |
+
client = InferenceClient(
|
| 1547 |
+
provider="hf-inference",
|
| 1548 |
+
api_key=os.environ["HF_TOKEN"],
|
| 1549 |
+
)
|
| 1550 |
+
|
| 1551 |
+
# German translation
|
| 1552 |
+
german_result = client.translation(
|
| 1553 |
+
"This is an amazing moment! โจ",
|
| 1554 |
+
model="google-t5/t5-small",
|
| 1555 |
+
)
|
| 1556 |
+
|
| 1557 |
+
# Chinese translation
|
| 1558 |
+
chinese_result = client.translation(
|
| 1559 |
+
"Amazing content for Instagram",
|
| 1560 |
+
model="chence08/mt5-small-iwslt2017-zh-en",
|
| 1561 |
+
)
|
| 1562 |
+
|
| 1563 |
+
# Hindi translation
|
| 1564 |
+
hindi_result = client.translation(
|
| 1565 |
+
"Beautiful content creation",
|
| 1566 |
+
model="Helsinki-NLP/opus-mt-en-hi",
|
| 1567 |
+
)
|
| 1568 |
+
|
| 1569 |
+
# Arabic translation
|
| 1570 |
+
arabic_result = client.translation(
|
| 1571 |
+
"Social media content",
|
| 1572 |
+
model="marefa-nlp/marefa-mt-en-ar",
|
| 1573 |
+
)
|
| 1574 |
+
|
| 1575 |
+
# Features:
|
| 1576 |
+
# โ
4 language models
|
| 1577 |
+
# โ
Fallback support
|
| 1578 |
+
# โ
Instagram-optimized output
|
| 1579 |
+
# โ
Cultural adaptation
|
| 1580 |
+
""",
|
| 1581 |
+
language="python",
|
| 1582 |
+
label="๐ง Hugging Face Translation Code"
|
| 1583 |
+
)
|
| 1584 |
+
|
| 1585 |
# Event Handlers
|
| 1586 |
generate_btn.click(
|
| 1587 |
fn=generate_advanced_caption_interface,
|
|
|
|
| 1589 |
outputs=[output, base_caption_input]
|
| 1590 |
)
|
| 1591 |
|
| 1592 |
+
# Generate multiple alternatives
|
| 1593 |
alternatives_btn.click(
|
| 1594 |
fn=generate_multiple_captions_interface,
|
| 1595 |
inputs=[images, caption_style, target_audience, custom_prompt],
|
| 1596 |
outputs=alternatives_output
|
| 1597 |
)
|
| 1598 |
|
| 1599 |
+
# Multi-language translation
|
| 1600 |
translate_btn.click(
|
| 1601 |
fn=translate_caption_interface,
|
| 1602 |
inputs=[base_caption_input, language_selector],
|
|
|
|
| 1605 |
|
| 1606 |
return app
|
| 1607 |
|
| 1608 |
+
|
| 1609 |
def main():
|
| 1610 |
"""Main function to launch the Instagram Caption Generator"""
|
| 1611 |
print("๐ Starting Instagram Caption Generator...")
|