Create app.py

app.py

@@ -0,0 +1,76 @@
+import streamlit as st
+from huggingface_hub import InferenceClient
+import os
+
+# Get token from Space secrets
+API_TOKEN = os.getenv("HF_TOKEN_Mistral")
+client = InferenceClient(token=API_TOKEN)
+
+# Initialize session state
+if "chat_history" not in st.session_state:
+    st.session_state.chat_history = []
+if "corrected_sentence" not in st.session_state:
+    st.session_state.corrected_sentence = ""
+
+# Title of the app in Arabic
+st.title("مُحسِّن الجمل والدردشة")
+
+# --- Sentence Correction Section ---
+st.subheader("تحسين الجملة")
+user_input = st.text_input("أدخل جملة لتحسينها:", "ذهبتو إلى الحديقه و العب.")
+
+if st.button("تحسين الجملة"):
+    if user_input:
+        prompt = f"صحح وحسّن هذه الجملة باللغة العربية: '{user_input}'"
+        try:
+            response = client.text_generation(
+                prompt,
+                model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+                max_new_tokens=100,
+                temperature=0.7,
+            )
+            st.session_state.corrected_sentence = response.strip()
+            st.success(f"الجملة المُحسّنة: {st.session_state.corrected_sentence}")
+        except Exception as e:
+            st.error(f"خطأ: {str(e)}")
+    else:
+        st.warning("من فضلك أدخل جملة أولاً!")
+
+# --- Chat Section ---
+st.subheader("الدردشة حول الجملة المُحسّنة")
+
+if st.session_state.corrected_sentence:
+    # Display chat history
+    for speaker, message in st.session_state.chat_history:
+        st.write(f"**{speaker}:** {message}")
+
+    # Define the chat submission function
+    def submit_chat():
+        chat_text = st.session_state.chat_input
+        if chat_text:
+            prompt = (
+                f"الجملة المُحسّنة هي: '{st.session_state.corrected_sentence}'. "
+                f"يسأل المستخدم: '{chat_text}'. أجب بشكل طبيعي وباللغة العربية."
+            )
+            try:
+                response = client.text_generation(
+                    prompt,
+                    model="mistralai/Mixtral-8x7B-Instruct-v0.1",
+                    max_new_tokens=150,
+                    temperature=0.7,
+                )
+                st.session_state.chat_history.append(("أنت", chat_text))
+                st.session_state.chat_history.append(("النموذج", response.strip()))
+                st.session_state.chat_input = ""
+            except Exception as e:
+                st.error(f"حدث خطأ في الدردشة: {str(e)}")
+
+    # Chat input field
+    st.text_input(
+        "اسأل شيئاً حول الجملة المُحسّنة (اضغط Enter للإرسال):",
+        key="chat_input",
+        on_change=submit_chat,
+    )
+
+else:
+    st.write("يرجى تحسين جملة أولاً لبدء الدردشة!")