Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ client = InferenceClient(
|
|
12 |
token=os.getenv("token")
|
13 |
)
|
14 |
|
15 |
-
# Chatbot response function
|
16 |
def respond(
|
17 |
message,
|
18 |
history: list[tuple[str, str]],
|
@@ -31,19 +31,14 @@ def respond(
|
|
31 |
}
|
32 |
|
33 |
messages = [system_message]
|
34 |
-
|
35 |
-
# Add chat history
|
36 |
for val in history:
|
37 |
if val[0]:
|
38 |
messages.append({"role": "user", "content": val[0]})
|
39 |
if val[1]:
|
40 |
messages.append({"role": "assistant", "content": val[1]})
|
41 |
-
|
42 |
-
# Add current message
|
43 |
messages.append({"role": "user", "content": message})
|
44 |
response = ""
|
45 |
|
46 |
-
# Streaming safe decoding
|
47 |
for message_chunk in client.chat_completion(
|
48 |
messages,
|
49 |
max_tokens=max_tokens,
|
@@ -65,39 +60,238 @@ def respond(
|
|
65 |
|
66 |
print(f"{datetime.datetime.now()}::{messages[-1]['content']}->{response}\n")
|
67 |
|
68 |
-
|
69 |
# Encode image to base64
|
70 |
def encode_image(image_path):
|
71 |
with open(image_path, "rb") as image_file:
|
72 |
return base64.b64encode(image_file.read()).decode('utf-8')
|
73 |
|
74 |
-
|
75 |
# Load and encode logos
|
76 |
github_logo_encoded = encode_image("Images/github-logo.png")
|
77 |
linkedin_logo_encoded = encode_image("Images/linkedin-logo.png")
|
78 |
website_logo_encoded = encode_image("Images/ai-logo.png")
|
79 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
80 |
# Gradio interface
|
81 |
-
with gr.Blocks(
|
82 |
-
gr.
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
94 |
)
|
95 |
-
gr.Markdown(md.description)
|
96 |
|
97 |
-
|
98 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
99 |
|
100 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
101 |
|
102 |
if __name__ == "__main__":
|
103 |
main.launch(share=True)
|
|
|
12 |
token=os.getenv("token")
|
13 |
)
|
14 |
|
15 |
+
# Chatbot response function (unchanged)
|
16 |
def respond(
|
17 |
message,
|
18 |
history: list[tuple[str, str]],
|
|
|
31 |
}
|
32 |
|
33 |
messages = [system_message]
|
|
|
|
|
34 |
for val in history:
|
35 |
if val[0]:
|
36 |
messages.append({"role": "user", "content": val[0]})
|
37 |
if val[1]:
|
38 |
messages.append({"role": "assistant", "content": val[1]})
|
|
|
|
|
39 |
messages.append({"role": "user", "content": message})
|
40 |
response = ""
|
41 |
|
|
|
42 |
for message_chunk in client.chat_completion(
|
43 |
messages,
|
44 |
max_tokens=max_tokens,
|
|
|
60 |
|
61 |
print(f"{datetime.datetime.now()}::{messages[-1]['content']}->{response}\n")
|
62 |
|
|
|
63 |
# Encode image to base64
|
64 |
def encode_image(image_path):
|
65 |
with open(image_path, "rb") as image_file:
|
66 |
return base64.b64encode(image_file.read()).decode('utf-8')
|
67 |
|
|
|
68 |
# Load and encode logos
|
69 |
github_logo_encoded = encode_image("Images/github-logo.png")
|
70 |
linkedin_logo_encoded = encode_image("Images/linkedin-logo.png")
|
71 |
website_logo_encoded = encode_image("Images/ai-logo.png")
|
72 |
|
73 |
+
# Custom CSS for enhanced UI
|
74 |
+
custom_css = """
|
75 |
+
/* General styling */
|
76 |
+
body {
|
77 |
+
background-color: #f5f7fa;
|
78 |
+
}
|
79 |
+
|
80 |
+
/* Chatbot styling */
|
81 |
+
.chatbot .message {
|
82 |
+
border-radius: 12px;
|
83 |
+
padding: 12px;
|
84 |
+
margin: 8px 0;
|
85 |
+
box-shadow: 0 2px 4px rgba(0,0,0,0.1);
|
86 |
+
}
|
87 |
+
.chatbot .user-message {
|
88 |
+
background-color: #007bff;
|
89 |
+
color: white;
|
90 |
+
}
|
91 |
+
.chatbot .bot-message {
|
92 |
+
background-color: #e9ecef;
|
93 |
+
color: #333;
|
94 |
+
}
|
95 |
+
|
96 |
+
/* Tab styling */
|
97 |
+
.tab-nav button {
|
98 |
+
font-weight: 600;
|
99 |
+
transition: all 0.3s ease;
|
100 |
+
}
|
101 |
+
.tab-nav button:hover {
|
102 |
+
background-color: #007bff;
|
103 |
+
color: white;
|
104 |
+
}
|
105 |
+
|
106 |
+
/* Footer styling */
|
107 |
+
.footer {
|
108 |
+
display: flex;
|
109 |
+
justify-content: center;
|
110 |
+
gap: 20px;
|
111 |
+
padding: 20px;
|
112 |
+
background-color: #343a40;
|
113 |
+
border-radius: 8px;
|
114 |
+
}
|
115 |
+
.footer img {
|
116 |
+
width: 32px;
|
117 |
+
height: 32px;
|
118 |
+
transition: transform 0.3s ease;
|
119 |
+
}
|
120 |
+
.footer img:hover {
|
121 |
+
transform: scale(1.2);
|
122 |
+
}
|
123 |
+
|
124 |
+
/* Sidebar styling */
|
125 |
+
.sidebar {
|
126 |
+
background-color: #ffffff;
|
127 |
+
padding: 20px;
|
128 |
+
border-right: 1px solid #dee2e6;
|
129 |
+
box-shadow: 2px 0 5px rgba(0,0,0,0.1);
|
130 |
+
}
|
131 |
+
|
132 |
+
/* Responsive design */
|
133 |
+
@media (max-width: 768px) {
|
134 |
+
.chatbot {
|
135 |
+
height: 60vh !important;
|
136 |
+
}
|
137 |
+
.footer img {
|
138 |
+
width: 24px;
|
139 |
+
height: 24px;
|
140 |
+
}
|
141 |
+
}
|
142 |
+
|
143 |
+
/* Loading spinner */
|
144 |
+
.spinner {
|
145 |
+
border: 4px solid #f3f3f3;
|
146 |
+
border-top: 4px solid #007bff;
|
147 |
+
border-radius: 50%;
|
148 |
+
width: 24px;
|
149 |
+
height: 24px;
|
150 |
+
animation: spin 1s linear infinite;
|
151 |
+
margin: 10px auto;
|
152 |
+
}
|
153 |
+
@keyframes spin {
|
154 |
+
0% { transform: rotate(0deg); }
|
155 |
+
100% { transform: rotate(360deg); }
|
156 |
+
}
|
157 |
+
|
158 |
+
/* Download button */
|
159 |
+
.download-btn {
|
160 |
+
background-color: #28a745;
|
161 |
+
color: white;
|
162 |
+
padding: 10px 20px;
|
163 |
+
border-radius: 8px;
|
164 |
+
text-decoration: none;
|
165 |
+
transition: background-color 0.3s ease;
|
166 |
+
}
|
167 |
+
.download-btn:hover {
|
168 |
+
background-color: #218838;
|
169 |
+
}
|
170 |
+
"""
|
171 |
+
|
172 |
# Gradio interface
|
173 |
+
with gr.Blocks(
|
174 |
+
theme=gr.themes.Soft(
|
175 |
+
primary_hue="blue",
|
176 |
+
secondary_hue="gray",
|
177 |
+
font=[gr.themes.GoogleFont("Roboto"), "Arial", "sans-serif"],
|
178 |
+
radius_size="lg",
|
179 |
+
),
|
180 |
+
css=custom_css
|
181 |
+
) as main:
|
182 |
+
# Welcome banner
|
183 |
+
gr.Markdown(
|
184 |
+
"""
|
185 |
+
# Welcome to SARATH's Portfolio
|
186 |
+
**BTech Final-Year Student | AI & ML Enthusiast | Job Seeker**
|
187 |
+
Explore my projects, skills, and resume below.
|
188 |
+
""",
|
189 |
+
elem_classes=["welcome-banner"]
|
190 |
+
)
|
191 |
+
|
192 |
+
with gr.Row():
|
193 |
+
# Sidebar for navigation
|
194 |
+
with gr.Column(scale=1, min_width=200):
|
195 |
+
gr.Markdown("## Navigation", elem_classes=["sidebar-title"])
|
196 |
+
gr.Button("Chat with SARATH", link="#chat", elem_classes=["nav-btn"])
|
197 |
+
gr.Button("View Resume", link="#resume", elem_classes=["nav-btn"])
|
198 |
+
gr.Button("Download Resume", link="path/to/resume.pdf", elem_classes=["download-btn"])
|
199 |
+
gr.Markdown("### Connect with Me")
|
200 |
+
gr.HTML(
|
201 |
+
f"""
|
202 |
+
<div class='footer'>
|
203 |
+
<a href='https://github.com/your-profile' aria-label='GitHub Profile'>
|
204 |
+
<img src='data:image/png;base64,{github_logo_encoded}' alt='GitHub Logo'>
|
205 |
+
</a>
|
206 |
+
<a href='https://linkedin.com/in/your-profile' aria-label='LinkedIn Profile'>
|
207 |
+
<img src='data:image/png;base64,{linkedin_logo_encoded}' alt='LinkedIn Logo'>
|
208 |
+
</a>
|
209 |
+
<a href='https://your-website.com' aria-label='Personal Website'>
|
210 |
+
<img src='data:image/png;base64,{website_logo_encoded}' alt='Website Logo'>
|
211 |
+
</a>
|
212 |
+
</div>
|
213 |
+
"""
|
214 |
)
|
|
|
215 |
|
216 |
+
# Main content
|
217 |
+
with gr.Column(scale=4):
|
218 |
+
with gr.Tabs():
|
219 |
+
with gr.TabItem("Chat with SARATH", id="chat"):
|
220 |
+
gr.Markdown(
|
221 |
+
"""
|
222 |
+
### Ask Me Anything!
|
223 |
+
I'm SARATH, a BTech student specializing in AI/ML. Ask about my projects, skills, or experience.
|
224 |
+
""",
|
225 |
+
elem_classes=["tab-header"]
|
226 |
+
)
|
227 |
+
gr.ChatInterface(
|
228 |
+
respond,
|
229 |
+
chatbot=gr.Chatbot(
|
230 |
+
height="60vh",
|
231 |
+
show_copy_button=True,
|
232 |
+
bubble_full_width=False,
|
233 |
+
avatar_images=(None, "Images/ai-logo.png"),
|
234 |
+
elem_classes=["chatbot"]
|
235 |
+
),
|
236 |
+
examples=[
|
237 |
+
{"text": "Tell me about yourself", "tooltip": "Learn about SARATH's background and interests."},
|
238 |
+
{"text": "Can you walk me through some of your recent projects?", "tooltip": "Explore SARATH's project portfolio."},
|
239 |
+
{"text": "What specific skills do you bring to AI/ML initiatives?", "tooltip": "Discover SARATH's technical expertise."},
|
240 |
+
{"text": "How do you stay updated with AI/ML trends?", "tooltip": "Understand SARATH's learning approach."},
|
241 |
+
],
|
242 |
+
additional_inputs=[
|
243 |
+
gr.Slider(
|
244 |
+
minimum=0.1,
|
245 |
+
maximum=1.0,
|
246 |
+
value=0.4,
|
247 |
+
label="Temperature",
|
248 |
+
info="Controls response creativity"
|
249 |
+
),
|
250 |
+
gr.Slider(
|
251 |
+
minimum=512,
|
252 |
+
maximum=2048,
|
253 |
+
value=1024,
|
254 |
+
step=512,
|
255 |
+
label="Max Tokens",
|
256 |
+
info="Controls response length"
|
257 |
+
)
|
258 |
+
],
|
259 |
+
submit_btn=gr.Button("Send", variant="primary", elem_classes=["submit-btn"]),
|
260 |
+
retry_btn=None,
|
261 |
+
undo_btn=None,
|
262 |
+
clear_btn=gr.Button("Clear Chat", variant="secondary"),
|
263 |
+
loading_component=gr.HTML("<div class='spinner'></div>")
|
264 |
+
)
|
265 |
+
gr.Markdown(md.description)
|
266 |
+
|
267 |
+
with gr.TabItem("Resume", id="resume"):
|
268 |
+
gr.Markdown(
|
269 |
+
"""
|
270 |
+
### My Resume
|
271 |
+
Below is a detailed overview of my education, skills, and projects.
|
272 |
+
""",
|
273 |
+
elem_classes=["tab-header"]
|
274 |
+
)
|
275 |
+
gr.Markdown(data)
|
276 |
+
gr.Button("Download Resume", link="path/to/resume.pdf", elem_classes=["download-btn"])
|
277 |
|
278 |
+
# Footer
|
279 |
+
gr.HTML(
|
280 |
+
f"""
|
281 |
+
<div class='footer' role='contentinfo' aria-label='Social Links and Contact'>
|
282 |
+
<a href='https://github.com/your-profile' aria-label='GitHub Profile'>
|
283 |
+
<img src='data:image/png;base64,{github_logo_encoded}' alt='GitHub Logo'>
|
284 |
+
</a>
|
285 |
+
<a href='https://linkedin.com/in/your-profile' aria-label='LinkedIn Profile'>
|
286 |
+
<img src='data:image/png;base64,{linkedin_logo_encoded}' alt='LinkedIn Logo'>
|
287 |
+
</a>
|
288 |
+
<a href='https://your-website.com' aria-label='Personal Website'>
|
289 |
+
<img src='data:image/png;base64,{website_logo_encoded}' alt='Website Logo'>
|
290 |
+
</a>
|
291 |
+
</div>
|
292 |
+
<p styleEAD: style='text-align: center;'>© {datetime.datetime.now().year} SARATH. All rights reserved.</p>
|
293 |
+
"""
|
294 |
+
)
|
295 |
|
296 |
if __name__ == "__main__":
|
297 |
main.launch(share=True)
|