Update app.py

app.py

@@ -52,16 +52,21 @@ llm = HuggingFacePipeline(pipeline=hf_pipeline)
 # ==========================================
 # 3. Setup RAG Chain
 # ==========================================
+
 prompt_template = """
-You are an expert in Unified Power Format (UPF). Generate a precise and complete UPF code block based on the following power intent. Use the retrieved context as a reference. The code must be correct and adhere to UPF 3.0 standards.
+You are a highly capable AI assistant. You possess specialized expertise in Unified Power Format (UPF) and power intent design, but you are also a helpful general conversationalist.
 
-Context:
+Context (Reference UPF examples, if relevant):
 {context}
 
-Power Intent Description:
+User's Input:
 {input}
 
-Generate only the UPF code. Do not include introductory text.
+Instructions:
+1. If the User's Input is asking to generate UPF code or discuss power intent, act as an expert UPF engineer. Use the Context above to generate precise, complete UPF 3.0 code. Enclose the code cleanly in ```tcl ... ``` blocks.
+2. If the User's Input is a general question, a greeting, or unrelated to UPF, respond normally as a helpful AI assistant and ignore the UPF context.
+
+Response:
 """
 prompt = PromptTemplate.from_template(prompt_template)
 document_chain = create_stuff_documents_chain(llm, prompt)
@@ -79,9 +84,6 @@ def generate_upf_code(power_intent_description):
 def user_interaction(user_message, history):
     history = history or []
     response = generate_upf_code(user_message)
-    
-    if "```" not in response:
-        response = f"```tcl\n{response}\n```"
 
     history.append({"role": "user", "content": user_message})
     history.append({"role": "assistant", "content": response})
@@ -93,9 +95,9 @@ with gr.Blocks() as interface:
     chatbot = gr.Chatbot(label="Chat History", elem_id="chatbot")
     
     with gr.Row():
-        user_input = gr.Textbox(show_label=False, placeholder="Enter power intent...", lines=3)
+        user_input = gr.Textbox(show_label=False, placeholder="Enter a general question or a UPF power intent...", lines=3)
         
-    send_button = gr.Button("Generate UPF Code", variant="primary")
+    send_button = gr.Button("Generate Response", variant="primary")
     
     send_button.click(fn=user_interaction, inputs=[user_input, chatbot], outputs=[chatbot, user_input])
     user_input.submit(fn=user_interaction, inputs=[user_input, chatbot], outputs=[chatbot, user_input])