changed streamlit app

app.py

@@ -1,51 +1,25 @@
 import streamlit as st
-
-from transformers import AutoTokenizer, AutoModelForCausalLM
-from transformers import pipeline
-import os
-import logging, sys
-from dotenv import load_dotenv
-
-from huggingface_hub import login
-#load_dotenv()
-
-#HF_TOKEN  = os.environ.get("HF_API_TOKEN")
-HF_TOKEN = st.secrets["HF_API_TOKEN"]
-login(token=HF_TOKEN)
-
-# Setup logging
-logging.basicConfig(stream=sys.stdout, level=logging.INFO)
-logging.getLogger().addHandler(logging.StreamHandler(stream=sys.stdout))
-
- 
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 
 # Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-7B-v0.1")
 model = AutoModelForCausalLM.from_pretrained("mistralai/Mistral-7B-v0.1")
 
-
-
-
-pipe = pipeline("text-generation", 
-                model=model,#"mistralai/Mistral-7B-v0.1",
-                tokenizer=tokenizer,
-                )
-
-# Generate text using the pipeline
-result = pipe("tell me about transformer.", max_length=50, truncation=True)
-print(result)
-
-
-pipe = pipeline("text-generation",  
-                model=model,        #'Mistral-7B-Instruct-v0.2'
-                tokenizer=tokenizer,
-                )
-
-
-
-
-question =st.text_input("enter your question","tell me about transformer.") 
-
-# Generate text using the pipeline
-result = pipe(question, max_length=50, truncation=True)
-print(result)
+# Create text generation pipeline
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
+
+# Continuously prompt the user for questions
+while True:
+    # User input and chat interaction
+    prompt = st.text_input("Please ask your question here (or type 'exit' to quit):")
+    if prompt.lower() == 'exit':
+        break
+
+    # Display user message in chat message container
+    st.markdown("<b style='color:Red;'>User:</b> {}".format(prompt), unsafe_allow_html=True)
+    
+    # Generate response
+    result = pipe(prompt, max_length=50, truncation=True)
+    
+    # Display bot response in chat message container
+    st.markdown("<b style='color:Green;'>Bot:</b> {}".format(result[0]['generated_text']), unsafe_allow_html=True)