Update app.py

app.py

@@ -382,25 +382,30 @@ def process_with_gpt(text_input):
 def process_with_claude(text_input):
     """Process text with Claude."""
     if text_input:
-        response = claude_client.messages.create(
-            model="claude-3-sonnet-20240229",
-            max_tokens=1000,
-            messages=[
-                {"role": "user", "content": text_input}
-            ]
-        )
-        response_text = response.content[0].text
-        st.write("Claude: " + response_text)
-        
-        #filename = generate_filename(text_input, "md")
-        filename = generate_filename("GPT-4o: " + response_text, "md")
-        create_file(filename, text_input, response_text)
-        
-        st.session_state.chat_history.append({
-            "user": text_input,
-            "claude": response_text
-        })
-        return response_text
+
+        with st.chat_message("user"):
+            st.markdown(text_input)
+
+        with st.chat_message("assistant"):
+            response = claude_client.messages.create(
+                model="claude-3-sonnet-20240229",
+                max_tokens=1000,
+                messages=[
+                    {"role": "user", "content": text_input}
+                ]
+            )
+            response_text = response.content[0].text
+            st.write("Claude: " + response_text)
+            
+            #filename = generate_filename(text_input, "md")
+            filename = generate_filename("GPT-4o: " + response_text, "md")
+            create_file(filename, text_input, response_text)
+            
+            st.session_state.chat_history.append({
+                "user": text_input,
+                "claude": response_text
+            })
+            return response_text
 
 # File Management Functions
 def load_file(file_name):