Prompt change

SkinGPT.py

@@ -10,6 +10,8 @@ import torch
 MODEL_DTYPE = torch.float16 if torch.cuda.is_available() else torch.float32
 token = os.getenv("HF_TOKEN")
 import streamlit as st
+import re
+
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 class Blip2QFormer(nn.Module):
@@ -250,8 +252,16 @@ class SkinGPT4(nn.Module):
 
         response = response.strip('"').strip()
 
-        print("Processed response:", response)  # Debug print
-        return response
+        cleaned = re.sub(r'^[ \t]+', '', response, flags=re.MULTILINE)
+
+        # Remove excessive blank lines (keep at most one blank line between paragraphs)
+        cleaned = re.sub(r'\n\s*\n', '\n\n', cleaned)
+
+        print(cleaned)
+        return cleaned
+
+        # print("Processed response:", response)  # Debug print
+        # return response
         # print("Split parts:", full_output.split("### Response:"))
         # # response = full_output.split("### Response:")[-1].strip()
         #

app.py

@@ -103,8 +103,8 @@ if uploaded_file is not None and uploaded_file != st.session_state.current_image
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         # st.markdown(remove_code_blocks(message["content"]))
-        # st.markdown(message["content"])
-        st.text(message["content"])
+        st.markdown(message["content"])
+        # st.text(message["content"])
 
 # for message in st.session_state.messages:
 #     role = "You" if message["role"] == "user" else "assistant"
@@ -135,8 +135,8 @@ if prompt := st.chat_input("Ask a follow-up question..."):
                 result = classifier.predict(image, user_input=prompt, reuse_embeddings=False)
 
             # st.markdown(remove_code_blocks(result["diagnosis"]))
-            # st.markdown(result["diagnosis"])
-            st.text(result["diagnosis"])
+            st.markdown(result["diagnosis"])
+            # st.text(result["diagnosis"])
             st.session_state.messages.append({"role": "assistant", "content": result["diagnosis"]})
 
 if st.session_state.messages and st.button("📄 Download Chat as PDF"):