Upload 2 files

app.py

@@ -13,10 +13,21 @@ import uuid
 
 from huggingface_hub import HfApi, snapshot_download
 
+from gradio.themes.utils import colors, fonts, sizes
+from gradio.themes import Base
+
+# Create a custom theme with larger text
+large_text_theme = Base(
+    # Increase all font sizes by ~25%
+    font=[fonts.GoogleFont("Roboto"), "ui-sans-serif", "sans-serif"],
+    font_mono=[fonts.GoogleFont("IBM Plex Mono"), "ui-monospace", "monospace"],
+    text_size=sizes.text_lg,     # Base text size (default is text_md)
+    radius_size=sizes.radius_md,
+)
 
 # Configuration
-VECTOR_STORE_DIR = "./vector_stores"
-EMBED_MODEL_PATH = "./datas/bge_onnx"
+# VECTOR_STORE_DIR = "./vector_stores"
+# EMBED_MODEL_PATH = "./datas/bge_onnx"
 CONFIG_PATH = "config.json"
 
 DEFAULT_LLM = "Jatin19K/unsloth-q5_k_m-mistral-nemo-instruct-2407"
@@ -45,7 +56,7 @@ local_dir = snapshot_download(
         )
 
 EMBED_MODEL_PATH = os.path.join(local_dir, "bge_onnx")
-# VECTOR_STORE_DIR = os.path.join(local_dir, "vector_stores")
+VECTOR_STORE_DIR = os.path.join(local_dir, "vector_stores")
 
 class ModelManager:
     def __init__(self):
@@ -239,11 +250,11 @@ def chat_response(message, history, llm_choice, vector_store_choice, session_sta
         
         # bot_message = f"{response.response}\n\nSources:\n" + "\n".join(sources)
         bot_message = f"{response.response}\n"
-        return history + [(message, bot_message)], session_state
+        return history + [(message, bot_message)], session_state, ""
         # return history + [(message)], session_state
     
     except Exception as e:
-        return history + [(message, f"Error: {str(e)}")], session_state
+        return history + [(message, f"Error: {str(e)}")], session_state, ""
 
 # Gradio interface with embedding status
 with gr.Blocks(title="De-KCIB(Deep Knowledge Center for Injury Biomechanics)") as demo:
@@ -258,7 +269,8 @@ with gr.Blocks(title="De-KCIB(Deep Knowledge Center for Injury Biomechanics)") a
             #      alt="Company Logo" 
             #      style="height: 100px; object-fit: contain;">
             # """)
-        gr.HTML("<img src='https://www.ussbchamber.org/wp-content/uploads/2021/04/innovisionlogo.png' />")
+        gr.Markdown("<img src='/gradio_api/file/logo.png' alt='Innovision Logo' height='150' width='390'>")
+        # gr.HTML("<img src='https://www.ussbchamber.org/wp-content/uploads/2021/04/innovisionlogo.png' />")
             # gr.Markdown("<img src='file/logo.png' alt='Company Logo' />")
     with gr.Row():
         gr.Markdown("# De-KCIB(Deep Knowledge Center for Injury Biomechanics)")
@@ -299,7 +311,7 @@ with gr.Blocks(title="De-KCIB(Deep Knowledge Center for Injury Biomechanics)") a
     msg.submit(
         chat_response,
         [msg, chatbot, llm_dropdown, vector_dropdown, session_state],
-        [chatbot, session_state]  # <-- Update outputs
+        [chatbot, session_state, msg]  # <-- Update outputs
     )
     
     clear_btn.click(
@@ -311,5 +323,5 @@ with gr.Blocks(title="De-KCIB(Deep Knowledge Center for Injury Biomechanics)") a
 
 # Deployment settings
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(allowed_paths=["logo.png"])
     # demo.launch(share=True)