zaid002 commited on
Commit
e454427
·
verified ·
1 Parent(s): 46d8f59

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -63
app.py DELETED
@@ -1,63 +0,0 @@
1
- import streamlit as st
2
- from deep_translator import GoogleTranslator
3
- from transformers import pipeline
4
- import os
5
-
6
- # Create a local cache directory
7
- os.environ["TRANSFORMERS_CACHE"] = "./models_cache"
8
- os.makedirs("./models_cache", exist_ok=True)
9
-
10
- @st.cache_resource
11
- def load_translator():
12
- def translate_to_english(text, src_lang='auto'):
13
- """Translate any input text to English."""
14
- try:
15
- translated = GoogleTranslator(source=src_lang, target='en').translate(text)
16
- return translated
17
- except Exception as e:
18
- return f"Translation error: {e}"
19
-
20
- def translate_from_english(text, target_lang):
21
- """Translate English response back to user language."""
22
- try:
23
- translated = GoogleTranslator(source='en', target=target_lang).translate(text)
24
- return translated
25
- except Exception as e:
26
- return f"Translation error: {e}"
27
- return translate_to_english, translate_from_english
28
-
29
- @st.cache_resource
30
- def load_chatbot_model():
31
- # Use a small multilingual model (or fine-tuned one from Hugging Face)
32
- chatbot = pipeline("text-generation", model="facebook/blenderbot-400M-distill")
33
- def get_chatbot_response(prompt):
34
- """Generate chatbot reply using a transformer model."""
35
- result = chatbot(prompt, max_length=120, num_return_sequences=1)
36
- return result[0]['generated_text']
37
- return get_chatbot_response
38
-
39
- translate_to_english, translate_from_english = load_translator()
40
- get_chatbot_response = load_chatbot_model()
41
-
42
- st.set_page_config(page_title="GUVI Multilingual Chatbot", layout="centered")
43
- st.title("🤖 GUVI Multilingual Chatbot")
44
- st.write("Chat with AI in your own language!")
45
-
46
- # User selects language
47
- language = st.selectbox("Select your language:",
48
- ["en", "ta", "hi", "ml", "te", "bn", "fr", "es"])
49
-
50
- user_input = st.text_input("You:", "")
51
-
52
- if st.button("Send"):
53
- if user_input:
54
- # Step 1: Translate to English
55
- english_text = translate_to_english(user_input, src_lang=language)
56
-
57
- # Step 2: Get GPT response
58
- english_reply = get_chatbot_response(english_text)
59
-
60
- # Step 3: Translate back
61
- final_reply = translate_from_english(english_reply, target_lang=language)
62
-
63
- st.markdown(f"**Bot:** {final_reply}")