Update app.py

app.py

@@ -5,62 +5,60 @@ import os
 import gradio as gr
 import time
 
-custom_prompt_template=""""
+custom_prompt_template = """
 You are an AI coding assistant and your task is to solve coding problems
 and return code snippets based on the user's query. Below is the user's query.
-Query:{query}
+Query: {query}
 You just return the helpful code and related details.
 Helpful code and related details:
 """
 
 def set_custom_prompt():
-    prompt=PromptTemplate(
+    prompt = PromptTemplate(
         template=custom_prompt_template,
         input_variables=['query']
     )
     return prompt
 
 def load_model():
-    llm=CTransformers(
+    llm = CTransformers(
         model='TheBloke/CodeLlama-7B-Instruct-GGML',
         model_type='llama',
         max_new_tokens=1096,
         temperature=0.2,
         repetition_penalty=1.13
-                
     )
-    
     return llm
 
 def chain_pipeline():
-    llm=load_model()
-    qa_prompt=set_custom_prompt()
-    qa_chain=LLMChain(
+    llm = load_model()
+    qa_prompt = set_custom_prompt()
+    qa_chain = LLMChain(
         prompt=qa_prompt,
         llm=llm
     )
     return qa_chain
 
-llmchain=chain_pipeline()
+llmchain = chain_pipeline()
 
 def bot(query):
-    llm_response=llmchain.run({'query':query})
-    return llm_response
+    llm_response = llmchain.run({'query': query})
+    # Wrap the response in triple backticks for code formatting
+    formatted_response = f"```\n{llm_response}\n```"
+    return formatted_response
 
-with gr.Blocks(title="Can AI code ? ") as demo:
+with gr.Blocks(title="Can AI code ?") as demo:
     gr.Markdown('# Code LLAMA demo')
-    chatbot=gr.Chatbot([],elem_id='chatbot',height=700)
-    msg=gr.Textbox()
-    clear=gr.ClearButton([msg,chatbot])
-    
-    
-    def respond(message,chat_history):
-        bot_message=bot(message)
-        chat_history.append((message,bot_message))
+    chatbot = gr.Chatbot([], elem_id='chatbot', height=700)
+    msg = gr.Textbox()
+    clear = gr.ClearButton([msg, chatbot])
+
+    def respond(message, chat_history):
+        bot_message = bot(message)
+        chat_history.append((message, bot_message))
         time.sleep(2)
-        return "",chat_history
-    
-    msg.submit(respond,[msg,chatbot],[msg,chatbot])
-    
+        return "", chat_history
+
+    msg.submit(respond, [msg, chatbot], [msg, chatbot])
 
-demo.launch()
+demo.launch()