gpaasch commited on
Commit
932f0ad
·
1 Parent(s): d6d864a
Files changed (1) hide show
  1. src/app.py +12 -14
src/app.py CHANGED
@@ -1,9 +1,18 @@
1
  import os
2
  import gradio as gr
3
- from llama_index.llms.huggingface import HuggingFaceLLM
4
- from parse_tabular import symptom_index
 
5
  import json
6
 
 
 
 
 
 
 
 
 
7
  # --- System prompt ---
8
  SYSTEM_PROMPT = """
9
  You are a medical assistant helping a user narrow down to the most likely ICD-10 code.
@@ -47,20 +56,9 @@ with gr.Blocks() as demo:
47
 
48
  microphone = gr.Microphone(
49
  streaming=True,
50
- type="filepath" # Return transcriptions as strings
51
  )
52
 
53
- # MCP metadata (store in a separate variable, not as an attribute of demo)
54
- mcp_metadata = {
55
- "title": "Medical Symptom to ICD-10 Code Assistant",
56
- "description": "Convert spoken symptoms to ICD-10 codes",
57
- "version": "1.0.0",
58
- "capabilities": {
59
- "speech_input": True,
60
- "streaming": True
61
- }
62
- }
63
-
64
  # Connect the streaming microphone to the chat
65
  microphone.stream(
66
  fn=process_speech,
 
1
  import os
2
  import gradio as gr
3
+ from llama_index.core import Settings
4
+ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
5
+ from parse_tabular import create_symptom_index # Change this import
6
  import json
7
 
8
+ # Configure embeddings globally
9
+ Settings.embed_model = HuggingFaceEmbedding(
10
+ model_name="sentence-transformers/all-MiniLM-L6-v2"
11
+ )
12
+
13
+ # Create the index at startup
14
+ symptom_index = create_symptom_index()
15
+
16
  # --- System prompt ---
17
  SYSTEM_PROMPT = """
18
  You are a medical assistant helping a user narrow down to the most likely ICD-10 code.
 
56
 
57
  microphone = gr.Microphone(
58
  streaming=True,
59
+ type="filepath" # Use 'filepath' to get the path to the recorded audio file
60
  )
61
 
 
 
 
 
 
 
 
 
 
 
 
62
  # Connect the streaming microphone to the chat
63
  microphone.stream(
64
  fn=process_speech,