first commit of the app

Dockerfile

@@ -0,0 +1,18 @@
+# Use the official Python 3.11.6 image
+FROM python:3.11.6
+
+# Install Streamlit and other dependencies
+COPY requirements.txt .
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+# Copy the rest of the app
+COPY . .
+
+# Expose the port (optionnel, mais propre)
+EXPOSE 7860
+
+# Set the working directory
+WORKDIR /
+
+# Start the Streamlit app
+CMD ["streamlit", "run", "chat-streamlit-app.py", "--server.port", "7860", "--server.address", "0.0.0.0"]

chat-streamlit-app.py

@@ -0,0 +1,79 @@
+import streamlit as st
+import logging
+import time
+from transformers import pipeline
+
+logging.basicConfig(level=logging.INFO)
+
+# Dictionnaire des modèles disponibles
+#     "Meta-LLaMA 3 8B": "meta-llama/Meta-Llama-3-8B-Instruct",
+MODEL_DICT = {
+    "DistilGPT-2": "distilgpt2",
+    "Falcon3-7B-Base": "tiiuae/Falcon3-7B-Base"
+}
+
+# Initialize chat history
+if 'messages' not in st.session_state:
+    st.session_state.messages = []
+
+if 'current_model' not in st.session_state:
+    st.session_state.current_model = None
+
+def load_generator(model_name):
+    return pipeline("text-generation", model=model_name)
+
+def stream_chat(generator, messages):
+    try:
+        prompt = "\n".join([f"{msg['role']}: {msg['content']}" for msg in messages])
+        output = generator(prompt, max_length=512, do_sample=True, temperature=0.7)
+        response = output[0]['generated_text']
+        return response
+    except Exception as e:
+        logging.error(f"Error during generation: {str(e)}")
+        raise e
+
+def main():
+    st.title("Chat with Multiple LLMs")
+    logging.info("App started")
+
+    # Sidebar for model selection
+    selected_model_key = st.sidebar.selectbox("Choose a model", list(MODEL_DICT.keys()))
+    selected_model_name = MODEL_DICT[selected_model_key]
+
+    # Load generator only if model changed
+    if st.session_state.current_model != selected_model_name:
+        with st.spinner(f"Loading model {selected_model_key}..."):
+            st.session_state.generator = load_generator(selected_model_name)
+        st.session_state.current_model = selected_model_name
+        st.success(f"Model {selected_model_key} loaded!")
+
+    if prompt := st.chat_input("Your question"):
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        logging.info(f"User input: {prompt}")
+
+        # Display conversation
+        for message in st.session_state.messages:
+            with st.chat_message(message["role"]):
+                st.write(message["content"])
+
+        if st.session_state.messages[-1]["role"] == "user":
+            with st.chat_message("assistant"):
+                start_time = time.time()
+                logging.info("Generating response")
+
+                with st.spinner("Writing..."):
+                    try:
+                        response_message = stream_chat(st.session_state.generator, st.session_state.messages)
+                        duration = time.time() - start_time
+                        response_with_duration = f"{response_message}\n\n⏱ Duration: {duration:.2f} seconds"
+                        st.session_state.messages.append({"role": "assistant", "content": response_with_duration})
+                        st.write(response_with_duration)
+                        logging.info(f"Response generated in {duration:.2f}s")
+
+                    except Exception as e:
+                        st.session_state.messages.append({"role": "assistant", "content": str(e)})
+                        st.error("An error occurred while generating the response.")
+                        logging.error(f"Error: {str(e)}")
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,109 @@
+aiohappyeyeballs==2.6.1
+aiohttp==3.11.18
+aiosignal==1.3.2
+altair==5.5.0
+annotated-types==0.7.0
+anyio==4.9.0
+attrs==25.3.0
+banks==2.1.2
+beautifulsoup4==4.13.4
+blinker==1.9.0
+cachetools==5.5.2
+certifi==2025.4.26
+charset-normalizer==3.4.2
+click==8.1.8
+colorama==0.4.6
+dataclasses-json==0.6.7
+Deprecated==1.2.18
+dirtyjson==1.0.8
+distro==1.9.0
+filelock==3.18.0
+filetype==1.2.0
+frozenlist==1.6.0
+fsspec==2025.3.2
+gitdb==4.0.12
+GitPython==3.1.44
+greenlet==3.2.1
+griffe==1.7.3
+h11==0.16.0
+hf-xet==1.1.0
+httpcore==1.0.9
+httpx==0.28.1
+huggingface-hub==0.31.1
+idna==3.10
+Jinja2==3.1.6
+jiter==0.9.0
+joblib==1.5.0
+jsonschema==4.23.0
+jsonschema-specifications==2025.4.1
+llama-cloud==0.1.19
+llama-cloud-services==0.6.21
+llama-index==0.12.34
+llama-index-agent-openai==0.4.7
+llama-index-cli==0.4.1
+llama-index-core==0.12.34.post1
+llama-index-embeddings-openai==0.3.1
+llama-index-indices-managed-llama-cloud==0.6.11
+llama-index-llms-ollama==0.5.4
+llama-index-llms-openai==0.3.38
+llama-index-multi-modal-llms-openai==0.4.3
+llama-index-program-openai==0.3.1
+llama-index-question-gen-openai==0.3.0
+llama-index-readers-file==0.4.7
+llama-index-readers-llama-parse==0.4.0
+llama-parse==0.6.21
+MarkupSafe==3.0.2
+marshmallow==3.26.1
+mpmath==1.3.0
+multidict==6.4.3
+mypy_extensions==1.1.0
+narwhals==1.38.0
+nest-asyncio==1.6.0
+networkx==3.4.2
+nltk==3.9.1
+numpy==2.2.5
+ollama==0.4.8
+openai==1.77.0
+packaging==24.2
+pandas==2.2.3
+pillow==11.2.1
+platformdirs==4.3.8
+propcache==0.3.1
+protobuf==6.30.2
+pyarrow==20.0.0
+pydantic==2.11.4
+pydantic_core==2.33.2
+pydeck==0.9.1
+pypdf==5.4.0
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.0
+pytz==2025.2
+PyYAML==6.0.2
+referencing==0.36.2
+regex==2024.11.6
+requests==2.32.3
+rpds-py==0.24.0
+safetensors==0.5.3
+six==1.17.0
+smmap==5.0.2
+sniffio==1.3.1
+soupsieve==2.7
+SQLAlchemy==2.0.40
+streamlit==1.45.0
+striprtf==0.0.26
+sympy==1.14.0
+tenacity==9.1.2
+tiktoken==0.9.0
+tokenizers==0.21.1
+toml==0.10.2
+torch==2.7.0
+tornado==6.4.2
+tqdm==4.67.1
+transformers==4.51.3
+typing-inspect==0.9.0
+typing-inspection==0.4.0
+typing_extensions==4.13.2
+tzdata==2025.2
+urllib3==2.4.0
+wrapt==1.17.2
+yarl==1.20.0