Update app.py

app.py

@@ -1,4 +1,11 @@
-import time, os, multiprocessing, torch, requests, asyncio, json, aiohttp
+import time
+import os
+import multiprocessing
+import torch
+import requests
+import asyncio
+import json
+import aiohttp
 from minivectordb.embedding_model import EmbeddingModel
 from minivectordb.vector_database import VectorDatabase
 from text_util_en_pt.cleaner import structurize_text, detect_language, Language
@@ -24,12 +31,12 @@ def index_and_search(query, text):
 
     # Indexing
     vector_db = VectorDatabase()
-    sentences = [ s['sentence'] for s in structurize_text(text)]
+    sentences = [s['sentence'] for s in structurize_text(text)]
 
     for idx, sentence in enumerate(sentences):
         sentence_embedding = model.extract_embeddings(sentence)
         vector_db.store_embedding(idx + 1, sentence_embedding, {'sentence': sentence})
-    
+
     embedding_time = time.time() - start
 
     # Retrieval
@@ -168,10 +175,9 @@ async def predict(message, history):
 
 # Define a custom theme with colors similar to the provided image
 custom_theme = gr.themes.Default(
-    primary_hue="#6a0dad",  # Custom primary color (similar to the purple in the image)
-    secondary_hue="#4b0082",  # Custom secondary color
+    primary_hue="purple",  # Custom primary color (similar to the purple in the image)
+    secondary_hue="indigo",  # Custom secondary color
     font="Roboto",  # Custom font
-    font_size="16px",  # Custom font size
     rounded_corners=True,  # Enable rounded corners
     background_color="#1a1a2e",  # Custom dark background color
     text_color="#f0f0f0"  # Custom text color for better contrast
@@ -180,13 +186,13 @@ custom_theme = gr.themes.Default(
 # Apply the custom theme to the Gradio interface
 gr.ChatInterface(
     predict,
-    title="Assistant pro",
+    title="Assistant Pro",
     retry_btn=None,
     undo_btn=None,
     examples=[
         'What is the current sentiment of the Brazil election?',
         'Compare the current economies of China and India?',
-        'What are new shoe design trends in 2024',
+        'What are new shoe design trends in 2024?',
     ],
     theme=custom_theme  # Apply the custom theme
-).launch()
+).launch()