Create app.py

app.py

@@ -0,0 +1,79 @@
+import os
+import streamlit as st
+import json
+import anthropic
+
+# Configuration de l'API
+api_key = os.environ.get("API_KEY")
+client = anthropic.Anthropic(api_key=api_key)
+
+def chatbot_interface():
+    st.title("Chatbot Interface")
+
+    # Choix du modèle
+    model_options = [
+        "claude-3-opus-20240229",
+        "claude-3-sonnet-20240229",
+        "claude-3-haiku-20240307"
+    ]
+    model_selected = st.sidebar.selectbox("Choisissez un modèle de Claude :", model_options)
+    if "ai_model" not in st.session_state or st.session_state["ai_model"] != model_selected:
+        st.session_state["ai_model"] = model_selected
+
+    # Initialisation de l'état de session
+    if "messages" not in st.session_state:
+        st.session_state.messages = []
+
+    # Charger l'historique
+    st.sidebar.title("Historique")
+    uploaded_file = st.sidebar.file_uploader("Chargez un fichier historique JSON")
+    if uploaded_file is not None:
+        st.session_state.messages = json.load(uploaded_file)
+
+    # Afficher les messages
+    for message in st.session_state.messages:
+        with st.chat_message(message["role"]):
+            st.markdown(message["content"])
+
+    # Entrée utilisateur
+    if prompt := st.chat_input("What is up?"):
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        # Appel API pour obtenir la réponse de l'assistant
+        with st.chat_message("assistant"):
+            message_placeholder = st.empty()  # Placeholder pour les réponses progressives
+            full_response = ""
+
+            # Préparation de la requête à envoyer à Claude
+            with client.messages.stream(
+                max_tokens=1024,
+                messages=[{"role": m["role"], "content": m["content"]} for m in st.session_state.messages],
+                model=st.session_state["ai_model"]
+            ) as stream:
+                for text in stream.text_stream:
+                    full_response += str(text) if text is not None else ""
+                    message_placeholder.markdown(full_response + "▌")  # Afficher la réponse progressive avec un curseur
+
+            message_placeholder.markdown(full_response)  # Afficher la réponse finale
+        st.session_state.messages.append({"role": "assistant", "content": full_response})
+
+    # Télécharger l'historique
+    if st.button("Télécharger l'historique"):
+        json_history = json.dumps(st.session_state.messages, indent=4)
+        st.download_button(
+            label="Télécharger l'historique",
+            data=json_history,
+            file_name="chat_history.json",
+            mime="application/json"
+        )
+
+def main():
+    logo_path = "logo.png"  # Remplacez 'logo.png' par le chemin réel vers votre image
+    st.sidebar.image(logo_path, width=150)  # Ajustez la largeur selon vos besoins
+
+    chatbot_interface()
+
+if __name__ == "__main__":
+    main()