karthikeyan-r commited on
Commit
171f684
·
verified ·
1 Parent(s): 402b346

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -76,6 +76,15 @@ if load_model_button:
76
  st.session_state["tokenizer"] = tokenizer
77
  st.session_state["qa_pipeline"] = qa_pipe
78
 
 
 
 
 
 
 
 
 
 
79
  st.success("Model loaded successfully and ready!")
80
  except Exception as e:
81
  st.error(f"Error loading model: {e}")
@@ -93,8 +102,7 @@ if clear_conversation_button:
93
  st.success("Conversation cleared.")
94
 
95
  # ----- Title -----
96
- st.title("Chat Conversation UI")
97
-
98
 
99
  user_input = None # We'll collect it below
100
 
 
76
  st.session_state["tokenizer"] = tokenizer
77
  st.session_state["qa_pipeline"] = qa_pipe
78
 
79
+ # ---------------------------
80
+ # Add a welcome message only once, if conversation is empty
81
+ # ---------------------------
82
+ if len(st.session_state["conversation"]) == 0:
83
+ st.session_state["conversation"].append({
84
+ "role": "assistant",
85
+ "content": "Hello! I’m your assistant. How can I help you today?"
86
+ })
87
+
88
  st.success("Model loaded successfully and ready!")
89
  except Exception as e:
90
  st.error(f"Error loading model: {e}")
 
102
  st.success("Conversation cleared.")
103
 
104
  # ----- Title -----
105
+ st.title("ChatBot")
 
106
 
107
  user_input = None # We'll collect it below
108